github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

fix: 修复迁移数据库应用账号缺少的问题

pull/9572/head
Bai 2023-02-15 20:08:02 +08:00 committed by Jiangjie.Bai
parent 8be25e1944
commit cf2089d826
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/assets/migrations/0100_auto_20220711_1413.py
View File

@ -81,7 +81,7 @@ def migrate_db_accounts(apps, schema_editor):
perms = app_perm_model.objects.filter(category__in=['db', 'cloud']) perms = app_perm_model.objects.filter(category__in=['db', 'cloud'])
same_attrs = [ same_attrs = [
'id', 'username', 'comment', 'date_created', 'date_updated', 'username', 'comment', 'date_created', 'date_updated',
'created_by', 'org_id', 'created_by', 'org_id',
] ]
auth_attrs = ['password', 'private_key', 'token'] auth_attrs = ['password', 'private_key', 'token']
@ -96,6 +96,7 @@ def migrate_db_accounts(apps, schema_editor):
index += 1 index += 1
start = time.time() start = time.time()
apps = perm.applications.all()
system_users = perm.system_users.all() system_users = perm.system_users.all()
accounts = [] accounts = []
for s in system_users: for s in system_users:
@ -125,13 +126,14 @@ def migrate_db_accounts(apps, schema_editor):
auth_infos.append((username, 'password', '')) auth_infos.append((username, 'password', ''))
for name, secret_type, secret in auth_infos: for name, secret_type, secret in auth_infos:
account = account_model(**values, name=name, secret=secret, secret_type=secret_type) values['name'] = name
accounts.append(account) values['secret_type'] = secret_type
values['secret'] = secret
apps = perm.applications.all() for app in apps:
for app in apps: values['asset_id'] = str(app.id)
for account in accounts: account = account_model(**values)
setattr(account, 'asset_id', str(app.id)) accounts.append(account)
account_model.objects.bulk_create(accounts, ignore_conflicts=True) account_model.objects.bulk_create(accounts, ignore_conflicts=True)