mysqlclient

import MySQLdb import os DB_HOST = os.environ.get('MYSQL_HOST', '127.0.0.1') DB_USER = os.environ.get('MYSQL_USER', 'root') DB_PASSWORD = os.environ.get('MYSQL_PASSWORD', 'your_password') # Replace with a secure method for production DB_NAME = os.environ.get('MYSQL_DATABASE', 'testdb') try: # Establish a connection conn = MySQLdb.connect( host=DB_HOST, user=DB_USER, password=DB_PASSWORD, database=DB_NAME ) cursor = conn.cursor() # Execute a query cursor.execute("SELECT VERSION();") version = cursor.fetchone() print(f"Database version: {version[0]}") # Example: Create a table (if it doesn't exist) cursor.execute("CREATE TABLE IF NOT EXISTS my_table (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255))") print("Table 'my_table' ensured.") # Example: Insert data cursor.execute("INSERT INTO my_table (name) VALUES (%s)", ("Test Name",)) conn.commit() print("Data inserted.") # Example: Read data cursor.execute("SELECT id, name FROM my_table") rows = cursor.fetchall() for row in rows: print(f"ID: {row[0]}, Name: {row[1]}") finally: # Close the cursor and connection if 'cursor' in locals() and cursor: cursor.close() if 'conn' in locals() and conn: conn.close() print("Database connection closed.")