v4.5.38: 实时数据查询硬规则(回复前必查)+ check_account.py封装

2026-07-17 22:14:19 +08:00
parent 629ac197de
commit 533c342305
55 changed files with 5437 additions and 599 deletions
@@ -0,0 +1,122 @@
+#!/usr/bin/env python3
+"""信号队列 - 先入库再处理，防止信号丢失"""
+
+import json, os, time, sys, sqlite3
+from datetime import datetime, timezone, timedelta
+
+DB_PATH = os.path.expanduser("~/.hermes/trading/signal_queue.db")
+
+def get_db():
+    os.makedirs(os.path.dirname(DB_PATH), exist_ok=True)
+    conn = sqlite3.connect(DB_PATH)
+    conn.execute("""
+        CREATE TABLE IF NOT EXISTS queue (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            raw_text TEXT NOT NULL,
+            status TEXT DEFAULT 'pending',  -- pending/processing/done/failed
+            created_at TEXT DEFAULT (datetime('now')),
+            processed_at TEXT,
+            result TEXT,
+            error TEXT,
+            retries INTEGER DEFAULT 0
+        )
+    """)
+    conn.commit()
+    return conn
+
+def enqueue(raw_text):
+    """信号入队"""
+    conn = get_db()
+    conn.execute("INSERT INTO queue (raw_text, status) VALUES (?, 'pending')", (raw_text,))
+    conn.commit()
+    row_id = conn.execute("SELECT last_insert_rowid()").fetchone()[0]
+    conn.close()
+    return row_id
+
+def get_pending(limit=10):
+    """获取待处理信号"""
+    conn = get_db()
+    rows = conn.execute(
+        "SELECT id, raw_text, retries FROM queue WHERE status IN ('pending','failed') AND retries < 3 ORDER BY id LIMIT ?",
+        (limit,)
+    ).fetchall()
+    conn.close()
+    return rows
+
+def mark_processing(row_id):
+    conn = get_db()
+    conn.execute("UPDATE queue SET status='processing' WHERE id=?", (row_id,))
+    conn.commit()
+    conn.close()
+
+def mark_done(row_id, result=""):
+    conn = get_db()
+    conn.execute("UPDATE queue SET status='done', processed_at=datetime('now'), result=? WHERE id=?",
+                 (result[:500], row_id))
+    conn.commit()
+    conn.close()
+
+def mark_failed(row_id, error=""):
+    conn = get_db()
+    conn.execute("UPDATE queue SET status='failed', processed_at=datetime('now'), error=?, retries=retries+1 WHERE id=?",
+                 (error[:500], row_id))
+    conn.commit()
+    conn.close()
+
+def get_stats():
+    conn = get_db()
+    stats = {}
+    for status in ['pending', 'processing', 'done', 'failed']:
+        count = conn.execute("SELECT COUNT(*) FROM queue WHERE status=?", (status,)).fetchone()[0]
+        stats[status] = count
+    conn.close()
+    return stats
+
+if __name__ == "__main__":
+    if len(sys.argv) < 2:
+        print("用法: signal_queue.py enqueue '信号原文'")
+        print("      signal_queue.py list")
+        print("      signal_queue.py retry")
+        sys.exit(1)
+
+    cmd = sys.argv[1]
+
+    if cmd == "enqueue":
+        raw = sys.argv[2] if len(sys.argv) > 2 else sys.stdin.read()
+        row_id = enqueue(raw)
+        print(f"✅ 已入队 #{row_id}")
+
+    elif cmd == "list":
+        pending = get_pending()
+        if not pending:
+            print("队列为空，无待处理信号")
+        else:
+            for row_id, raw, retries in pending:
+                print(f"#{row_id} (重试{retries}次): {raw[:80]}...")
+
+    elif cmd == "retry":
+        """重试所有失败信号"""
+        pending = get_pending()
+        print(f"待处理: {len(pending)}条")
+        for row_id, raw, retries in pending:
+            print(f"\n重试 #{row_id}...")
+            mark_processing(row_id)
+            import subprocess
+            try:
+                r = subprocess.run(
+                    ["python3", os.path.expanduser("~/.hermes/skills/trading/okx-auto-position/scripts/process_signal.py"), raw],
+                    capture_output=True, text=True, timeout=120
+                )
+                if r.returncode == 0:
+                    mark_done(row_id, r.stdout[:200])
+                    print(f"  ✅ 成功")
+                else:
+                    mark_failed(row_id, r.stderr[:200])
+                    print(f"  ❌ 失败: {r.stderr[:100]}")
+            except Exception as e:
+                mark_failed(row_id, str(e))
+                print(f"  ❌ 异常: {e}")
+
+    elif cmd == "stats":
+        stats = get_stats()
+        print(f"待处理: {stats['pending']} | 处理中: {stats['processing']} | 完成: {stats['done']} | 失败: {stats['failed']}")