Added extra check

This commit is contained in:
Raynaldo Rivera 2014-10-12 17:19:12 -07:00
parent 392c752eb8
commit b869c465bb
3 changed files with 17 additions and 7 deletions

View File

@ -1,5 +1,7 @@
from util import add_member_permission
from util import remove_member_permission
from util import check_if_user_has_permission
from django.conf import settings
from django.http import HttpResponseRedirect
from django.shortcuts import render_to_response
@ -84,6 +86,7 @@ def main_character_change(request, char_id):
generate_corp_group_name(EveManager.get_character_by_id(char_id).corporation_name))
else:
#TODO: disable serivces
if check_if_user_has_permission(request.user, 'alliance_member'):
remove_member_permission(request.user, 'alliance_member')
remove_user_from_group(request.user, settings.DEFAULT_ALLIANCE_GROUP)
remove_user_from_group(request.user,

View File

@ -23,3 +23,10 @@ def remove_member_permission(user, permission):
if user.has_perm(permission):
user.user_permissions.remove(stored_permission)
user.save()
def check_if_user_has_permission(user, permission):
ct = ContentType.objects.get_for_model(User)
stored_permission, created = Permission.objects.get_or_create(codename=permission,
content_type=ct, name=permission)
return user.has_perm(stored_permission)