diff --git a/api/kiwi_vpn_api/db/connection.py b/api/kiwi_vpn_api/db/connection.py index 23c537d..9f65c7c 100644 --- a/api/kiwi_vpn_api/db/connection.py +++ b/api/kiwi_vpn_api/db/connection.py @@ -29,29 +29,3 @@ class Connection: yield db finally: db.close() - - -ENGINE: Engine | None = None -SESSION_LOCAL: sessionmaker | None = None - - -def reconnect(engine: Engine) -> None: - global ENGINE, SESSION_LOCAL - - ENGINE = engine - SESSION_LOCAL = sessionmaker( - autocommit=False, autoflush=False, bind=engine, - ) - ORMBaseModel.metadata.create_all(bind=engine) - - -async def get() -> Generator[Session | None, None, None]: - if SESSION_LOCAL is None: - yield None - - else: - db = SESSION_LOCAL() - try: - yield db - finally: - db.close()