add
This commit is contained in:
@@ -39,6 +39,19 @@ BEGIN
|
|||||||
END IF;
|
END IF;
|
||||||
END $$;"""
|
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():
|
def update_sql_file():
|
||||||
"""将生成的SQL语句追加到03_create_table.sql文件中"""
|
"""将生成的SQL语句追加到03_create_table.sql文件中"""
|
||||||
|
|
||||||
@@ -67,8 +80,14 @@ def update_sql_file():
|
|||||||
existing_middle = match.group(2)
|
existing_middle = match.group(2)
|
||||||
after_insert = match.group(3)
|
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:
|
with open(sql_file_path, 'w', encoding='utf-8') as f:
|
||||||
|
|||||||
@@ -31,33 +31,7 @@ BEGIN
|
|||||||
END IF;
|
END IF;
|
||||||
END $$;
|
END $$;
|
||||||
|
|
||||||
DO $$
|
|
||||||
BEGIN
|
|
||||||
IF NOT EXISTS (
|
|
||||||
SELECT 1
|
|
||||||
FROM information_schema.tables
|
|
||||||
WHERE table_schema = 'public'
|
|
||||||
AND table_name = 'cn_futures_trading_records'
|
|
||||||
) THEN
|
|
||||||
CREATE TABLE cn_futures_trading_records (
|
|
||||||
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_cn_futures_trading_records_at
|
|
||||||
BEFORE UPDATE ON cn_futures_trading_records
|
|
||||||
FOR EACH ROW
|
|
||||||
EXECUTE FUNCTION moddatetime(updated_at);
|
|
||||||
|
|
||||||
RAISE NOTICE 'cn_futures_trading_records 表已创建';
|
|
||||||
ELSE
|
|
||||||
RAISE NOTICE 'cn_futures_trading_records 表已存在,跳过';
|
|
||||||
END IF;
|
|
||||||
END $$;
|
|
||||||
|
|
||||||
DO $$
|
DO $$
|
||||||
BEGIN
|
BEGIN
|
||||||
|
|||||||
Reference in New Issue
Block a user