github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: 优化报错代码

pull/9702/head
Aaron3S 2023-02-22 20:05:31 +08:00
parent 49c7a12bb3
commit 9b2f422622
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/ops/models/job.py
View File

@ -407,12 +407,14 @@ class JobExecution(JMSOrgBaseModel):
all_permed_assets = UserPermAssetUtil(self.creator).get_all_assets() all_permed_assets = UserPermAssetUtil(self.creator).get_all_assets()
has_permed_assets = set(self.current_job.assets.all()) & set(all_permed_assets) has_permed_assets = set(self.current_job.assets.all()) & set(all_permed_assets)
error_assets_count = 0
for asset in self.current_job.assets.all(): for asset in self.current_job.assets.all():
if asset not in has_permed_assets: if asset not in has_permed_assets:
print("\033[31mAsset {}({}) has no access permission\033[0m".format(asset.name, asset.address)) print("\033[31mAsset {}({}) has no access permission\033[0m".format(asset.name, asset.address))
error_assets_count += 1
if self.current_job.assets.count() != len(has_permed_assets): if error_assets_count > 0:
raise Exception("You do not have access rights to some assets") raise Exception("You do not have access rights to {} assets".format(error_assets_count))
def before_start(self): def before_start(self):
self.check_assets_perms() self.check_assets_perms()