From 8ae6863266257f8f0645389a1d5fae8322df17ab Mon Sep 17 00:00:00 2001 From: Bai Date: Mon, 28 Jun 2021 19:36:37 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E7=BB=88=E7=AB=AF?= =?UTF-8?q?=E6=9B=B4=E6=96=B0=E5=AD=98=E5=82=A8=E5=A4=B1=E8=B4=A5=E7=9A=84?= =?UTF-8?q?=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/terminal/models/terminal.py | 6 ------ apps/terminal/serializers/terminal.py | 2 ++ 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/apps/terminal/models/terminal.py b/apps/terminal/models/terminal.py index 0c0c36771..77c9b1ce8 100644 --- a/apps/terminal/models/terminal.py +++ b/apps/terminal/models/terminal.py @@ -176,12 +176,6 @@ class Terminal(StorageMixin, TerminalStatusMixin, models.Model): self.save() return - def save(self, **kwargs): - from .storage import CommandStorage, ReplayStorage - self.command_storage = CommandStorage.default().name - self.replay_storage = ReplayStorage.default().name - return super().save(**kwargs) - def __str__(self): status = "Active" if not self.is_accepted: diff --git a/apps/terminal/serializers/terminal.py b/apps/terminal/serializers/terminal.py index e4640e454..44cb2a3a5 100644 --- a/apps/terminal/serializers/terminal.py +++ b/apps/terminal/serializers/terminal.py @@ -119,5 +119,7 @@ class TerminalRegistrationSerializer(serializers.ModelSerializer): instance.remote_addr = get_request_ip(request) sa = self.service_account.save() instance.user = sa + instance.command_storage = CommandStorage.default().name + instance.replay_storage = ReplayStorage.default().name instance.save() return instance