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

Merge pull request #9702 from jumpserver/pr@dev@perf_ops_permed_assets_error_msg 

perf: 优化报错代码
pull/9704/head
Eric_Lee 2023-02-22 21:18:40 +08:00 committed by GitHub
parent 49c7a12bb3 9b2f422622
commit 3aa7d1841b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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()
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():
if asset not in has_permed_assets:
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):
raise Exception("You do not have access rights to some assets")
if error_assets_count > 0:
raise Exception("You do not have access rights to {} assets".format(error_assets_count))
def before_start(self):
self.check_assets_perms()