import pymysql
import os
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from dotenv import load_dotenv
load_dotenv()


DB_USER = os.getenv("DB_USER")
DB_PASSWORD = os.getenv("DB_PASSWORD")
DB_NAME = os.getenv("DB_NAME")
try:
    DATABASE_URL = f"mysql+pymysql://{DB_USER}:{DB_PASSWORD}@localhost/{DB_NAME}"
    print(DATABASE_URL)
    engine = create_engine(
    	DATABASE_URL,
    	pool_size=10,         # Increase base pool size
    	max_overflow=20,      # Allow additional overflow connections if needed
    	pool_timeout=60,      # Wait time in seconds before timing out
    	pool_recycle=1800,    # Recycle connections every 30 minutes to avoid stale connections
    	pool_pre_ping=True
    )
    SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
    db = SessionLocal()
    print("Connection successful!")
    db.close()
except pymysql.err.OperationalError as e:
   print(f"Error: {e}")