perf: change tree api

This commit is contained in:
ibuler
2022-12-27 13:20:18 +08:00
parent dd630f0e14
commit 795e952dc1
3 changed files with 30 additions and 4 deletions

View File

@@ -144,7 +144,7 @@ class CategoryTreeApi(SerializeToTreeNodeMixin, generics.ListAPIView):
return self.serialize_assets(assets, key)
def list(self, request, *args, **kwargs):
include_asset = bool(self.request.query_params.get('assets'))
include_asset = self.request.query_params.get('assets', '0') == '1'
if include_asset and self.request.query_params.get('key'):
nodes = self.get_assets()

View File

@@ -0,0 +1,23 @@
from django.db import models
from django.utils.translation import ugettext_lazy as _
from assets.const import AutomationTypes
from .change_secret import ChangeSecretAutomation
__all__ = ['CreateAccountAutomation']
class CreateAccountAutomation(ChangeSecretAutomation):
username = models.CharField(max_length=128, blank=True, verbose_name=_("Username"))
secret_type = models.CharField(max_length=16, verbose_name=_('Secret type'))
secret_strategy = models.CharField(max_length=16, default=SecretPolicy.random,
choices=SecretPolicy.choices, verbose_name=_("Secret strategy"))
secret = models.CharField(max_length=1024, blank=True, verbose_name=_("Secret"))
accounts = None
def save(self, *args, **kwargs):
self.type = AutomationTypes.push_account
super().save(*args, **kwargs)
class Meta:
verbose_name = _("Push asset account")