init_run_data.py 923 B

123456789101112131415161718192021222324
  1. from app.models.base_model import SessionLocal
  2. from app.service.v2.initialize_data import dialog_menu_sync, default_group_sync, default_role_sync, \
  3. basic_agent_sync, admin_account_sync, sync_rg_api_token, sync_complex_api_token, system_license_sync
  4. from app.task.sync_account_token import sync_token
  5. async def sync_default_data():
  6. db = SessionLocal()
  7. try:
  8. await dialog_menu_sync(db) # 小数
  9. await default_group_sync(db) # 默认组
  10. await default_role_sync(db) # 默认角色
  11. await basic_agent_sync(db) # 开发的agent
  12. await admin_account_sync(db) # 管理员账号
  13. await sync_rg_api_token(db) # rg token
  14. await sync_token() # 账号token登录
  15. await sync_complex_api_token(db) # 账号token同步
  16. await system_license_sync(db) # 同步系统license
  17. except Exception as e:
  18. print(e)
  19. finally:
  20. db.close()