add

2025-12-26 16:08:07 +08:00
parent 206cf30904
commit e51805749a
4 changed files with 237 additions and 1 deletions
--- a/create/create.py
+++ b/create/create.py
@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+create.py - 动态生成PostgreSQL建表SQL语句
+"""
+
+import os
+import re
+
+# 1. 定义表名称变量，方便调整
+table_name = "cn_futures_trading_records"  # 可以根据需要修改表名
+
+# 2. 定义SQL模板
+sql_template = f"""DO $$
+BEGIN
+    IF NOT EXISTS (
+        SELECT 1
+        FROM information_schema.tables
+        WHERE table_schema = 'public'
+          AND table_name   = '{table_name}'
+    ) THEN
+        CREATE TABLE {table_name} (
+            id            UUID         DEFAULT gen_random_uuid() PRIMARY KEY,  -- id
+            payload       JSONB        NOT NULL,                               -- 数据
+            deleted       BOOLEAN      NOT NULL DEFAULT FALSE,                 -- 删除状态
+            created_at    TIMESTAMPTZ  NOT NULL DEFAULT CURRENT_TIMESTAMP,     -- 记录创建时间
+            updated_at    TIMESTAMPTZ  NOT NULL DEFAULT CURRENT_TIMESTAMP      -- 记录修改时间
+        );
+
+        -- 3 触发器：自动刷新 updated_at
+        CREATE TRIGGER trg_{table_name}_at
+            BEFORE UPDATE ON {table_name}
+            FOR EACH ROW
+            EXECUTE FUNCTION moddatetime(updated_at);
+
+        RAISE NOTICE '{table_name} 表已创建';
+    ELSE
+        RAISE NOTICE '{table_name} 表已存在，跳过';
+    END IF;
+END $$;"""
+
+def update_sql_file():
+    """将生成的SQL语句追加到03_create_table.sql文件中"""
+    
+    # 定义文件路径
+    sql_file_path = os.path.join(
+        os.path.dirname(os.path.dirname(__file__)), 
+        'infra', 
+        'postgres', 
+        'sql', 
+        '03_create_table.sql'
+    )
+    
+    try:
+        # 读取现有文件内容
+        with open(sql_file_path, 'r', encoding='utf-8') as f:
+            content = f.read()
+        
+        # 查找插入位置（在\"部署完成\"日志前）
+        insert_pattern = r'(DO \$\$.*?RAISE NOTICE \'🚀============ 数据库表部署开始 ============🚀\'.*?END \$\$;)(.*?)(DO \$\$.*?RAISE NOTICE \'============ 数据库表部署完成 ============\'.*?END \$\$;)'
+        
+        match = re.search(insert_pattern, content, re.DOTALL)
+        
+        if match:
+            # 分割内容
+            before_insert = match.group(1)
+            existing_middle = match.group(2)
+            after_insert = match.group(3)
+            
+            # 组合新内容
+            new_content = f"""{before_insert}\n\n{sql_template}\n{existing_middle}\n{after_insert}"""
+            
+            # 写回文件
+            with open(sql_file_path, 'w', encoding='utf-8') as f:
+                f.write(new_content)
+            
+            print(f"✅ 成功更新 {sql_file_path}")
+            print(f"📋 生成的表名: {table_name}")
+            print(f"📝 SQL内容已追加到文件中")
+            
+        else:
+            print("❌ 无法找到插入位置，请检查文件格式")
+            
+    except FileNotFoundError:
+        print(f"❌ 文件 {sql_file_path} 不存在")
+    except Exception as e:
+        print(f"❌ 处理文件时出错: {e}")
+
+if __name__ == "__main__":
+    print(f"🚀 开始生成表 '{table_name}' 的SQL语句...")
+    update_sql_file()