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

perf: 账号备份性能优化

pull/7629/head
feng626 2022-02-16 18:29:55 +08:00 committed by Jiangjie.Bai
parent c7f8ebb613
commit b088362ae3
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/assets/task_handlers/backup/handlers.py
View File

@ -52,10 +52,11 @@ class BaseAccountHandler:
return header_fields return header_fields
@classmethod @classmethod
def create_row(cls, account, serializer_cls): def create_row(cls, account, serializer_cls, header_fields=None):
serializer = serializer_cls(account) serializer = serializer_cls(account)
data = cls.unpack_data(serializer.data) if not header_fields:
header_fields = cls.get_header_fields(serializer) header_fields = cls.get_header_fields(serializer)
data = cls.unpack_data(serializer.data)
row_dict = {} row_dict = {}
for field, header_name in header_fields.items(): for field, header_name in header_fields.items():
row_dict[header_name] = data[field] row_dict[header_name] = data[field]
@ -74,12 +75,16 @@ class AssetAccountHandler(BaseAccountHandler):
def create_df(cls): def create_df(cls):
df_dict = defaultdict(list) df_dict = defaultdict(list)
sheet_name = AuthBook._meta.verbose_name sheet_name = AuthBook._meta.verbose_name
accounts = AuthBook.get_queryset()
accounts = AuthBook.get_queryset().select_related('systemuser')
if not accounts.first():
return df_dict
header_fields = cls.get_header_fields(AccountSecretSerializer(accounts.first()))
for account in accounts: for account in accounts:
account.load_auth() account.load_auth()
row = cls.create_row(account, AccountSecretSerializer) row = cls.create_row(account, AccountSecretSerializer, header_fields)
df_dict[sheet_name].append(row) df_dict[sheet_name].append(row)
for k, v in df_dict.items(): for k, v in df_dict.items():
df_dict[k] = pd.DataFrame(v) df_dict[k] = pd.DataFrame(v)
@ -98,7 +103,7 @@ class AppAccountHandler(BaseAccountHandler):
@classmethod @classmethod
def create_df(cls): def create_df(cls):
df_dict = defaultdict(list) df_dict = defaultdict(list)
accounts = Account.get_queryset() accounts = Account.get_queryset().select_related('systemuser')
for account in accounts: for account in accounts:
account.load_auth() account.load_auth()
app_type = account.type app_type = account.type