日报增加Cookie有效期检测(ALF字段), 按剩余天数分级预警

backend/task_scheduler/app/main.py

@@ -651,6 +651,7 @@ async def _build_daily_report() -> str:
     from shared.models.account import Account
     from shared.models.signin_log import SigninLog
     from shared.models.user import User
+    from shared.crypto import decrypt_cookie, derive_key
 
     today_start = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
     now = datetime.now()
@@ -729,19 +730,37 @@ async def _build_daily_report() -> str:
                 if msg:
                     rank_details.append({"name": name, "topic": topic, "message": msg})
 
-            # 4. Cookie 即将失效的账号（超过 3 天未检查）
+            # 4. Cookie 过期时间检测（从 ALF 字段解析）
-            stale_cutoff = now - timedelta(days=3)
+            cookie_expiry = []
-            stale_result = await session.execute(
+            all_accounts_result = await session.execute(
-                select(Account.remark, Account.weibo_user_id, Account.last_checked_at)
+                select(Account.id, Account.remark, Account.weibo_user_id,
-                .where(Account.status == "active")
+                       Account.encrypted_cookies, Account.iv, Account.status)
-                .where(
+                .where(Account.status.in_(["active", "pending"]))
-                    (Account.last_checked_at < stale_cutoff) | (Account.last_checked_at == None)
-                )
             )
-            stale_accounts = [
+            key = derive_key(shared_settings.COOKIE_ENCRYPTION_KEY)
-                (row[0] or row[1], row[2].strftime("%m-%d %H:%M") if row[2] else "从未")
+            for acc_row in all_accounts_result.all():
-                for row in stale_result.all()
+                acc_id, remark, uid, enc_cookies, iv, status = acc_row
-            ]
+                name = remark or uid
+                try:
+                    cookie_str = decrypt_cookie(enc_cookies, iv, key)
+                    alf = ""
+                    for pair in cookie_str.split(";"):
+                        pair = pair.strip()
+                        if pair.startswith("ALF="):
+                            alf = pair.split("=", 1)[1].strip()
+                            break
+                    if alf and alf.isdigit():
+                        expire_dt = datetime.fromtimestamp(int(alf))
+                        remain_days = (expire_dt - now).days
+                        cookie_expiry.append({
+                            "name": name,
+                            "expire": expire_dt.strftime("%m-%d"),
+                            "remain": remain_days,
+                        })
+                    else:
+                        cookie_expiry.append({"name": name, "expire": "未知", "remain": -1})
+                except Exception:
+                    cookie_expiry.append({"name": name, "expire": "解密失败", "remain": -1})
 
     finally:
         await eng.dispose()
@@ -777,10 +796,22 @@ async def _build_daily_report() -> str:
         for r in rank_details:
             lines.append(f"  {r['name']} - {r['topic']}: {r['message']}")
 
-    if stale_accounts:
+    if cookie_expiry:
-        lines += ["", "⚠️ 需要关注"]
+        lines += ["", "🍪 Cookie 有效期"]
-        for name, last in stale_accounts:
+        expiring_soon = []
-            lines.append(f"  {name} (上次检查: {last})")
+        for ce in cookie_expiry:
+            remain = ce["remain"]
+            if remain < 0:
+                lines.append(f"  ⚠️ {ce['name']}: {ce['expire']}")
+            elif remain <= 3:
+                lines.append(f"  🔴 {ce['name']}: {ce['expire']} (剩 {remain} 天，即将过期!)")
+                expiring_soon.append(ce["name"])
+            elif remain <= 7:
+                lines.append(f"  🟡 {ce['name']}: {ce['expire']} (剩 {remain} 天)")
+            else:
+                lines.append(f"  🟢 {ce['name']}: {ce['expire']} (剩 {remain} 天)")
+        if expiring_soon:
+            lines.append(f"  ⚠️ 请尽快重新扫码: {', '.join(expiring_soon)}")
 
     if total_logs == 0:
         lines += ["", "💤 今日暂无签到记录"]

test_fetch_topics.py

@@ -0,0 +1,124 @@
+"""
+验证签到流程是否正常。
+在服务器上执行: docker exec -it weibo-scheduler python -m test_fetch_topics
+或本地: cd backend && python ../test_fetch_topics.py
+
+会从数据库读取第一个 active 账号，解密 Cookie，模拟 _fetch_topics 流程。
+"""
+
+import os
+import sys
+import re
+import asyncio
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "backend"))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__)))
+
+# 兼容容器内运行（/app 目录）
+if os.path.exists("/app/shared"):
+    sys.path.insert(0, "/app")
+
+from shared.config import shared_settings
+from shared.crypto import decrypt_cookie, derive_key
+
+WEIBO_HEADERS = {
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+    "Referer": "https://weibo.com/",
+    "Accept": "*/*",
+    "Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
+}
+
+
+async def main():
+    import httpx
+    from sqlalchemy import select
+    from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
+    from sqlalchemy.orm import sessionmaker
+    from shared.models.account import Account
+
+    engine = create_async_engine(shared_settings.DATABASE_URL, echo=False)
+    Session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+
+    async with Session() as session:
+        result = await session.execute(
+            select(Account).where(Account.status.in_(["active", "pending"])).limit(1)
+        )
+        account = result.scalar_one_or_none()
+        if not account:
+            print("❌ 没有 active/pending 账号")
+            return
+
+        print(f"📱 账号: {account.remark or account.weibo_user_id} (status={account.status})")
+
+        key = derive_key(shared_settings.COOKIE_ENCRYPTION_KEY)
+        cookie_str = decrypt_cookie(account.encrypted_cookies, account.iv, key)
+        cookies = {}
+        for pair in cookie_str.split(";"):
+            pair = pair.strip()
+            if "=" in pair:
+                k, v = pair.split("=", 1)
+                cookies[k.strip()] = v.strip()
+
+        print(f"🍪 Cookie 数量: {len(cookies)}, keys: {list(cookies.keys())}")
+
+        # 检查 Cookie 有效期
+        from datetime import datetime
+        alf = cookies.get("ALF", "")
+        if alf and alf.isdigit():
+            expire_time = datetime.fromtimestamp(int(alf))
+            remain = (expire_time - datetime.now()).days
+            print(f"📅 Cookie 过期时间: {expire_time.strftime('%Y-%m-%d %H:%M:%S')} (还剩 {remain} 天)")
+        else:
+            print(f"📅 ALF 字段: {alf or '无'} (无法判断过期时间)")
+
+    await engine.dispose()
+
+    # 测试 1: 访问 weibo.com
+    print("\n--- 测试 1: GET https://weibo.com/ ---")
+    async with httpx.AsyncClient(timeout=20, follow_redirects=True) as client:
+        resp = await client.get("https://weibo.com/", headers=WEIBO_HEADERS, cookies=cookies)
+        final_url = str(resp.url)
+        print(f"  状态码: {resp.status_code}")
+        print(f"  最终URL: {final_url}")
+        print(f"  是否登录页: {'login.sina.com.cn' in final_url or 'passport' in final_url}")
+
+        if "login.sina.com.cn" in final_url or "passport" in final_url:
+            print("\n❌ Cookie 已失效，被重定向到登录页")
+            return
+
+        xsrf = client.cookies.get("XSRF-TOKEN", "")
+        print(f"  XSRF-TOKEN: {'有' if xsrf else '无'}")
+
+        # 测试 2: 获取超话列表
+        print("\n--- 测试 2: 获取超话列表 ---")
+        headers = {**WEIBO_HEADERS, "X-Requested-With": "XMLHttpRequest"}
+        if xsrf:
+            headers["X-XSRF-TOKEN"] = xsrf
+
+        resp = await client.get(
+            "https://weibo.com/ajax/profile/topicContent",
+            params={"tabid": "231093_-_chaohua", "page": "1"},
+            headers=headers, cookies=cookies,
+        )
+        print(f"  状态码: {resp.status_code}")
+        print(f"  最终URL: {resp.url}")
+
+        try:
+            data = resp.json()
+            print(f"  ok: {data.get('ok')}")
+            topics = data.get("data", {}).get("list", [])
+            print(f"  超话数量: {len(topics)}")
+            for t in topics[:5]:
+                title = t.get("topic_name", "") or t.get("title", "")
+                print(f"    - {title}")
+            if topics:
+                print("\n✅ Cookie 有效，超话获取正常")
+            else:
+                print("\n⚠️ Cookie 可能有效但没有关注超话")
+        except Exception as e:
+            print(f"  ❌ 响应非 JSON: {resp.text[:300]}")
+            print(f"\n❌ 获取超话失败: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())