diff --git a/scoreboard_injector/main.py b/scoreboard_injector/main.py index 2d597fa..71a3131 100644 --- a/scoreboard_injector/main.py +++ b/scoreboard_injector/main.py @@ -34,6 +34,13 @@ async def verify_token(authorization: str = Header(None)): raise HTTPException(status_code=403, detail="Invalid token") return token +# Database functions +async def get_db(): + return await db_pool.acquire() + +async def release_db(conn): + await db_pool.release(conn) + async def send_telegram_alert(message: str, service_id: int = None, service_name: str = None): """Send alert to telegram bot"""