From 59388655ea7c96b84624f89504a47fe22b8ad89a Mon Sep 17 00:00:00 2001 From: feng <1304903146@qq.com> Date: Fri, 18 Nov 2022 16:55:53 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20es=20=E9=BB=98=E8=AE=A4=E5=AD=98?= =?UTF-8?q?=E5=82=A8500?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/terminal/models/storage.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/apps/terminal/models/storage.py b/apps/terminal/models/storage.py index c333c5498..0793a7325 100644 --- a/apps/terminal/models/storage.py +++ b/apps/terminal/models/storage.py @@ -19,7 +19,6 @@ from .terminal import Terminal from .command import Command from .. import const - logger = get_logger(__file__) @@ -37,10 +36,10 @@ class CommonStorageModelMixin(models.Model): def set_to_default(self): self.is_default = True - self.save() - self.__class__.objects.select_for_update()\ - .filter(is_default=True)\ - .exclude(id=self.id)\ + self.save(update_fields=['is_default']) + self.__class__.objects.select_for_update() \ + .filter(is_default=True) \ + .exclude(id=self.id) \ .update(is_default=False) @classmethod @@ -128,7 +127,10 @@ class CommandStorage(CommonStorageModelMixin, CommonModelMixin): def save(self, force_insert=False, force_update=False, using=None, update_fields=None): - super().save() + super().save( + force_insert=force_insert, force_update=force_update, + using=using, update_fields=update_fields + ) if self.type in TYPE_ENGINE_MAPPING: engine_mod = import_module(TYPE_ENGINE_MAPPING[self.type])