diff --git a/create/create_table.py b/create/create_table.py index 3b2c5f1..9a7014c 100644 --- a/create/create_table.py +++ b/create/create_table.py @@ -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 (