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

修改授权问题

pull/2/head
guanghongwei 2014-09-27 20:58:50 +08:00
parent 5fd15478a8
commit 52e567b157
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
webroot/AutoSa/AutoSa/views.py
View File

@ -166,6 +166,13 @@ def group_member(username):
return list(set(member)) return list(set(member))
def user_assets(username):
assets = []
user = User.objects.get(username=username)
for asset in user.assetsuser_set.all():
assets.append(asset.aid)
return assets
def login_required(func): def login_required(func):
"""要求登录的装饰器""" """要求登录的装饰器"""
def _deco(request, *args, **kwargs): def _deco(request, *args, **kwargs):
@ -559,7 +566,10 @@ def addPerm(request):
for asset_user in assets_user: for asset_user in assets_user:
have_assets.append(asset_user.aid) have_assets.append(asset_user.aid)
if request.session.get('admin') == 2:
all_assets = Assets.objects.all() all_assets = Assets.objects.all()
else:
all_assets = user_assets(request.session.get('username'))
other_assets = list(set(all_assets) - set(have_assets)) other_assets = list(set(all_assets) - set(have_assets))
return render_to_response('addUserPerm.html', return render_to_response('addUserPerm.html',
{'user': user, 'assets': other_assets, 'perm_menu': 'active'}, {'user': user, 'assets': other_assets, 'perm_menu': 'active'},