新增素材更新功能

2026-06-23 14:16:31 +08:00
parent b88546ab66
commit 242030c50b
11 changed files with 1109 additions and 7 deletions
@@ -0,0 +1,68 @@
+from datetime import datetime, timedelta, timezone
+import asyncio
+import os
+import logging
+
+from app.services.material_consumption_queue import sync_all_advertisers_consumption
+
+LOG_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "logs")
+os.makedirs(LOG_DIR, exist_ok=True)
+
+logger = logging.getLogger("material_consumption_task")
+logger.setLevel(logging.INFO)
+
+
+class DailyRotatingFileHandler(logging.FileHandler):
+    def __init__(self, directory, encoding=None):
+        self.directory = directory
+        filename = self._get_log_filename()
+        super().__init__(filename, encoding=encoding)
+    
+    def _get_log_filename(self):
+        return os.path.join(self.directory, f"material_consumption_task-{datetime.now(timezone.utc).strftime('%Y-%m-%d')}.log")
+    
+    def emit(self, record):
+        current_filename = self._get_log_filename()
+        if self.baseFilename != current_filename:
+            self.close()
+            self.baseFilename = current_filename
+            self.stream = self._open()
+        super().emit(record)
+
+
+if not logger.handlers:
+    handler = DailyRotatingFileHandler(LOG_DIR, encoding="utf-8")
+    handler.setFormatter(logging.Formatter("%(asctime)s - %(levelname)s - %(message)s", "%Y-%m-%d %H:%M:%S"))
+    logger.addHandler(handler)
+
+
+async def schedule_daily_sync():
+    """每天9点自动同步素材消耗数据"""
+    logger.info("Daily material consumption sync task started")
+    
+    while True:
+        try:
+            now = datetime.now(timezone.utc)
+            target_time = now.replace(hour=9, minute=0, second=0, microsecond=0)
+            
+            if now >= target_time:
+                target_time += timedelta(days=1)
+            
+            wait_seconds = (target_time - now).total_seconds()
+            logger.info(f"Waiting {wait_seconds/3600:.1f} hours until next sync at {target_time}")
+            
+            await asyncio.sleep(wait_seconds)
+            
+            logger.info("Starting daily material consumption sync...")
+            try:
+                result = await sync_all_advertisers_consumption()
+                logger.info(f"Daily sync completed: {result['message']}")
+            except Exception as e:
+                logger.error(f"Daily sync failed: {e}")
+        
+        except asyncio.CancelledError:
+            logger.info("Daily sync task cancelled")
+            break
+        except Exception as e:
+            logger.error(f"Error in schedule_daily_sync: {e}")
+            await asyncio.sleep(60)