klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2025-07-11 04:19:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
Skia
2016-04-20 03:30:49 +02:00
parent e5c085c706
commit b4c3f2f4dc
5 changed files with 59 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
core/models.py
View File

@ -128,24 +128,25 @@ class User(AbstractBaseUser, PermissionsMixin):
if group_name == settings.SITH_GROUPS['public']['name']:
return True
if group_name == settings.SITH_MAIN_MEMBERS_GROUP: # We check the subscription if asked
try: # TODO: change for a test in settings.INSTALLED_APP
if 'subscription' in settings.INSTALLED_APPS:
from subscription import Subscriber
s = Subscriber.objects.filter(pk=self.pk).first()
if s is not None and s.is_subscribed():
return True
else:
return False
except Exception as e:
print(e)
else:
return False
if group_name[-6:] == settings.SITH_BOARD_SUFFIX:
try: # TODO: change for a test in settings.INSTALLED_APP
if 'club' in settings.INSTALLED_APPS:
from club.models import Club
name = group_name[:-6]
c = Club.objects.filter(unix_name=name).first()
return c.get_membership_for(self).role >= 2
except Exception as e:
print(e)
mem = c.get_membership_for(self)
if mem:
return mem.role >= 2
return False
else:
return False
return self.groups.filter(name=group_name).exists()