add
This commit is contained in:
@@ -6,11 +6,18 @@ create.py - 动态生成PostgreSQL建表SQL语句
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
import re
|
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 $$
|
sql_template = f"""DO $$
|
||||||
BEGIN
|
BEGIN
|
||||||
IF NOT EXISTS (
|
IF NOT EXISTS (
|
||||||
|
|||||||
Reference in New Issue
Block a user