Update quality_audit_job.py

quality_audit_job.py

@@ -2,20 +2,18 @@ import os
 import re
 import time
 import sys
-import logging
 from supabase import create_client
 
 # 强制不缓存输出
-logging.basicConfig(level=logging.INFO, format='%(message)s', stream=sys.stdout)
-logger = logging.getLogger(__name__)
+def log(msg):
+    print(msg, flush=True)
 
-# --- 核心修正点：匹配你截图里的变量名 ---
+# 1. 匹配你 Space 里的真实变量名
 url = os.environ.get("SUPABASE_URL")
-# 尝试读取截图中的 SERVICE_ROLE_KEY
 key = os.environ.get("SUPABASE_SERVICE_ROLE_KEY") 
 
 if not url or not key:
-    print(f"❌ 环境变量缺失! URL: {'OK' if url else 'MISSING'}, KEY: {'OK' if key else 'MISSING'}", flush=True)
+    log(f"❌ 环境变量缺失! URL: {'OK' if url else 'MISSING'}, KEY: {'OK' if key else 'MISSING'}")
     sys.exit(1)
 
 supabase = create_client(url, key)
@@ -39,29 +37,30 @@ def analyze_text(text):
     return {"is_filled": not is_empty, "char_count": char_count, "has_garbage_chars": has_garbage, "quality_score": score}
 
 def run_audit():
-    # 你的数据湖大小
+    # 你的数据湖总数
     total_count = 7038 
-    logger.info(f"🚀 审计引擎启动 | 目标量: {total_count}", flush=True)
+    log(f"🚀 审计引擎启动 | 目标量: {total_count}")
 
     while True:
         try:
-            # --- 修正点：不使用 RPC，改用普通 COUNT ---
-            # 每个植物审计3个字段，所以 count / 3
+            # 2. 统计已审计数量 (每个植物有 3 个字段，所以除以 3)
             count_res = supabase.table("data_quality_audit").select("id", count="exact").execute()
-            processed_plants = (count_res.count or 0) // 3
+            processed_count = (count_res.count or 0) // 3
             
-            # 打印进度条
-            print(f"📊 进度: {get_progress_bar(processed_plants, total_count)} ({processed_plants}/{total_count})", flush=True)
+            # 显示进度条
+            bar = get_progress_bar(processed_count, total_count)
+            log(f"📊 进度: {bar} ({processed_count}/{total_count})")
 
-            # 水位线逻辑
+            # 3. 获取水位线 (基于审计表的最后更新时间)
             last_res = supabase.table("data_quality_audit").select("updated_at").order("updated_at", desc=True).limit(1).execute()
             last_time = last_res.data[0]['updated_at'] if last_res.data else "1970-01-01T00:00:00Z"
             
-            # 抓取主表变动
+            # 4. 抓取主表新变动
             res = supabase.table("bs4_plants").select("plant_id, edible_uses, medicinal_uses, cultivation_details, updated_at")\
                 .gt("updated_at", last_time).order("updated_at").limit(20).execute()
 
             if not res.data:
+                # 如果没数据，休眠 30 秒
                 time.sleep(30)
                 continue
 
@@ -76,7 +75,7 @@ def run_audit():
                     }).execute()
             
         except Exception as e:
-            print(f"❌ 运行异常: {e}", flush=True)
+            log(f"❌ 运行异常: {e}")
             time.sleep(10)
 
 if __name__ == "__main__":