100 lines
3.7 KiB
Python
100 lines
3.7 KiB
Python
|
"""临时联调数据：为能耗报表验证准备 pms_dev 数据，并打印清理所需的 ID。
|
|||
|
|
|
|||
|
|
产出：
|
|||
|
|
- 1 个临时用户（role=2，总管理员）→ 用于自签 JWT
|
|||
|
|
- 1 个临时项目（org=1）+ 2 个临时柜子（含板/仓体）→ 用于验证项目/柜分组
|
|||
|
|
- 6 条 charge_records（3 柜 × 2 天）→ 用于验证分组与趋势
|
|||
|
|
|
|||
|
|
所有 ID 写入 temp_ids.json，供 cleanup_temp_data.py 精确删除。
|
|||
|
|
"""
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import pymysql
|
|||
|
|
|
|||
|
|
HERE = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
DB = dict(host="10.8.0.252", user="root", password="Hbhyg731024@", database="pms_dev")
|
|||
|
|
|
|||
|
|
conn = pymysql.connect(**DB)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
|
|||
|
|
created = {
|
|||
|
|
"user_id": None,
|
|||
|
|
"project_id": None,
|
|||
|
|
"cabinet_ids": [],
|
|||
|
|
"board_ids": [],
|
|||
|
|
"compartment_ids": [],
|
|||
|
|
"record_ids": [],
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# ---------------------------------------------------------------- 临时总管理员
|
|||
|
|
cur.execute(
|
|||
|
|
"INSERT INTO users (phone, name, password, role, organization_id, status) "
|
|||
|
|
"VALUES (%s, %s, %s, %s, %s, %s)",
|
|||
|
|
("19900000001", "能耗验证临时管理员", "x", 2, None, 1),
|
|||
|
|
)
|
|||
|
|
created["user_id"] = cur.lastrowid
|
|||
|
|
|
|||
|
|
# ------------------------------------------------- 临时项目（org=1，验证项目分组）
|
|||
|
|
cur.execute(
|
|||
|
|
"INSERT INTO projects (organization_id, name) VALUES (%s, %s)",
|
|||
|
|
(1, "能耗验证临时项目"),
|
|||
|
|
)
|
|||
|
|
temp_project_id = cur.lastrowid
|
|||
|
|
created["project_id"] = temp_project_id
|
|||
|
|
|
|||
|
|
# ------------------------------------- 临时柜子 ×2（挂临时项目；imei 15 位、abstract_id 唯一）
|
|||
|
|
for idx, name in enumerate(["能耗验证临时柜A", "能耗验证临时柜B"], start=1):
|
|||
|
|
cur.execute(
|
|||
|
|
"INSERT INTO cabinets (project_id, abstract_id, imei, name, status) "
|
|||
|
|
"VALUES (%s, %s, %s, %s, %s)",
|
|||
|
|
(temp_project_id, f"99-0000000{idx}", f"99900000000000{idx}", name, 1),
|
|||
|
|
)
|
|||
|
|
cabinet_id = cur.lastrowid
|
|||
|
|
created["cabinet_ids"].append(cabinet_id)
|
|||
|
|
cur.execute(
|
|||
|
|
"INSERT INTO cabin_boards (cabinet_id, board_id, status) VALUES (%s, %s, %s)",
|
|||
|
|
(cabinet_id, 1, 1),
|
|||
|
|
)
|
|||
|
|
board_id = cur.lastrowid
|
|||
|
|
created["board_ids"].append(board_id)
|
|||
|
|
cur.execute(
|
|||
|
|
"INSERT INTO compartments (cabin_board_id, channel_id, status) VALUES (%s, %s, %s)",
|
|||
|
|
(board_id, 1, 0),
|
|||
|
|
)
|
|||
|
|
created["compartment_ids"].append(cur.lastrowid)
|
|||
|
|
|
|||
|
|
comp_a, comp_b = created["compartment_ids"]
|
|||
|
|
cab_a, cab_b = created["cabinet_ids"]
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------- 6 条充电记录
|
|||
|
|
# 3 个柜子（已有柜 1 + 临时柜 A/B）× 2 天（今天 / 昨天）：
|
|||
|
|
# 柜1 今天 220 Wh 昨天 1000 Wh （项目 1「联调测试项目」）
|
|||
|
|
# 柜A 今天 330 Wh 昨天 2000 Wh （临时项目）
|
|||
|
|
# 柜B 今天 440 Wh 昨天 3000 Wh （临时项目）
|
|||
|
|
# 期望：今日合计 990 Wh；本月合计 990 + 6000 = 6990 Wh
|
|||
|
|
plan = [
|
|||
|
|
(1, 1, 220, "CURDATE()"),
|
|||
|
|
(1, 1, 1000, "DATE_SUB(CURDATE(), INTERVAL 1 DAY)"),
|
|||
|
|
(cab_a, comp_a, 330, "CURDATE()"),
|
|||
|
|
(cab_a, comp_a, 2000, "DATE_SUB(CURDATE(), INTERVAL 1 DAY)"),
|
|||
|
|
(cab_b, comp_b, 440, "CURDATE()"),
|
|||
|
|
(cab_b, comp_b, 3000, "DATE_SUB(CURDATE(), INTERVAL 1 DAY)"),
|
|||
|
|
]
|
|||
|
|
for cabinet_id, compartment_id, energy, day_expr in plan:
|
|||
|
|
cur.execute(
|
|||
|
|
"INSERT INTO charge_records "
|
|||
|
|
"(compartment_id, cabinet_id, start_time, end_time, energy, max_temperature, status) "
|
|||
|
|
f"VALUES (%s, %s, {day_expr} + INTERVAL 10 HOUR, {day_expr} + INTERVAL 12 HOUR, %s, %s, %s)",
|
|||
|
|
(compartment_id, cabinet_id, energy, 29815, 0),
|
|||
|
|
)
|
|||
|
|
created["record_ids"].append(cur.lastrowid)
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
|
|||
|
|
with open(os.path.join(HERE, "temp_ids.json"), "w", encoding="utf-8") as fh:
|
|||
|
|
json.dump(created, fh, ensure_ascii=False, indent=2)
|
|||
|
|
|
|||
|
|
print(json.dumps(created, ensure_ascii=False, indent=2))
|
|||
|
|
cur.close()
|
|||
|
|
conn.close()
|