database.py

import os
from pathlib import Path

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

try:
	from dotenv import load_dotenv
except ImportError:
	load_dotenv = None


if load_dotenv:
	load_dotenv()
else:
	env_file = Path(__file__).resolve().parent / ".env"
	if env_file.exists():
		for line in env_file.read_text().splitlines():
			line = line.strip()
			if not line or line.startswith("#") or "=" not in line:
				continue
			key, value = line.split("=", 1)
			if key.strip() == "DATABASE_URL" and not os.getenv("DATABASE_URL"):
				os.environ["DATABASE_URL"] = value.strip().strip('"').strip("'")
				break

db_url = os.getenv("DATABASE_URL")

if not db_url:
	raise ValueError("DATABASE_URL is not set. Add it to your .env file.")

engine = create_engine(db_url, echo=True)

session = sessionmaker(autocommit=False, autoflush=False, bind=engine)