From ae364ac3730c2ae3397add93a3ce57c50d469053 Mon Sep 17 00:00:00 2001 From: wangruidong <940853815@qq.com> Date: Tue, 14 Nov 2023 15:49:47 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E5=BD=95=E5=83=8F=E5=AD=98=E5=82=A8?= =?UTF-8?q?=E4=B8=8B=E8=BD=BD=E6=8A=A5=E9=94=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/common/storage/base.py | 13 +++++++------ apps/terminal/models/component/storage.py | 4 ++++ 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/apps/common/storage/base.py b/apps/common/storage/base.py index 66b4a7096..cb1d1b5b7 100644 --- a/apps/common/storage/base.py +++ b/apps/common/storage/base.py @@ -7,7 +7,6 @@ from django.conf import settings from terminal.models import default_storage, ReplayStorage from common.utils import get_logger, make_dirs - logger = get_logger(__name__) @@ -26,11 +25,13 @@ class BaseStorageHandler(object): def download(self): replay_storages = ReplayStorage.objects.all() - configs = { - storage.name: storage.config - for storage in replay_storages - if not storage.type_null_or_server - } + configs = {} + for storage in replay_storages: + if storage.type_sftp: + continue + if storage.type_null_or_server: + continue + configs[storage.name] = storage.config if settings.SERVER_REPLAY_STORAGE: configs['SERVER_REPLAY_STORAGE'] = settings.SERVER_REPLAY_STORAGE if not configs: diff --git a/apps/terminal/models/component/storage.py b/apps/terminal/models/component/storage.py index 6a34f3f5a..f122c90b3 100644 --- a/apps/terminal/models/component/storage.py +++ b/apps/terminal/models/component/storage.py @@ -162,6 +162,10 @@ class ReplayStorage(CommonStorageModelMixin, JMSBaseModel): def type_swift(self): return self.type == const.ReplayStorageType.swift.value + @property + def type_sftp(self): + return self.type == const.ReplayStorageType.sftp.value + @property def type_ceph(self): return self.type == const.ReplayStorageType.ceph.value