Update achievements.py

achievements.py

@@ -1,24 +1,37 @@
-import sqlite3
+import os
 from datetime import datetime, timedelta
 import pytz
+from supabase import create_client, Client
 
 tw_tz = pytz.timezone('Asia/Taipei')
 
+# 💡 初始化 Supabase 雲端連線
+SUPABASE_URL = os.getenv("SUPABASE_URL", "")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY", "")
+
 def update_user_stats_and_get_awards(user_id, db_file, action="chat"):
-    conn = sqlite3.connect(db_file, timeout=20)
-    c = conn.cursor()
+    # 備註：db_file 參數保留是為了不改動 app.py 的呼叫方式，但實際已不使用
+    if not SUPABASE_URL or not SUPABASE_KEY:
+        print("Achievement Error: 找不到 Supabase 金鑰")
+        return []
+        
+    supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
     today_str = datetime.now(tw_tz).strftime('%Y-%m-%d')
     yesterday_str = (datetime.now(tw_tz) - timedelta(days=1)).strftime('%Y-%m-%d')
     awards = []
 
     try:
-        c.execute("SELECT count, streak, last_active_date FROM users WHERE user_id = ?", (user_id,))
-        row = c.fetchone()
-        if not row: return []
+        # 💡 1. 從雲端獲取使用者狀態
+        res = supabase.table("users").select("count, streak, last_active_date").eq("user_id", user_id).execute()
+        if not res.data: 
+            return []
         
-        current_count, current_streak, last_active_date = row
+        row = res.data[0]
+        current_count = row.get("count") or 0
+        current_streak = row.get("streak") or 0
+        last_active_date = row.get("last_active_date")
 
-        # 1. 互動與連續天數結算
+        # 💡 2. 互動與連續天數結算 (並更新上雲端)
         if action == "chat":
             current_count += 1
             if last_active_date == yesterday_str:
@@ -26,30 +39,31 @@ def update_user_stats_and_get_awards(user_id, db_file, action="chat"):
             elif last_active_date != today_str:
                 current_streak = 1
             
-            c.execute("UPDATE users SET count = ?, streak = ?, last_active_date = ? WHERE user_id = ?", 
-                      (current_count, current_streak, today_str, user_id))
-            conn.commit()
+            # 將最新次數與天數寫回雲端金庫
+            supabase.table("users").update({
+                "count": current_count, 
+                "streak": current_streak, 
+                "last_active_date": today_str
+            }).eq("user_id", user_id).execute()
 
-        # 2. 結算反饋次數
-        c.execute("SELECT COUNT(*) FROM user_surveys WHERE user_id = ? AND suggestion IS NOT NULL", (user_id,))
-        feedback_count = c.fetchone()[0]
+        # 💡 3. 從雲端精算反饋次數
+        fb_res = supabase.table("user_surveys").select("id", count="exact").eq("user_id", user_id).not_is("suggestion", "null").execute()
+        feedback_count = fb_res.count or 0
 
-        # 3. 檢查級距 (改成字典格式)
+        # 💡 4. 檢查級距與派發獎狀
         # 🥇 勤學獎 (天數) -> 綠色
         if action == "chat" and last_active_date != today_str:
             s_m = {1: "🐣【破殼而出】", 3: "🥉【初燃火種】", 7: "🥈【週日守候】", 15: "🥇【半月恆心】", 30: "💎【月滿族語】", 
                    60: "🌊【雙月波瀾】", 100: "⛰️【百日築基】", 200: "⚔️【堅韌意志】", 300: "☀️【歲月傳人】"}
             if current_streak in s_m: 
-                # 💡 針對第 1 天給予專屬的歡迎詞
                 desc = "踏出學習的第一步，歡迎加入族語的大家庭！" if current_streak == 1 else f"連續堅持 {current_streak} 天，族語火種永不熄滅！"
                 awards.append({"type": "勤學獎", "title": s_m[current_streak], "desc": desc})
 
         # 🏹 積極獎 (次數) -> 黑色
         if action == "chat":
             c_m = {1: "💬【初次發聲】", 10: "🌱【初試啼聲】", 30: "🍃【語感啟蒙】", 50: "🦌【林間尋蹤】", 100: "🏹【語林勇士】", 
-                   300: "🦅【文化獵人】", 500: "🗺️【部落嚮導】", 700: "🧭【語海導航】", 1000: "👑【傳奇語者】"}
+                   300: "🦅【文化 সাংস্কৃতিক獵人】", 500: "🗺️【部落嚮導】", 700: "🧭【語海導航】", 1000: "👑【傳奇語者】"}
             if current_count in c_m: 
-                # 💡 針對第 1 次互動給予專屬的鼓勵詞
                 desc = "勇敢說出第一句族語，您的聲音已經被聽見了！" if current_count == 1 else f"累積互動 {current_count} 次，您的聲音已被大家看見！"
                 awards.append({"type": "積極獎", "title": c_m[current_count], "desc": desc})
 
@@ -58,9 +72,10 @@ def update_user_stats_and_get_awards(user_id, db_file, action="chat"):
             f_m = {1: "💌【首份心意】", 10: "🔍【細心觀察家】", 30: "🪓【磨刀尖兵】", 50: "🧶【編織能手】", 100: "🛡️【族語護衛】", 
                    300: "🌾【智慧播種者】", 500: "💡【指路明燈】", 700: "🧱【文化奠基石】", 1000: "🧬【語脈建築師】"}
             if feedback_count in f_m: 
-                # 💡 針對第 1 次建議給予專屬的感謝詞
                 desc = "送出第一份寶貴建議，您是幫助 AI 成長的重要推手！" if feedback_count == 1 else f"您已提供 {feedback_count} 次建議，是 AI 精進的重要養分！"
                 awards.append({"type": "反饋獎", "title": f_m[feedback_count], "desc": desc})
-    except Exception as e: print(f"Achievement Error: {e}")
-    finally: conn.close()
+                
+    except Exception as e: 
+        print(f"Achievement Error: {e}")
+        
     return awards