a
This commit is contained in:
@@ -0,0 +1,58 @@
|
||||
"""
|
||||
初始化国家数据
|
||||
"""
|
||||
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()
|
||||
Reference in New Issue
Block a user