starrocks

import os from sqlalchemy import create_engine, text # Configure connection details via environment variables STARROCKS_USER = os.environ.get('STARROCKS_USER', 'root') STARROCKS_PASSWORD = os.environ.get('STARROCKS_PASSWORD', '') STARROCKS_HOST = os.environ.get('STARROCKS_HOST', 'localhost') STARROCKS_PORT = os.environ.get('STARROCKS_PORT', '9030') STARROCKS_DATABASE = os.environ.get('STARROCKS_DATABASE', 'mydatabase') # Construct connection string connection_string = ( f"starrocks://{STARROCKS_USER}:" # User and optional password f"{STARROCKS_PASSWORD}@{STARROCKS_HOST}:{STARROCKS_PORT}/" f"{STARROCKS_DATABASE}" ) # Create a SQLAlchemy engine engine = create_engine(connection_string) try: # Establish a connection and execute a basic query with engine.connect() as connection: print("Connection successful!") # Ensure 'mytable' exists in 'mydatabase' for this example result = connection.execute(text("SELECT 1 + 1")).scalar() print(f"Query result: {result}") # Example: Fetching data from a table (uncomment and replace if 'mytable' exists) # rows = connection.execute(text("SELECT * FROM mytable LIMIT 2")).fetchall() # print(rows) except Exception as e: print(f"An error occurred: {e}")