perf: 修改网关自动化任务 (#9335)

Co-authored-by: feng <1304903146@qq.com>
This commit is contained in:
fit2bot
2023-01-18 17:14:02 +08:00
committed by GitHub
parent 571e9b1878
commit c2a8acb73b
10 changed files with 85 additions and 26 deletions

View File

@@ -0,0 +1,21 @@
from common.utils import get_logger
from accounts.const import AutomationTypes
from assets.automations.ping_gateway.manager import PingGatewayManager
logger = get_logger(__name__)
class VerifyGatewayAccountManager(PingGatewayManager):
@classmethod
def method_type(cls):
return AutomationTypes.verify_gateway_account
@staticmethod
def before_runner_start():
logger.info(">>> 开始执行测试网关账号可连接性任务")
def get_accounts(self, gateway):
usernames = self.execution.snapshot['accounts']
accounts = gateway.accounts.filter(username__in=usernames)
return accounts