From 6c3e1bd2db888ac0ba6afc595766a29c1af19865 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20P=C3=A9pin?= Date: Sat, 11 Feb 2017 17:13:08 +0100 Subject: [PATCH 1/2] Fix the loaddevdata script --- bda/management/commands/loadbdadevdata.py | 2 +- cof/management/commands/loaddevdata.py | 16 +++++++++++----- 2 files changed, 12 insertions(+), 6 deletions(-) diff --git a/bda/management/commands/loadbdadevdata.py b/bda/management/commands/loadbdadevdata.py index a8851e3d..f348b310 100644 --- a/bda/management/commands/loadbdadevdata.py +++ b/bda/management/commands/loadbdadevdata.py @@ -77,7 +77,7 @@ class Command(MyBaseCommand): self.stdout.write("Inscription des utilisateurs aux tirages") ChoixSpectacle.objects.all().delete() choices = [] - for user in User.objects.filter(profile__is_cof=True): + for user in User.objects.filter(profile__cof__is_cof=True): for tirage in tirages: part, _ = Participant.objects.get_or_create( user=user, diff --git a/cof/management/commands/loaddevdata.py b/cof/management/commands/loaddevdata.py index 4627c2d0..1525152c 100644 --- a/cof/management/commands/loaddevdata.py +++ b/cof/management/commands/loaddevdata.py @@ -19,6 +19,8 @@ from cof.petits_cours_models import ( PetitCoursAttributionCounter ) +from cof.models import CofProfile + # Où sont stockés les fichiers json DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'data') @@ -47,8 +49,10 @@ class Command(MyBaseCommand): # Gaulois gaulois = self.from_json('gaulois.json', DATA_DIR, User) for user in gaulois: - user.profile.is_cof = True - user.profile.save() + CofProfile.objects.create( + profile=user.profile, + is_cof=True + ) # Romains self.from_json('romains.json', DATA_DIR, User) @@ -65,10 +69,12 @@ class Command(MyBaseCommand): root.set_password('root') root.is_staff = True root.is_superuser = True - root.profile.is_cof = True - root.profile.is_buro = True - root.profile.save() root.save() + CofProfile.objects.create( + profile=root.profile, + is_cof=True, + is_buro=True + ) # --- # Petits cours From 3365d7b9a1cc1163ce6b351a1c2e4d358b1a48c9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20P=C3=A9pin?= Date: Sat, 11 Feb 2017 17:21:59 +0100 Subject: [PATCH 2/2] Updates the decorators --- cof/decorators.py | 8 ++++---- cof/templates/home.html | 12 ++++++------ 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/cof/decorators.py b/cof/decorators.py index d7e70608..a7ced8f7 100644 --- a/cof/decorators.py +++ b/cof/decorators.py @@ -9,8 +9,8 @@ from django_cas_ng.decorators import user_passes_test def is_cof(user): try: - profile = user.profile - return profile.is_cof + cofprofile = user.profile.cof + return cofprofile.is_cof except: return False @@ -21,8 +21,8 @@ cof_required_customdenied = user_passes_test(lambda u: is_cof(u), def is_buro(user): try: - profile = user.profile - return profile.is_buro + cofprofile = user.profile.cof + return cofprofile.is_buro except: return False diff --git a/cof/templates/home.html b/cof/templates/home.html index 5e9732c7..3fc92bfe 100644 --- a/cof/templates/home.html +++ b/cof/templates/home.html @@ -7,7 +7,7 @@
-
+
- {% if not user.profile.is_buro %} + {% if not user.profile.cof.is_buro %}