feat: add celery & redis for background data processing

This commit is contained in:
2026-03-03 12:27:14 +00:00
parent 2e0e842525
commit 3a58705635
3 changed files with 45 additions and 7 deletions

19
server/queue/tasks.py Normal file
View File

@@ -0,0 +1,19 @@
import pandas as pd
from server.queue.celery_app import celery
from server.analysis.enrichment import DatasetEnrichment
@celery.task(bind=True, max_retries=3)
def process_dataset(self, dataset_id: int, posts: list, topics: dict):
from server.db.database import PostgresConnector
from server.core.datasets import DatasetManager
db = PostgresConnector()
dataset_manager = DatasetManager(db)
df = pd.DataFrame(posts)
processor = DatasetEnrichment(df, topics)
enriched_df = processor.enrich()
dataset_manager.save_dataset_content(dataset_id, enriched_df)