mirror of https://github.com/jumpserver/jumpserver
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
# Generated by Django 3.2.12 on 2022-07-11 06:13
|
|
|
|
import time
|
|
from django.db import migrations
|
|
from assets.models import Platform
|
|
|
|
|
|
def migrate_platform_set_ops(apps, *args):
|
|
platform_model = apps.get_model('assets', 'Platform')
|
|
Platform.set_default_platforms_ops(platform_model)
|
|
|
|
|
|
def migrate_accounts(apps, schema_editor):
|
|
auth_book_model = apps.get_model('assets', 'AuthBook')
|
|
account_model = apps.get_model('assets', 'Account')
|
|
|
|
count = 0
|
|
bulk_size = 1000
|
|
print("\nStart migrate accounts")
|
|
while True:
|
|
start = time.time()
|
|
auth_books = auth_book_model.objects \
|
|
.prefetch_related('systemuser') \
|
|
.all()[count:count+bulk_size]
|
|
if not auth_books:
|
|
break
|
|
|
|
count += len(auth_books)
|
|
accounts = []
|
|
# auth book 和 account 相同的属性
|
|
same_attrs = [
|
|
'id', 'comment', 'date_created', 'date_updated',
|
|
'created_by', 'asset_id', 'org_id',
|
|
]
|
|
# 认证的属性，可能是 authbook 的，可能是 systemuser 的
|
|
auth_attrs = ['username', 'password', 'private_key', 'public_key']
|
|
|
|
for auth_book in auth_books:
|
|
values = {attr: getattr(auth_book, attr) for attr in same_attrs}
|
|
values['version'] = 1
|
|
|
|
system_user = auth_book.systemuser
|
|
if system_user:
|
|
values.update({attr: getattr(system_user, attr) for attr in auth_attrs})
|
|
values['created_by'] = str(system_user.id)
|
|
values['privileged'] = system_user.type == 'admin'
|
|
|
|
auth_book_auth = {attr: getattr(auth_book, attr) for attr in auth_attrs}
|
|
auth_book_auth = {attr: value for attr, value in auth_book_auth.items() if value}
|
|
values.update(auth_book_auth)
|
|
values['name'] = values['username']
|
|
|
|
account = account_model(**values)
|
|
accounts.append(account)
|
|
|
|
account_model.objects.bulk_create(accounts, ignore_conflicts=True)
|
|
print("Create accounts: {}-{} using: {:.2f}s".format(
|
|
count - len(auth_books), count, time.time()-start
|
|
))
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('assets', '0099_auto_20220711_1409'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(migrate_accounts),
|
|
migrations.RunPython(migrate_platform_set_ops)
|
|
]
|