config.py

"""
Database configuration for ORM benchmarks.

You can use a local PostgreSQL instance or a free cloud database like:
- Neon (https://neon.tech) - Free tier available
- Supabase (https://supabase.com) - Free tier available
- ElephantSQL (https://elephantsql.com) - Free tier available

For accurate benchmarks, a local database is recommended to eliminate network latency.
"""

import os

# Database configuration - set these environment variables or modify directly
DATABASE_CONFIG = {
    "host": os.getenv("DB_HOST", "localhost"),
    "port": int(os.getenv("DB_PORT", "5432")),
    "user": os.getenv("DB_USER", "postgres"),
    "password": os.getenv("DB_PASSWORD", "postgres"),
    "database": os.getenv("DB_NAME", "orm_benchmark"),
}

# Connection URLs
def get_database_url() -> str:
    """Get PostgreSQL connection URL for SQLAlchemy."""
    return (
        f"postgresql+asyncpg://{DATABASE_CONFIG['user']}:{DATABASE_CONFIG['password']}"
        f"@{DATABASE_CONFIG['host']}:{DATABASE_CONFIG['port']}/{DATABASE_CONFIG['database']}"
    )

def get_tortoise_config() -> dict:
    """Get Tortoise ORM configuration."""
    return {
        "connections": {
            "default": {
                "engine": "tortoise.backends.asyncpg",
                "credentials": {
                    "host": DATABASE_CONFIG["host"],
                    "port": DATABASE_CONFIG["port"],
                    "user": DATABASE_CONFIG["user"],
                    "password": DATABASE_CONFIG["password"],
                    "database": DATABASE_CONFIG["database"],
                },
            }
        },
        "apps": {
            "models": {
                "models": ["models.tortoise_models"],
                "default_connection": "default",
            }
        },
    }

# Benchmark settings
BENCHMARK_SETTINGS = {
    "num_records": 1000,      # Number of records to create for bulk operations
    "num_iterations": 100,    # Number of iterations for each benchmark
    "warmup_iterations": 10,  # Warmup iterations before actual benchmarking
}