Call the credit update function

Author: derjogi

app/core/scheduler.py

@@ -0,0 +1,15 @@
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from apscheduler.triggers.cron import CronTrigger
+from app.services.credits import CreditService
+
+scheduler = AsyncIOScheduler()
+
+def init_schedulers():
+    scheduler.add_job(
+        CreditService.refresh_daily_credits,
+        trigger=CronTrigger(hour=0, minute=0),  # Run at midnight
+        id='refresh_credits',
+        name='Refresh daily credits',
+        replace_existing=True
+    )
+    scheduler.start()

main.py

@@ -4,6 +4,8 @@
 from slowapi.errors import RateLimitExceeded
 from slowapi import _rate_limit_exceeded_handler
 from app.core.config import settings
+from contextlib import asynccontextmanager
+from app.core.scheduler import init_schedulers, scheduler
 
 from app.services.analyzer import init_nltk_resources
 import app.api.v1.routes as v1
@@ -13,9 +15,19 @@
 os.environ.clear()
 load_dotenv(override=True)
 
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
+    init_schedulers()
+    init_nltk_resources()
+    yield
+    # Shutdown
+    scheduler.shutdown()
+
 # Main app for requests at the base url. This will serve documentation etc, but all API endpoints must go to a versioned one.
 app = FastAPI(
-  title=settings.PROJECT_NAME
+  title=settings.PROJECT_NAME,
+  lifespan=lifespan,
 )
 
 ### V1 ###
@@ -28,8 +40,6 @@
 v1_app.include_router(v1.auth.router)
 ### /V1 ###
 
-init_nltk_resources()
-
 ACCESS_CONTROL_ALLOW_CREDENTIALS = os.environ.get(
     'ACCESS_CONTROL_ALLOW_CREDENTIALS', 
     True)
@@ -50,4 +60,4 @@
                    allow_headers=ACCESS_CONTROL_ALLOW_HEADERS
                    )
 
-app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
+app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)

poetry.lock

@@ -30,6 +30,7 @@ openpyxl = "^3.1.5"
 pytest-asyncio = "^0.25.2"
 supabase = "^2.11.0"
 supabase-pydantic = "^0.15.2"
+apscheduler = "^3.11.0"
 
 
 [tool.poetry.group.dev.dependencies]