fix: CTP 需 zh_CN.GB18030 中文 locale 而非仅 UTF-8

Co-authored-by: Cursor <cursoragent@cursor.com>

locale_fix.py

@@ -11,29 +11,48 @@ logger = logging.getLogger(__name__)
 _LOCALE_DONE = False
 _LOCALE_NAME = ""
 
+# CTP C++ API 登录回调依赖中文 locale（见 vnpy/vnpy_ctp#24）
+_CTP_REQUIRED_LOCALES = ("zh_CN.GB18030", "zh_CN.gb18030")
+
+
+def _available_locales() -> set[str]:
+    try:
+        out = subprocess.check_output(["locale", "-a"], text=True, stderr=subprocess.DEVNULL)
+        return {line.strip() for line in out.splitlines() if line.strip()}
+    except (OSError, subprocess.SubprocessError):
+        return set()
+
+
+def missing_ctp_locales() -> list[str]:
+    """CTP 所需的 zh_CN.GB18030 是否已安装。"""
+    avail = {x.lower() for x in _available_locales()}
+    if any(x.lower() in avail for x in _CTP_REQUIRED_LOCALES):
+        return []
+    return ["zh_CN.GB18030"]
+
 
 def _list_locale_candidates() -> list[str]:
+    avail = _available_locales()
     names: list[str] = []
+    # CTP 回调优先尝试中文 locale
     for item in (
-        "C.UTF-8",
+        "zh_CN.GB18030",
+        "zh_CN.gb18030",
+        "zh_CN.UTF-8",
+        "zh_CN.utf8",
         "en_US.UTF-8",
         "en_US.utf8",
+        "C.UTF-8",
+        "C.utf8",
         "POSIX",
         "C",
     ):
-        if item not in names:
+        if item in avail and item not in names:
             names.append(item)
-    try:
-        out = subprocess.check_output(["locale", "-a"], text=True, stderr=subprocess.DEVNULL)
-        for line in out.splitlines():
-            loc = line.strip()
-            if not loc:
-                continue
-            low = loc.lower()
-            if "utf" in low and loc not in names:
-                names.insert(0, loc)
-    except (OSError, subprocess.SubprocessError):
-        pass
+    for loc in sorted(avail):
+        low = loc.lower()
+        if "utf" in low and loc not in names:
+            names.append(loc)
     return names
 
 
@@ -43,6 +62,14 @@ def ensure_process_locale() -> str:
     if _LOCALE_DONE:
         return _LOCALE_NAME
 
+    missing = missing_ctp_locales()
+    if missing:
+        raise RuntimeError(
+            "CTP 需要中文 locale zh_CN.GB18030，当前系统未安装。"
+            "请执行: sed -i '/^# zh_CN.GB18030/s/^# //' /etc/locale.gen && "
+            "locale-gen zh_CN.GB18030"
+        )
+
     last_err: locale.Error | None = None
     for name in _list_locale_candidates():
         try:
@@ -60,6 +87,5 @@ def ensure_process_locale() -> str:
 
     raise RuntimeError(
         "未找到可用 locale，vnpy_ctp 会在 CTP 登录后崩溃。"
-        "请执行: apt install -y locales && locale-gen en_US.UTF-8 && "
-        "update-locale LANG=en_US.UTF-8 LC_ALL=en_US.UTF-8"
+        "请执行: apt install -y locales && locale-gen zh_CN.GB18030 en_US.UTF-8"
     ) from last_err