import os
import pymysql
import random

# 默认指向 SSH 隧道（tools/dev-tunnel.ps1），不再使用已移除的 ZeroTier/nginx 转发
conn = pymysql.connect(
    host=os.environ.get('PMS_MYSQL_HOST', '127.0.0.1'),
    port=int(os.environ.get('PMS_MYSQL_PORT', '13306')),
    user='root', password='Hbhyg731024@', database='pms',
)
cur = conn.cursor()

# 获取所有项目ID
cur.execute('SELECT id FROM projects')
proj_ids = [row[0] for row in cur.fetchall()]

# 为每个项目添加2-3个柜子
count = 0
for proj_id in proj_ids:
    num = random.randint(2, 3)
    for j in range(num):
        count += 1
        abstract_id = f'10-{count:08d}'
        imei = f'86{random.randint(10000000000000, 99999999999999)}'
        auth = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=8))
        name = f'CAB-{count:03d}'
        try:
            cur.execute(f"INSERT IGNORE INTO cabinets (project_id, abstract_id, imei, auth_str, name, status) VALUES ({proj_id}, '{abstract_id}', '{imei}', '{auth}', '{name}', 1)")
        except:
            pass

conn.commit()
cur.execute('SELECT COUNT(*) FROM cabinets')
print(f'柜子总数: {cur.fetchone()[0]}')
conn.close()