"""
初始化国家数据
"""
import json
from database import SessionLocal, engine, Base
from models import Country


def init_countries():
    # 读取国家数据
    with open('/home/ubuntu/nanocode/fumamxCRM/countryinfo.json', 'r', encoding='utf-8') as f:
        data = json.load(f)
    
    countries_data = data.get('data', [])
    
    db = SessionLocal()
    try:
        # 清空现有数据
        db.query(Country).delete()
        
        # 插入新数据
        for item in countries_data:
            country = Country(
                country_id=item.get('countryid'),
                continent_id=item.get('continentid'),
                sort_order=item.get('sortorder', 0),
                cn_name=item.get('cnname'),
                en_name=item.get('enname'),
                en_name_ali=item.get('ennameAli'),
                short_name=item.get('shortname'),
                short_name_ali=item.get('shortnameAli'),
                pinyin=item.get('pinyin'),
                del_state=item.get('delstate', 0)
            )
            db.add(country)
        
        db.commit()
        print(f"✅ 成功导入 {len(countries_data)} 个国家/地区")
        
        # 显示前10个
        print("\n前10个国家示例:")
        for c in db.query(Country).limit(10).all():
            print(f"  {c.cn_name} ({c.short_name}) - {c.en_name}")
            
    except Exception as e:
        db.rollback()
        print(f"❌ 导入失败: {e}")
    finally:
        db.close()


if __name__ == "__main__":
    # 创建表
    Base.metadata.create_all(bind=engine)
    print("📦 国家表创建完成")
    
    # 导入数据
    init_countries()