Update 配置.py

配置.py

@@ -6,6 +6,7 @@ from datetime import datetime, timezone
 默认保存根 = "/tmp/app_data"
 os.makedirs(默认保存根, exist_ok=True)
 
+# ---------- 工具 ----------
 def _guess_mapping(headers):
     m = {"时间": None, "开": None, "高": None, "低": None, "收": None, "量": None}
     cand = {
@@ -31,13 +32,11 @@ def _parse_time_to_ms(s):
     s = str(s).strip()
     if not s:
         return None
-    # 数字：秒或毫秒
     if re.fullmatch(r"\d{10,13}", s):
         v = int(s)
-        if v < 10**12:  # 秒
+        if v < 10**12:
             return v * 1000
         return v
-    # ISO 格式
     try:
         ss = s.replace("Z", "+00:00")
         dt = datetime.fromisoformat(ss)
@@ -46,14 +45,7 @@ def _parse_time_to_ms(s):
         return int(dt.timestamp() * 1000)
     except Exception:
         pass
-    # 常见格式补刀
-    fmts = [
-        "%Y-%m-%d %H:%M:%S",
-        "%Y/%m/%d %H:%M:%S",
-        "%Y-%m-%d %H:%M",
-        "%Y/%m/%d %H:%M",
-        "%Y-%m-%dT%H:%M:%S",
-    ]
+    fmts = ["%Y-%m-%d %H:%M:%S","%Y/%m/%d %H:%M:%S","%Y-%m-%d %H:%M","%Y/%m/%d %H:%M","%Y-%m-%dT%H:%M:%S"]
     for f in fmts:
         try:
             dt = datetime.strptime(s, f).replace(tzinfo=timezone.utc)
@@ -68,7 +60,6 @@ def _sniff_delimiter(path):
     try:
         return csv.Sniffer().sniff(sample).delimiter
     except Exception:
-        # 兜底
         return ',' if sample.count(',') >= sample.count('\t') else '\t'
 
 def 读取CSV预览(file_obj):
@@ -87,7 +78,7 @@ def 读取CSV预览(file_obj):
                 if i < 5:
                     head_rows.append(row)
                 total += 1
-        info = f"分隔符: '{delim}' | 列: {len(headers)} | 总行(不含表头): ~{total}"
+        info = f"分隔符: '{delim}' | 列数: {len(headers)} | 预估行数(不含表头): ~{total}"
         preview = "\n".join([",".join(row) for row in head_rows])
         return info, ",".join(headers), preview
     except Exception as e:
@@ -103,12 +94,10 @@ def 标准化CSV(file_obj, 输出文件名, 时间列, 开列, 高列, 低列,
     os.makedirs(save_dir, exist_ok=True)
     out_path = os.path.join(save_dir, 输出文件名)
 
-    # 校验必要列
     必要 = [时间列, 开列, 高列, 低列, 收列]
     if not all(必要):
-        return f"请确保时间、开、高、低、收列都已选择", ""
-    
-    # 读取、转换并写出标准格式
+        return "请确保时间/开/高/低/收列均已选择", ""
+
     delim = _sniff_delimiter(path)
     写入 = 0
     try:
@@ -116,14 +105,11 @@ def 标准化CSV(file_obj, 输出文件名, 时间列, 开列, 高列, 低列,
              open(out_path, "w", encoding="utf-8", newline="") as w:
             r = csv.DictReader(f, delimiter=delim)
             wr = csv.writer(w)
-            wr.writerow(["时间","开","高","低","收","量"])  # 标准表头
+            wr.writerow(["时间","开","高","低","收","量"])
             for row in r:
                 try:
                     tms = _parse_time_to_ms(row[时间列])
-                    o = float(row[开列])
-                    h = float(row[高列])
-                    l = float(row[低列])
-                    c = float(row[收列])
+                    o = float(row[开列]); h = float(row[高列]); l = float(row[低列]); c = float(row[收列])
                     v = float(row[量列]) if 量列 and 量列 in row and row[量列] not in (None, "") else 0.0
                     if tms is None:
                         continue
@@ -135,6 +121,56 @@ def 标准化CSV(file_obj, 输出文件名, 时间列, 开列, 高列, 低列,
     except Exception as e:
         return f"保存失败: {e}", ""
 
+# ---------- 合约登记 ----------
+合同根 = os.path.join(默认保存根, "contracts")
+os.makedirs(合同根, exist_ok=True)
+
+def _key(exchange, inst): return f"{exchange}__{inst}"
+
+def 登记合约(交易所, 合约ID, 别名, csv路径):
+    if not 交易所 or not 合约ID or not csv路径:
+        return "登记失败：缺少交易所/合约ID/CSV路径"
+    meta = {
+        "exchange": 交易所,
+        "contract_id": 合约ID,
+        "alias": 别名 or "",
+        "csv_path": csv路径,
+        "saved_at": datetime.utcnow().isoformat() + "Z"
+    }
+    path = os.path.join(合同根, _key(交易所, 合约ID) + ".json")
+    with open(path, "w", encoding="utf-8") as f:
+        json.dump(meta, f, ensure_ascii=False, indent=2)
+    return f"已登记合约：{交易所} | {合约ID}", path
+
+def 列出已登记合约():
+    items = []
+    for fn in os.listdir(合同根):
+        if fn.endswith(".json"):
+            try:
+                with open(os.path.join(合同根, fn), "r", encoding="utf-8") as f:
+                    m = json.load(f)
+                label = f"{m.get('exchange','?')} | {m.get('contract_id','?')}" + (f" ({m.get('alias')})" if m.get('alias') else "")
+                items.append((label, _key(m.get('exchange','?'), m.get('contract_id','?'))))
+            except Exception:
+                continue
+    # 返回 choices 列表（仅值）
+    return [v for _, v in sorted(items)]
+
+def 读取登记(key):
+    path = os.path.join(合同根, key + ".json")
+    if not os.path.exists(path):
+        return None
+    with open(path, "r", encoding="utf-8") as f:
+        return json.load(f)
+
+def 删除登记(key):
+    path = os.path.join(合同根, key + ".json")
+    if os.path.exists(path):
+        os.remove(path)
+        return True
+    return False
+
+# ---------- 参数 ----------
 def 加载参数文件(param_file):
     if param_file is None:
         return json.dumps({"ver":"v1"}, ensure_ascii=False, indent=2)
@@ -169,11 +205,8 @@ def 保存参数(JSON文本, 文件名):
     except Exception as e:
         return f"保存失败: {e}"
 
+# ---------- 构建设置面板 ----------
 def 构建设置面板(容器, 默认参数=None, 记录函数=None):
-    """
-    在给定容器内构建设置面板。
-    返回一个字典，包含用于回填到运行页的组件（参数编辑框、复制按钮）。
-    """
     if 默认参数 is None:
         默认参数 = {"ver":"v1"}
 
@@ -182,12 +215,8 @@ def 构建设置面板(容器, 默认参数=None, 记录函数=None):
             with gr.TabItem("参数配置"):
                 with gr.Row():
                     with gr.Column(scale=1):
-                        gr.Markdown("### 参数编辑")
-                        参数编辑框 = gr.Textbox(
-                            label="策略参数（JSON可编辑）",
-                            value=json.dumps(默认参数, ensure_ascii=False, indent=2),
-                            lines=18
-                        )
+                        gr.Markdown("### 策略参数")
+                        参数编辑框 = gr.Textbox(label="策略参数（JSON可编辑）", value=json.dumps(默认参数, ensure_ascii=False, indent=2), lines=18)
                         with gr.Row():
                             参数文件 = gr.File(label="上传参数JSON文件", file_types=[".json"])
                             加载按钮 = gr.Button("加载到编辑框")
@@ -197,13 +226,12 @@ def 构建设置面板(容器, 默认参数=None, 记录函数=None):
                             参数保存名 = gr.Textbox(label="保存文件名", value="params.json")
                             参数保存按钮 = gr.Button("保存参数到磁盘")
                         保存结果 = gr.Textbox(label="保存结果", interactive=False)
-                        复制按钮 = gr.Button("将上方参数回填到运行页", variant="primary")
+                        复制参数按钮 = gr.Button("将上方参数回填到运行页", variant="primary")
 
                     with gr.Column(scale=1):
                         gr.Markdown("小提示")
                         gr.Markdown("- 可上传回测生成的最优参数JSON\n- 编辑后点“格式化/校验”再保存\n- 点击“将上方参数回填到运行页”即可带回主界面")
 
-                # 事件
                 加载按钮.click(加载参数文件, inputs=参数文件, outputs=参数编辑框)
                 格式化按钮.click(格式化参数, inputs=参数编辑框, outputs=[参数编辑框, 加载结果])
                 参数保存按钮.click(保存参数, inputs=[参数编辑框, 参数保存名], outputs=保存结果)
@@ -211,74 +239,107 @@ def 构建设置面板(容器, 默认参数=None, 记录函数=None):
             with gr.TabItem("CSV上传"):
                 with gr.Row():
                     with gr.Column(scale=1):
+                        gr.Markdown("#### 1) 上传与预览")
                         csv文件 = gr.File(label="上传 1m K线 CSV", file_types=[".csv"])
-                        
-                        # 预览区域
                         预览信息 = gr.Textbox(label="文件信息", lines=2, interactive=False)
                         表头显示 = gr.Textbox(label="检测到的表头", lines=2, interactive=False)
                         预览前几行 = gr.Textbox(label="前几行示例", lines=8, interactive=False)
-                        
-                        # 映射下拉菜单（静态创建）
-                        gr.Markdown("### 列映射")
+
+                        gr.Markdown("#### 2) 列映射")
                         时间列 = gr.Dropdown(label="时间列", choices=[], value=None)
                         开列 = gr.Dropdown(label="开盘价列", choices=[], value=None)
                         高列 = gr.Dropdown(label="最高价列", choices=[], value=None)
                         低列 = gr.Dropdown(label="最低价列", choices=[], value=None)
                         收列 = gr.Dropdown(label="收盘价列", choices=[], value=None)
                         量列 = gr.Dropdown(label="成交量列（可选）", choices=[], value=None)
-                        
-                        输出文件名 = gr.Textbox(label="输出文件名", value="uploaded_1m.csv")
-                        保存按钮 = gr.Button("标准化并保存")
+
+                        gr.Markdown("#### 3) 合约登记")
+                        交易所选择 = gr.Dropdown(label="交易所", choices=["OKX","BINANCE","BYBIT","DERIBIT","OTHER"], value="OKX")
+                        合约ID框 = gr.Textbox(label="合约ID（例：UXLINK-USDT-SWAP）", value="UXLINK-USDT-SWAP")
+                        合约别名 = gr.Textbox(label="合约别名（可选）", value="")
+                        输出文件名 = gr.Textbox(label="标准化输出文件名", value="uploaded_1m.csv")
+                        保存按钮 = gr.Button("标准化并登记", variant="primary")
                         保存结果 = gr.Textbox(label="结果", interactive=False)
                         输出路径 = gr.Textbox(label="输出路径", interactive=False)
-                        
+
+                        gr.Markdown("#### 4) 已登记合约")
+                        已登记下拉 = gr.Dropdown(label="已登记合约", choices=列出已登记合约(), value=None)
+                        with gr.Row():
+                            刷新登记按钮 = gr.Button("刷新列表")
+                            加载登记按钮 = gr.Button("加载登记到上方")
+                            删除登记按钮 = gr.Button("删除所选登记")
+                            回填合约按钮 = gr.Button("回填合约到运行页", variant="secondary")
+
                     with gr.Column(scale=1):
-                        gr.Markdown("### 使用说明")
-                        gr.Markdown("1. 上传 CSV 文件\n2. 系统自动检测表头\n3. 选择对应的列映射\n4. 点击保存生成标准格式")
-                        gr.Markdown("标准格式: 时间(ms), 开, 高, 低, 收, 量")
+                        gr.Markdown("使用步骤")
+                        gr.Markdown("1. 上传 CSV → 预览及列映射\n2. 选择交易所/合约ID → 标准化并登记\n3. 可在右侧主界面启动模拟\n4. 如需再次使用，直接在“已登记合约”选择并加载")
 
-                # CSV 上传后的处理函数
+                # 上传后更新映射下拉
                 def 处理CSV上传(file_obj):
-                    if file_obj is None:
-                        return ("", "", "", [], None, None, None, None, None, None)
-                    
                     info, headers_str, preview = 读取CSV预览(file_obj)
                     if not headers_str:
-                        return (info, "", preview, [], None, None, None, None, None, None)
-                    
+                        empty = gr.update(choices=[], value=None)
+                        return info, headers_str, preview, empty, empty, empty, empty, empty, empty
                     headers = headers_str.split(",")
                     映射 = _guess_mapping(headers)
-                    
                     return (
-                        info, 
-                        headers_str, 
-                        preview, 
-                        headers,
-                        映射.get("时间"), 
-                        映射.get("开"), 
-                        映射.get("高"), 
-                        映射.get("低"), 
-                        映射.get("收"), 
-                        映射.get("量")
+                        info, headers_str, preview,
+                        gr.update(choices=headers, value=映射.get("时间")),
+                        gr.update(choices=headers, value=映射.get("开")),
+                        gr.update(choices=headers, value=映射.get("高")),
+                        gr.update(choices=headers, value=映射.get("低")),
+                        gr.update(choices=headers, value=映射.get("收")),
+                        gr.update(choices=headers, value=映射.get("量")),
                     )
 
-                # 更新下拉菜单选项和默认值
                 csv文件.change(
                     fn=处理CSV上传,
                     inputs=csv文件,
-                    outputs=[
-                        预览信息, 表头显示, 预览前几行,
-                        时间列, 时间列, 开列, 高列, 低列, 收列, 量列
-                    ]
+                    outputs=[预览信息, 表头显示, 预览前几行, 时间列, 开列, 高列, 低列, 收列, 量列]
                 )
-                
-                # 保存按钮点击事件
+
+                # 保存并登记
+                def 保存并登记(file_obj, 输出名, 时间c, 开c, 高c, 低c, 收c, 量c, 交易所, 合约ID, 别名):
+                    msg, path = 标准化CSV(file_obj, 输出名, 时间c, 开c, 高c, 低c, 收c, 量c)
+                    if not path:
+                        return msg, "", gr.update(choices=列出已登记合约()), ""
+                    jmsg = ""
+                    try:
+                        jmsg, _ = 登记合约(交易所, 合约ID, 别名, path)
+                    except Exception as e:
+                        jmsg = f"登记失败：{e}"
+                    # 更新列表并选中新登记
+                    key = _key(交易所, 合约ID)
+                    return f"{msg}\n{jmsg}", path, gr.update(choices=列出已登记合约(), value=key), key
+
+                新登记键 = gr.Textbox(visible=False)
                 保存按钮.click(
-                    fn=标准化CSV,
-                    inputs=[csv文件, 输出文件名, 时间列, 开列, 高列, 低列, 收列, 量列],
-                    outputs=[保存结果, 输出路径]
+                    保存并登记,
+                    inputs=[csv文件, 输出文件名, 时间列, 开列, 高列, 低列, 收列, 量列, 交易所选择, 合约ID框, 合约别名],
+                    outputs=[保存结果, 输出路径, 已登记下拉, 新登记键]
                 )
 
+                # 刷新/加载/删除
+                刷新登记按钮.click(lambda: gr.update(choices=列出已登记合约(), value=None), outputs=已登记下拉)
+
+                def do_load(key):
+                    m = 读取登记(key) if key else None
+                    if not m:
+                        return (gr.update(), gr.update(), gr.update(), "")
+                    return (gr.update(value=m.get("exchange","OKX")),
+                            gr.update(value=m.get("contract_id","")),
+                            gr.update(value=m.get("alias","")),
+                            m.get("csv_path",""))
+                加载登记按钮.click(do_load, inputs=已登记下拉, outputs=[交易所选择, 合约ID框, 合约别名, 输出路径])
+
+                def do_del(key):
+                    if not key:
+                        return "请先选择一条登记", gr.update(choices=列出已登记合约(), value=None)
+                    ok = 删除登记(key)
+                    msg = "已删除" if ok else "删除失败（不存在或权限）"
+                    return msg, gr.update(choices=列出已登记合约(), value=None)
+                删除登记按钮.click(do_del, inputs=已登记下拉, outputs=[保存结果, 已登记下拉])
+
             with gr.TabItem("系统"):
                 修复按钮 = gr.Button("修复导入路径/别名（重试）")
                 修复结果 = gr.Textbox(label="结果", interactive=False)
@@ -302,5 +363,10 @@ def 构建设置面板(容器, 默认参数=None, 记录函数=None):
                         return f"失败：{e}"
                 修复按钮.click(do_fix, outputs=修复结果)
 
-    # 返回可用于回填运行页的关键组件
-    return {"参数编辑框": 参数编辑框, "复制按钮": 复制按钮}
+    # 返回用于与主界面联动的组件
+    return {
+        "参数编辑框": 参数编辑框,
+        "复制参数按钮": 复制参数按钮,
+        "合约ID框": 合约ID框,
+        "回填合约按钮": 回填合约按钮
+    }