Spaces:
Running
Running
| # db/database.py | |
| import os | |
| from sqlalchemy import create_engine | |
| from sqlalchemy.ext.declarative import declarative_base | |
| from sqlalchemy.orm import sessionmaker | |
| from fastapi.security import OAuth2PasswordBearer | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| # β PostgreSQL URL from Neon | |
| SQLALCHEMY_DATABASE_URL = os.getenv("NEON_DATABASE_URL") # put this in .env | |
| # β Create PostgreSQL engine | |
| engine = create_engine(SQLALCHEMY_DATABASE_URL) | |
| # β PostgreSQL session | |
| SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) | |
| # β Base class for models | |
| Base = declarative_base() | |
| # β Token config | |
| oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/login") | |