79 lines
2.0 KiB
Python
79 lines
2.0 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎
|
|||
|
|
@file： common.py
|
|||
|
|
@date：2025/3/25 16:26
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
from django.db.models import QuerySet
|
|||
|
|
|
|||
|
|
from common.util.common import encryption
|
|||
|
|
from setting.models import Model
|
|||
|
|
from users.models import User
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_model_operation_object(model_id):
|
|||
|
|
model_model = QuerySet(model=Model).filter(id=model_id).first()
|
|||
|
|
if model_model is not None:
|
|||
|
|
return {
|
|||
|
|
"name": model_model.name
|
|||
|
|
}
|
|||
|
|
return {}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_member_operation_object(member_id):
|
|||
|
|
user_model = QuerySet(model=User).filter(id=member_id).first()
|
|||
|
|
if user_model is not None:
|
|||
|
|
return {
|
|||
|
|
"name": user_model.username
|
|||
|
|
}
|
|||
|
|
return {}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_member_operation_object_batch(member_id_list):
|
|||
|
|
user_model_list = QuerySet(model=User).filter(id__in=member_id_list)
|
|||
|
|
if user_model_list is not None:
|
|||
|
|
return {
|
|||
|
|
"name": f'[{",".join([user.username for user in user_model_list])}]',
|
|||
|
|
"user_list": [{'name': user.username} for user in user_model_list]
|
|||
|
|
}
|
|||
|
|
return {}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def encryption_str(_value):
|
|||
|
|
if isinstance(_value, str):
|
|||
|
|
return encryption(_value)
|
|||
|
|
return _value
|
|||
|
|
|
|||
|
|
|
|||
|
|
def encryption_credential(credential):
|
|||
|
|
if isinstance(credential, dict):
|
|||
|
|
return {key: encryption_str(credential.get(key)) for key in credential}
|
|||
|
|
return credential
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_edit_model_details(request):
|
|||
|
|
path = request.path
|
|||
|
|
body = request.data
|
|||
|
|
query = request.query_params
|
|||
|
|
credential = body.get('credential', {})
|
|||
|
|
credential_encryption_ed = encryption_credential(credential)
|
|||
|
|
return {
|
|||
|
|
'path': path,
|
|||
|
|
'body': {**body, 'credential': credential_encryption_ed},
|
|||
|
|
'query': query
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_email_details(request):
|
|||
|
|
path = request.path
|
|||
|
|
body = request.data
|
|||
|
|
query = request.query_params
|
|||
|
|
email_host_password = body.get('email_host_password', '')
|
|||
|
|
return {
|
|||
|
|
'path': path,
|
|||
|
|
'body': {**body, 'email_host_password': encryption_str(email_host_password)},
|
|||
|
|
'query': query
|
|||
|
|
}
|