Initial project commit

2026-05-09 16:40:29 +08:00
commit 02b0259a9e
267 changed files with 54891 additions and 0 deletions
@@ -0,0 +1,55 @@
+import logging
+from datetime import datetime, timezone
+
+from app.celery_app import celery_app
+
+logger = logging.getLogger(__name__)
+
+
+@celery_app.task
+def aggregate_daily_stats():
+    """Aggregate daily statistics and store to Redis for dashboard consumption."""
+    from sqlalchemy import func, select
+    from sqlalchemy.orm import Session
+
+    from app.db.session import sync_engine
+    from app.models.comment import Comment
+    from app.models.event import Event
+    from app.models.favorite import Favorite
+    from app.models.rating import Rating
+    from app.models.shooting import ShootingRequest
+    from app.models.spot import Spot
+    from app.models.user import User
+
+    try:
+        with Session(sync_engine) as session:
+            stats = {
+                "users": session.execute(select(func.count(User.id))).scalar() or 0,
+                "spots": session.execute(select(func.count(Spot.id))).scalar() or 0,
+                "approved_spots": session.execute(
+                    select(func.count(Spot.id)).where(Spot.audit_status == "approved")
+                ).scalar() or 0,
+                "comments": session.execute(select(func.count(Comment.id))).scalar() or 0,
+                "ratings": session.execute(select(func.count(Rating.id))).scalar() or 0,
+                "favorites": session.execute(select(func.count(Favorite.id))).scalar() or 0,
+                "shootings": session.execute(select(func.count(ShootingRequest.id))).scalar() or 0,
+                "events": session.execute(select(func.count(Event.id))).scalar() or 0,
+                "generated_at": datetime.now(timezone.utc).isoformat(),
+            }
+
+        import redis
+        r = redis.from_url(str(_get_redis_url()))
+        r.hset("ciyuan:daily_stats", mapping={k: str(v) for k, v in stats.items()})
+        r.expire("ciyuan:daily_stats", 86400 * 2)
+
+        logger.info("Daily stats aggregated: %s", stats)
+        return stats
+
+    except Exception:
+        logger.exception("Failed to aggregate daily stats")
+        raise
+
+
+def _get_redis_url():
+    from app.core.config import settings
+    return settings.REDIS_URL