Sin descripción

utils.py 747B

123456789101112131415161718192021
  1. from django.conf import settings
  2. import pyodbc
  3. def create_db_connection(env="dev"):
  4. try:
  5. connection_details = settings.DATABASE_CONNECTIONS.get(env)
  6. if not connection_details:
  7. raise ValueError(f"Environment '{env}' not found in DATABASE_CONNECTIONS")
  8. connection_string = (
  9. f"DRIVER={connection_details['DRIVER']};"
  10. f"SERVER={connection_details['SERVER']};"
  11. f"DATABASE={connection_details['DATABASE']};"
  12. f"UID={connection_details['UID']};"
  13. f"PWD={connection_details['PWD']}"
  14. )
  15. return pyodbc.connect(connection_string)
  16. except Exception as e:
  17. raise RuntimeError(f"Failed to establish database connection: {e}")