Spaces:
Sleeping
Sleeping
| from sqlalchemy import create_engine | |
| from sqlalchemy.ext.declarative import declarative_base | |
| from sqlalchemy.orm import sessionmaker | |
| import redis | |
| from .settings import settings | |
| # PostgreSQL Setup | |
| engine = create_engine(settings.database_url) | |
| SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) | |
| Base = declarative_base() | |
| # Redis Setup | |
| redis_client = redis.from_url(settings.redis_url, decode_responses=True) | |
| # Dependency to get a DB session | |
| def get_db(): | |
| db = SessionLocal() | |
| try: | |
| yield db | |
| finally: | |
| db.close() | |
| # Dependency to get a Redis client | |
| def get_redis(): | |
| return redis_client | |