add scheduler celery

Author: sumaro2101

api_v1/users/tasks.py

@@ -1,4 +1,4 @@
-from config import celery_app
+from config import celery_app, settings
 import asyncio
 
 
@@ -9,3 +9,12 @@ async def time_sleep_task():
     """
     await asyncio.sleep(2.0)
     return 'Task is done'
+
+
+celery_app.conf.beat_schedule = {
+    'test-every-10-seconds': {
+        'task': 'llm_analizer.tasks.test',
+        'schedule': settings.celery.TEST_TIMEDELTA,
+        'args': ('hello',)
+    },
+}

config/celery/connection.py

@@ -41,4 +41,5 @@ def wrapper(*args,
 
 app = Celery(__name__)
 app.conf.broker_url = settings.rabbit.broker_url
+app.conf.timezone = settings.celery.TIMEZONE
 app.autodiscover_tasks(packages=['api_v1.users'])

config/config.py

@@ -1,7 +1,8 @@
 from pathlib import Path
 from pydantic_settings import BaseSettings, SettingsConfigDict
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 from starlette.config import Config
+from celery.schedules import crontab
 
 
 base_dir = Path(__file__).resolve().parent.parent
@@ -11,6 +12,14 @@
 config = Config('.env')
 
 
+class AlembicSettings(BaseModel):
+    """
+    Настройки Alembic
+    """
+    CONFIG_PATH: Path = Path('alembic.ini')
+    MIGRATION_PATH: Path = Path('async_alembic/')
+
+
 class TestDBSettings(BaseModel):
     """
     Настройки тестовой базы данных
@@ -35,6 +44,23 @@ class DBSettings(BaseModel):
     url: str = f'{_engine}://{_owner}:{_password}@{_name}/{_db_name}'
 
 
+class CelerySettings(BaseModel):
+    """
+    Настройки Celery
+    """
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
+    TIMEZONE: str = 'Europe/Moscow'
+    TIMEDELTA_PER_DAY: crontab = crontab(minute=0,
+                                         hour=2,
+                                         day_of_week='*/1',
+                                         day_of_month='*/1',
+                                         month_of_year='*/1',
+                                         )
+    TEST_TIMEDELTA: crontab = crontab(minute='*/1')
+
+
 class RabbitSettings(BaseModel):
     """
     Настройки RabbitMQ
@@ -62,7 +88,9 @@ class Settings(BaseSettings):
     )
     db: DBSettings = DBSettings()
     test_db: TestDBSettings = TestDBSettings()
+    celery: CelerySettings = CelerySettings()
     rabbit: RabbitSettings = RabbitSettings()
+    alembic: AlembicSettings = AlembicSettings()
     debug: bool = bool(int(config('DEBUG')))
     API_PREFIX: str = '/api/v1'
     BASE_DIR: Path = base_dir