chore(db): 比较于niushop的差异

docs/db/update_sql_comments.py

@@ -0,0 +1,117 @@
+import re
+import os
+
+# 读取 SQL 文件内容
+def read_sql_file(file_path):
+    with open(file_path, 'r', encoding='utf-8') as f:
+        return f.read()
+
+# 解析 database.sql，提取表结构和注释
+def parse_database_sql(sql_content):
+    tables = {}
+    
+    # 匹配 CREATE TABLE 语句，更灵活的格式
+    # 匹配所有 CREATE TABLE 语句，不依赖于 ENGINE、CHARACTER SET 等子句的顺序
+    table_pattern = re.compile(r"CREATE TABLE\s+`?([^`\s]+)`?\s*\(([^;]+)\)\s*(?:[^;]+COMMENT\s*=\s*'([^']+)'[^;]*)?\s*;", re.DOTALL | re.IGNORECASE)
+    
+    matches = table_pattern.findall(sql_content)
+    
+    for table_name, table_def, table_comment in matches:
+        if not table_comment:
+            # 如果没有匹配到表注释，尝试从其他位置获取
+            comment_match = re.search(r"COMMENT\s*=\s*'([^']+)'", table_def, re.IGNORECASE)
+            if comment_match:
+                table_comment = comment_match.group(1)
+            else:
+                table_comment = ''
+        
+        # 解析列定义和注释
+        columns = {}
+        # 匹配列定义，包括 COMMENT
+        column_lines = table_def.split('\n')
+        for line in column_lines:
+            # 匹配列名、类型和注释
+            column_match = re.search(r"\s*([^\s,]+)\s+([^\s,]+)\s*(?:[^,]+COMMENT\s*=\s*'([^']+)'[^,]*|[^,]*)", line)
+            if column_match:
+                column_name = column_match.group(1)
+                column_comment = column_match.group(3) or ''
+                if column_comment:
+                    columns[column_name] = column_comment
+        
+        tables[table_name] = {
+            'comment': table_comment,
+            'columns': columns
+        }
+    
+    return tables
+
+# 更新 init_v2.0.sql 文件中的注释
+def update_init_sql(init_sql_path, database_tables):
+    # 读取 init_v2.0.sql 内容
+    init_content = read_sql_file(init_sql_path)
+    
+    # 匹配 CREATE TABLE 语句，适应 init_v2.0.sql 的格式
+    table_pattern = re.compile(r"(create table if not exists lucky_([^\s]+)\s*\(([^;]+)\)\s*comment\s*=\s*'[^']*'\s*(.*?);)", re.DOTALL | re.IGNORECASE)
+    
+    def replace_table(match):
+        full_match = match.group(0)
+        table_name = match.group(2)
+        table_def = match.group(3)
+        table_suffix = match.group(4)
+        
+        if table_name in database_tables:
+            # 获取数据库表的注释和列注释
+            db_table = database_tables[table_name]
+            table_comment = db_table['comment']
+            columns = db_table['columns']
+            
+            # 更新列注释
+            new_table_def = table_def
+            for column_name, column_comment in columns.items():
+                # 匹配列定义，替换注释
+                # 格式：列名 类型 default 默认值 not null comment '注释'
+                column_pattern = re.compile(r"(\s*" + column_name + r"\s+[^\s,]+\s*(?:default\s+[^\s,]+\s*)?(?:not null\s*)?comment\s*=\s*')([^']*)'([^,]*)", re.IGNORECASE)
+                new_table_def = column_pattern.sub(r"\1" + column_comment + r"'\3", new_table_def)
+            
+            # 重新构建 CREATE TABLE 语句
+            new_full_match = f"create table if not exists lucky_{table_name} ({new_table_def}) comment = '{table_comment}' {table_suffix};"
+            return new_full_match
+        
+        return full_match
+    
+    # 替换所有表
+    updated_content = table_pattern.sub(replace_table, init_content)
+    
+    # 写回文件
+    with open(init_sql_path, 'w', encoding='utf-8') as f:
+        f.write(updated_content)
+    
+    print(f"Updated {init_sql_path}")
+
+# 主函数
+def main():
+    # 文件路径
+    database_sql_path = r'./niushop_database.sql'
+    init_v20_sql_path = r'./init_v2.0.sql'
+    init_v20_with_data_sql_path = r'./init_v2.0_with_data.sql'
+    
+    # 解析 database.sql
+    print("Parsing database.sql...")
+    database_content = read_sql_file(database_sql_path)
+    database_tables = parse_database_sql(database_content)
+    print(f"Found {len(database_tables)} tables in database.sql")
+    
+    # 更新 init_v2.0.sql
+    if os.path.exists(init_v20_sql_path):
+        print("Updating init_v2.0.sql...")
+        update_init_sql(init_v20_sql_path, database_tables)
+    
+    # 更新 init_v2.0_with_data.sql
+    if os.path.exists(init_v20_with_data_sql_path):
+        print("Updating init_v2.0_with_data.sql...")
+        update_init_sql(init_v20_with_data_sql_path, database_tables)
+    
+    print("All files updated successfully!")
+
+if __name__ == "__main__":
+    main()