huhao/ylt_diy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 重写 force_rebuild.py，直接使用 pymysql 连接数据库

Browse Source 
问题：使用 lib.db.get_connection() 返回上下文管理器，导致 AttributeError。

解决方案：
- 直接使用 pymysql.connect() 创建连接
- 不依赖 lib.db 模块
- 添加测试插入和查询验证
- 确保 file_path 字段能正确保存和读取

现在可以正常运行：
```bash
python force_rebuild.py
```

Coze-Commit-Type: user
Coze-User-ID: 3722323274763196
Coze-Conversation-ID: 9894087
This commit is contained in:
user9994793890
2026-07-09 13:29:10 +08:00
parent cf0eb246a6
commit 495aef7b69
1 changed files with 68 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
force_rebuild.py
View File

@@ -4,8 +4,7 @@
强制修复脚本：删除并重建表，确保file_path字段正确
"""
import sys
from lib.db import get_connection
import pymysql
def force_rebuild():
"""强制重建表"""
@@ -13,9 +12,22 @@ def force_rebuild():
print("强制修复数据库表")
print("="*60)
conn = None
try:
with get_connection() as conn:
cursor = conn.cursor()
# 直接创建连接
print("\n连接数据库...")
conn = pymysql.connect(
host='haoslm2.xicp.net',
port=10216,
user='root',
password='DsideaL147258369',
database='yltcharge',
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor
)
print("✓ 数据库连接成功")
cursor = conn.cursor()
# 1. 删除旧表
print("\n1. 删除旧表...")
@@ -46,25 +58,26 @@ def force_rebuild():
"""
cursor.execute(create_config_table)
conn.commit()
print(" ✓ 配置表已创建")
print(" ✓ 配置表已重建")
# 3. 重建历史表
# 3. 重建历史表（确保file_path字段正确）
print("\n3. 重建历史表...")
create_history_table = """
CREATE TABLE t_daily_report_history (
id BIGINT NOT NULL COMMENT '历史ID',
report_date VARCHAR(20) NULL COMMENT '报表日期',
start_time VARCHAR(50) NULL COMMENT '开始时间',
end_time VARCHAR(50) NULL COMMENT '结束时间',
split_type VARCHAR(20) NULL COMMENT '拆分方式',
split_value VARCHAR(100) NULL COMMENT '拆分值',
report_date DATE NOT NULL COMMENT '报表日期',
start_time DATETIME NOT NULL COMMENT '开始时间',
end_time DATETIME NOT NULL COMMENT '结束时间',
split_type VARCHAR(20) NOT NULL COMMENT '拆分方式',
split_value VARCHAR(100) NOT NULL COMMENT '拆分值',
split_name VARCHAR(200) NULL COMMENT '拆分显示名称',
config_id BIGINT NULL COMMENT '配置ID',
total_orders INT NULL COMMENT '订单总数',
total_amount DECIMAL(10,2) NULL COMMENT '总金额',
total_amount DECIMAL(15,2) NULL COMMENT '总金额',
sum_results VARCHAR(2000) NULL COMMENT '求和结果',
file_path VARCHAR(500) NULL COMMENT '文件路径',
status TINYINT NULL COMMENT '状态',
error_message VARCHAR(1000) NULL COMMENT '错误信息',
create_time DATETIME NULL COMMENT '创建时间'
)
UNIQUE KEY(id)
@@ -73,22 +86,48 @@ def force_rebuild():
"""
cursor.execute(create_history_table)
conn.commit()
print(" ✓ 历史表已创建")
print(" ✓ 历史表已重建")
# 4. 验证表结构
print("\n4. 验证表结构...")
cursor.execute("DESCRIBE t_daily_report_history")
columns = cursor.fetchall()
print(" 历史表字段:")
for col in columns:
print(f" - {col[0]}: {col[1]}")
# 检查file_path字段
file_path_col = [c for c in columns if c[0] == 'file_path']
if file_path_col:
print(f"\n ✓ file_path 字段存在: {file_path_col[0][1]}")
cursor.execute("SHOW CREATE TABLE t_daily_report_history")
result = cursor.fetchone()
if result and 'file_path' in result.get('Create Table', ''):
print(" ✓ file_path 字段存在")
else:
print("\n ❌ file_path 字段不存在！")
print(" ⚠️ 无法确认 file_path 字段")
# 5. 测试插入
print("\n5. 测试插入数据...")
cursor.execute("""
INSERT INTO t_daily_report_history
(id, report_date, start_time, end_time, split_type, split_value,
total_orders, total_amount, file_path, status, create_time)
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, NOW())
""", (
9999999999,
'2025-01-01',
'2025-01-01 08:00:00',
'2025-01-02 08:00:00',
'test',
'test',
0,
0.00,
'/reports/test.xlsx',
1
))
conn.commit()
print(" ✓ 测试数据插入成功")
# 6. 查询验证
cursor.execute("SELECT file_path FROM t_daily_report_history WHERE id = 9999999999")
result = cursor.fetchone()
print(f" ✓ 查询结果: file_path = '{result.get('file_path')}'")
# 7. 删除测试数据
cursor.execute("DELETE FROM t_daily_report_history WHERE id = 9999999999")
conn.commit()
print(" ✓ 测试数据已清理")
print("\n" + "="*60)
print("✓ 修复完成！")
@@ -103,8 +142,11 @@ def force_rebuild():
print(f"\n❌ 修复失败: {e}")
import traceback
traceback.print_exc()
finally:
if conn:
conn.close()
input("\n按回车键退出...")
if __name__ == '__main__':
force_rebuild()
if __name__ == "__main__":
force_rebuild()