fix docker config

This commit is contained in:
h z
2024-12-03 14:35:25 +00:00
parent 973d24ddd2
commit 23b9db19a8
2 changed files with 9 additions and 3 deletions

View File

@@ -2,13 +2,14 @@
from contextlib import contextmanager
from db.models import Base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine
import env_provider
from env_provider import DB_HOST, DB_PORT, DB_NAME, DB_USER, DB_PASSWORD
engine = create_engine(env_provider.DB_CONNECT_STRING)
engine = create_engine(f"mysql+pymysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}")
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@contextmanager
def get_db():
db = SessionLocal()