import os

import pymysql

# 默认指向 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()

# 添加parent_id字段
try:
    cur.execute("ALTER TABLE organizations ADD COLUMN parent_id BIGINT DEFAULT NULL COMMENT '上级组织ID' AFTER name")
    print("✓ parent_id字段添加成功")
except Exception as e:
    print(f"✓ {e}")

# 添加外键
try:
    cur.execute("ALTER TABLE organizations ADD FOREIGN KEY (parent_id) REFERENCES organizations(id)")
    print("✓ 外键添加成功")
except Exception as e:
    print(f"✓ {e}")

# 创建一些下级机构
sub_orgs = [
    (1, "无锡太湖新城分部"),
    (1, "无锡梁溪区分部"),
    (2, "苏州工业园区分部"),
    (3, "杭州西湖区分部"),
    (3, "杭州滨江区分部"),
]

for parent_id, name in sub_orgs:
    try:
        cur.execute(f"INSERT INTO organizations (name, parent_id) VALUES ('{name}', {parent_id})")
    except:
        pass

conn.commit()
print("✓ 下级机构创建完成")

# 统计
cur.execute("SELECT COUNT(*) FROM organizations")
print(f"组织总数: {cur.fetchone()[0]}")
cur.execute("SELECT COUNT(*) FROM organizations WHERE parent_id IS NOT NULL")
print(f"有上级的组织: {cur.fetchone()[0]}")

conn.close()