This commit is contained in:
vipg
2025-12-26 17:07:53 +08:00
parent 902ef36976
commit 5f01137a3f

View File

@@ -6,11 +6,18 @@ create.py - 动态生成PostgreSQL建表SQL语句
import os
import re
import sys
# 1. 从环境变量获取表名,如果没有设置则使用默认值
table_name = os.environ.get('TABLE_NAME', 'records')
# 从环境变量获取表名
table_name = os.environ.get('TABLE_NAME')
# 2. 定义SQL模板
if not table_name:
print("❌ 错误: 未设置 TABLE_NAME 环境变量")
sys.exit(1)
print(f"📋 接收到的表名: {table_name}")
# 定义SQL模板
sql_template = f"""DO $$
BEGIN
IF NOT EXISTS (