diff --git a/conf/db.py b/conf/db.py index df6b738..e63f790 100644 --- a/conf/db.py +++ b/conf/db.py @@ -21,5 +21,8 @@ db_url = f"mysql+pymysql://{db_cfg['user']}:{db_cfg['password']}@{db_cfg['host'] engine = create_engine(db_url, pool_pre_ping=True) Session = sessionmaker(bind=engine) +def get_engine(): + return engine + def get_session(): return Session()