迁移 psycopg3 并修复 Postgres 18 兼容性问题

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

tushare/src/storage.py

@@ -3,8 +3,8 @@ import os
 from typing import Optional
 
 import pandas as pd
-import psycopg2
-from psycopg2.extras import RealDictCursor
+import psycopg
+from psycopg.rows import dict_row
 
 from .models import ScoreResult
 
@@ -12,7 +12,7 @@ DEFAULT_DB_URL = os.environ.get("DATABASE_URL", "postgresql://trade:trade@postgr
 
 
 def _get_conn(db_url: str = DEFAULT_DB_URL):
-    return psycopg2.connect(db_url)
+    return psycopg.connect(db_url)
 
 
 def init_db(db_url: str = DEFAULT_DB_URL):
@@ -38,7 +38,7 @@ def init_db(db_url: str = DEFAULT_DB_URL):
             """)
             cur.execute("""
                 CREATE TABLE IF NOT EXISTS scores (
-                    id          UUID DEFAULT gen_random_uuid_v7() PRIMARY KEY,
+                    id          UUID DEFAULT uuidv7() PRIMARY KEY,
                     ts_code     TEXT NOT NULL,
                     trade_date  TEXT NOT NULL,
                     close       REAL,
@@ -143,7 +143,7 @@ def get_latest_score(ts_code: str, db_url: str = DEFAULT_DB_URL) -> Optional[dic
     """查询最新打分记录。"""
     conn = _get_conn(db_url)
     try:
-        with conn.cursor(cursor_factory=RealDictCursor) as cur:
+        with conn.cursor(row_factory=dict_row) as cur:
             cur.execute(
                 "SELECT * FROM scores WHERE ts_code = %s ORDER BY trade_date DESC LIMIT 1",
                 (ts_code,),