ibm-db-sa

import os from sqlalchemy import create_engine, text # Get credentials from environment variables # Ensure these are set in your environment or replace with actual values DB2_USER = os.environ.get('DB2_USER', 'db2inst1') DB2_PASSWORD = os.environ.get('DB2_PASSWORD', 'password') DB2_HOST = os.environ.get('DB2_HOST', 'localhost') DB2_PORT = os.environ.get('DB2_PORT', '50000') DB2_DATABASE = os.environ.get('DB2_DATABASE', 'SAMPLE') INFORMIX_SERVER = os.environ.get('INFORMIX_SERVER', '') # Optional for Informix # Construct the connection string connection_string = f"ibm_db_sa://{DB2_USER}:{DB2_PASSWORD}@{DB2_HOST}:{DB2_PORT}/{DB2_DATABASE}" if INFORMIX_SERVER: connection_string += f"?INFORMIXSERVER={INFORMIX_SERVER}" print(f"Attempting to connect to: {connection_string.split('@')[-1].split('?')[0]}") try: engine = create_engine(connection_string) with engine.connect() as connection: # Example query for Db2. For Informix, adjust query (e.g., SELECT FIRST 1 current FROM systables) result = connection.execute(text("SELECT CURRENT_TIMESTAMP AS current_time FROM SYSIBM.SYSDUMMY1")) for row in result: print(f"Successfully connected! Current Db2 timestamp: {row.current_time}") except Exception as e: print(f"Error connecting or executing query: {e}") print("Please ensure your Db2/Informix server is running, credentials are correct, and necessary client drivers are installed.")