Create scoring.py

src/scoring.py

@@ -0,0 +1,34 @@
+from datetime import datetime, timedelta
+from sqlmodel import select
+from .models import Visti, TokenBalance
+from .db import get_session
+
+# 重みは適宜調整
+W_RECENCY = 0.5
+W_FREQUENCY = 0.3
+W_POINTS30 = 0.2
+
+def calc_score(user_id: int) -> float:
+    now = datetime.utcnow()
+    cutoff_30 = now - timedelta(days=30)
+    with get_session() as s:
+        visits = s.exec(select(Visit).where(Visit.user_id == user_id).order_by(Visit.created_at.desc())).all()
+        if not visits:
+            return 0.0
+        last_visit = visits[0].created_at
+        days_since = (now - last_visit).days
+        recency_score = max(0.0, 1.0 -days_since/ 30.0) #30日で減衰
+        freq_score = min(1.0, len([v for v in visits if v.created_at > now -timedelta(days=90)])/ 10.0) #90日10回で満点
+        points30 = s.exec(select(TokenBalance).where(TokenBalance.user_id == user_id)).all()
+        points_amounts = sum(tb.amount for tb in points30)
+        points30_score = min(1.0, points_amounts / 1000.0)
+        return W_RECENCY * recency_score + W_FREQUENCY * freq_score + W_POINTS30 *points30_score
+    
+def segment(socre: float) -> str:
+    if score >= 0.75:
+        return "loyal"
+    if score >= 0.4:
+        return "active"
+    if score > 0.0:
+        return "at-risk"
+    return "new"