From 52e567b157dda1d10ba8cc3cb52551203e9b278b Mon Sep 17 00:00:00 2001 From: guanghongwei Date: Sat, 27 Sep 2014 20:58:50 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=87=EE=86=BD=E6=95=BC=E9=8E=BA=E5=A0=9F?= =?UTF-8?q?=E6=BD=88=E9=97=82=EE=87=80=EE=95=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- webroot/AutoSa/AutoSa/views.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/webroot/AutoSa/AutoSa/views.py b/webroot/AutoSa/AutoSa/views.py index 1f360f9c1..afd533791 100644 --- a/webroot/AutoSa/AutoSa/views.py +++ b/webroot/AutoSa/AutoSa/views.py @@ -166,6 +166,13 @@ def group_member(username): 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 _deco(request, *args, **kwargs): @@ -559,7 +566,10 @@ def addPerm(request): for asset_user in assets_user: have_assets.append(asset_user.aid) - all_assets = Assets.objects.all() + if request.session.get('admin') == 2: + all_assets = Assets.objects.all() + else: + all_assets = user_assets(request.session.get('username')) other_assets = list(set(all_assets) - set(have_assets)) return render_to_response('addUserPerm.html', {'user': user, 'assets': other_assets, 'perm_menu': 'active'},