From 5f01137a3f3328f1d9bd715cad292a8a801deb03 Mon Sep 17 00:00:00 2001
From: vipg <vipg@vipg.com>
Date: Fri, 26 Dec 2025 17:07:53 +0800
Subject: [PATCH] add

---
 create/create_table.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

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 (