This commit is contained in:
vipg
2025-12-26 16:15:43 +08:00
parent 9abd667211
commit 4118483ba2
2 changed files with 20 additions and 27 deletions

View File

@@ -39,6 +39,19 @@ BEGIN
END IF;
END $$;"""
def normalize_blank_lines(text):
"""规范化空行确保END $$;和DO $$之间只有一个空行"""
# 将多个空行替换为单个空行
text = re.sub(r'\n{3,}', '\n\n', text)
# 确保END $$;后面有一个空行再接DO $$
text = re.sub(r'END \$\$;(\s*)DO \$\$', r'END $$;\n\nDO $$', text)
# 清理开头和结尾的多余空行
text = text.strip() + '\n'
return text
def update_sql_file():
"""将生成的SQL语句追加到03_create_table.sql文件中"""
@@ -67,8 +80,14 @@ def update_sql_file():
existing_middle = match.group(2)
after_insert = match.group(3)
# 规范化空行
existing_middle = normalize_blank_lines(existing_middle)
# 组合新内容
new_content = f"""{before_insert}\n\n{sql_template}\n{existing_middle}\n{after_insert}"""
new_content = f"""{before_insert}\n\n{sql_template}\n\n{existing_middle}\n{after_insert}"""
# 规范化整个内容的空行
new_content = normalize_blank_lines(new_content)
# 写回文件
with open(sql_file_path, 'w', encoding='utf-8') as f: