add
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user