Factorise les get_instance dans le mixin acl

cotisations/models.py

@@ -59,7 +59,7 @@ from machines.models import regen
 from re2o.field_permissions import FieldPermissionModelMixin
 from re2o.mixins import AclMixin
 
-class Facture(FieldPermissionModelMixin, models.Model):
+class Facture(AclMixin, FieldPermissionModelMixin, models.Model):
     """ Définition du modèle des factures. Une facture regroupe une ou
     plusieurs ventes, rattachée à un user, et reliée à un moyen de paiement
     et si il y a lieu un numero pour les chèques. Possède les valeurs
@@ -114,13 +114,6 @@ class Facture(FieldPermissionModelMixin, models.Model):
             ).values_list('name', flat=True))
         return name
 
-    def get_instance(factureid, *args, **kwargs):
-        return Facture.objects.get(pk=factureid)
-
-    def can_create(user_request, *args, **kwargs):
-        return user_request.has_perm('cotisations.add_facture'), u"Vous n'avez pas le\
-            droit de créer des factures"
-
     def can_edit(self, user_request, *args, **kwargs):
         if not user_request.has_perm('cotisations.change_facture'):
             return False, u"Vous n'avez pas le droit d'éditer les factures"
@@ -144,11 +137,6 @@ class Facture(FieldPermissionModelMixin, models.Model):
         else:
             return True, None
 
-    def can_view_all(user_request, *args, **kwargs):
-        if not user_request.has_perm('cotisations.view_facture'):
-            return False, u"Vous n'avez pas le droit de voir les factures"
-        return True, None
-
     def can_view(self, user_request, *args, **kwargs):
         if not user_request.has_perm('cotisations.view_facture') and\
             self.user != user_request:
@@ -192,7 +180,7 @@ def facture_post_delete(sender, **kwargs):
     user.ldap_sync(base=False, access_refresh=True, mac_refresh=False)
 
 
-class Vente(models.Model):
+class Vente(AclMixin, models.Model):
     """Objet vente, contient une quantité, une facture parente, un nom,
     un prix. Peut-être relié à un objet cotisation, via le boolean
     iscotisation"""
@@ -277,14 +265,6 @@ class Vente(models.Model):
         self.update_cotisation()
         super(Vente, self).save(*args, **kwargs)
 
-    def get_instance(venteid, *args, **kwargs):
-        return Vente.objects.get(pk=venteid)
-
-    def can_create(user_request, *args, **kwargs):
-        return user_request.has_perm('cotisations.add_vente'), u"Vous n'avez pas le\
-            droit de créer des ventes"
-        return True, None
-
     def can_edit(self, user_request, *args, **kwargs):
         if not user_request.has_perm('cotisations.change_vente'):
             return False, u"Vous n'avez pas le droit d'éditer les ventes"
@@ -308,11 +288,6 @@ class Vente(models.Model):
         else:
             return True, None
 
-    def can_view_all(user_request, *args, **kwargs):
-        if not user_request.has_perm('cotisations.view_vente'):
-            return False, u"Vous n'avez pas le droit de voir les ventes"
-        return True, None
-
     def can_view(self, user_request, *args, **kwargs):
         if not user_request.has_perm('cotisations.view_vente') and\
             self.facture.user != user_request:
@@ -402,9 +377,6 @@ class Article(AclMixin, models.Model):
                 "La durée est obligatoire si il s'agit d'une cotisation"
             )
 
-    def get_instance(articleid, *args, **kwargs):
-        return Article.objects.get(pk=articleid)
-
     def __str__(self):
         return self.name
 
@@ -420,9 +392,6 @@ class Banque(AclMixin, models.Model):
             ("view_banque", "Peut voir un objet banque"),
         )
 
-    def get_instance(banqueid, *args, **kwargs):
-        return Banque.objects.get(pk=banqueid)
-
     def __str__(self):
         return self.name
 
@@ -443,9 +412,6 @@ class Paiement(AclMixin, models.Model):
             ("view_paiement", "Peut voir un objet paiement"),
         )
 
-    def get_instance(paiementid, *args, **kwargs):
-        return Paiement.objects.get(pk=paiementid)
-
     def __str__(self):
         return self.moyen
 
@@ -460,7 +426,7 @@ class Paiement(AclMixin, models.Model):
         super(Paiement, self).save(*args, **kwargs)
 
 
-class Cotisation(models.Model):
+class Cotisation(AclMixin, models.Model):
     """Objet cotisation, debut et fin, relié en onetoone à une vente"""
     PRETTY_NAME = "Cotisations"
 
@@ -485,14 +451,6 @@ class Cotisation(models.Model):
             ("change_all_cotisation", "Superdroit, peut modifier toutes les cotisations"),
         )
 
-    def get_instance(cotisationid, *args, **kwargs):
-        return Cotisations.objects.get(pk=cotisationid)
-
-    def can_create(user_request, *args, **kwargs):
-        return user_request.has_perm('cotisations.add_cotisation'), u"Vous n'avez pas le\
-            droit de créer des cotisations"
-        return True, None
-
     def can_edit(self, user_request, *args, **kwargs):
         if not user_request.has_perm('cotisations.change_cotisation'):
             return False, u"Vous n'avez pas le droit d'éditer les cotisations"
@@ -512,11 +470,6 @@ class Cotisation(models.Model):
         else:
             return True, None
 
-    def can_view_all(user_request, *args, **kwargs):
-        if not user_request.has_perm('cotisations.view_cotisation'):
-            return False, u"Vous n'avez pas le droit de voir les cotisations"
-        return True, None
-
     def can_view(self, user_request, *args, **kwargs):
         if not user_request.has_perm('cotisations.view_cotisation') and\
             self.vente.facture.user != user_request:

re2o/mixins.py

@@ -3,9 +3,7 @@
 # se veut agnostique au réseau considéré, de manière à être installable en
 # quelques clics.
 #
-# Copyright © 2017  Gabriel Détraz
-# Copyright © 2017  Goulven Kermarec
-# Copyright © 2017  Augustin Lemesle
+# Copyright © 2018  Gabriel Détraz
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -30,6 +28,14 @@ class AclMixin(object):
     def get_modulename(cls):
         return str(cls.__module__).split('.')[0].lower()
 
+    @classmethod
+    def get_instance(cls, *args, **kwargs):
+        """Récupère une instance
+        :param objectid: Instance id à trouver
+        :return: Une instance de la classe évidemment"""
+        object_id = kwargs.get(cls.get_classname() + 'id')
+        return cls.objects.get(pk=object_id)
+
     @classmethod
     def can_create(cls, user_request, *args, **kwargs):
         """Verifie que l'user a les bons droits infra pour créer

topologie/models.py

@@ -67,9 +67,6 @@ class Stack(AclMixin, models.Model):
             ("view_stack", "Peut voir un objet stack"),
         )
 
-    def get_instance(stack_id, *args, **kwargs):
-        return Stack.objects.get(pk=stack_id)
-
     def __str__(self):
         return " ".join([self.name, self.stack_id])
 
@@ -105,9 +102,6 @@ class AccessPoint(AclMixin, Machine):
             ("view_accesspoint", "Peut voir une borne"),
         )
 
-    def get_instance(ap_id, *args, **kwargs):
-        return AccessPoint.objects.get(pk=ap_id)
-
 
 class Switch(AclMixin, Machine):
     """ Definition d'un switch. Contient un nombre de ports (number),
@@ -146,9 +140,6 @@ class Switch(AclMixin, Machine):
             ("view_switch", "Peut voir un objet switch"),
         )
 
-    def get_instance(switch_id, *args, **kwargs):
-        return Switch.objects.get(pk=switch_id)
-
     def clean(self):
         """ Verifie que l'id stack est dans le bon range
         Appelle également le clean de la classe parente"""
@@ -210,9 +201,6 @@ class ModelSwitch(AclMixin, models.Model):
             ("view_modelswitch", "Peut voir un objet modelswitch"),
         )
 
-    def get_instance(model_switch_id, *args, **kwargs):
-        return ModelSwitch.objects.get(pk=model_switch_id)
-
     def __str__(self):
         return str(self.constructor) + ' ' + self.reference
 
@@ -227,9 +215,6 @@ class ConstructorSwitch(AclMixin, models.Model):
             ("view_constructorswitch", "Peut voir un objet constructorswitch"),
         )
 
-    def get_instance(constructor_switch_id, *args, **kwargs):
-        return ConstructorSwitch.objects.get(pk=constructor_switch_id)
-
     def __str__(self):
         return self.name
 
@@ -363,9 +348,6 @@ class Room(AclMixin, models.Model):
             ("view_room", "Peut voir un objet chambre"),
         )
 
-    def get_instance(room_id, *args, **kwargs):
-        return Room.objects.get(pk=room_id)
-
     def __str__(self):
         return self.name
 

users/models.py

@@ -990,9 +990,6 @@ class ServiceUser(AclMixin, AbstractBaseUser):
             )]).values_list('dn', flat=True))
         group.save()
 
-    def get_instance(userid, *args, **kwargs):
-        return ServiceUser.objects.get(pk=userid)
-
     def __str__(self):
         return self.pseudo
 
@@ -1021,9 +1018,6 @@ class School(AclMixin, models.Model):
             ("view_school", "Peut voir un objet school"),
         )
 
-    def get_instance(schoolid, *args, **kwargs):
-        return School.objects.get(pk=schoolid)
-
     def __str__(self):
         return self.name
 
@@ -1058,9 +1052,6 @@ class ListRight(AclMixin, Group):
             ("view_listright", "Peut voir un objet Group/ListRight"),
         )
 
-    def get_instance(listrightid, *args, **kwargs):
-        return ListRight.objects.get(pk=listrightid)
-
     def __str__(self):
         return self.name
 
@@ -1110,9 +1101,6 @@ class ListShell(AclMixin, models.Model):
             ("view_listshell", "Peut voir un objet shell quelqu'il soit"),
         )
 
-    def get_instance(shellid, *args, **kwargs):
-        return ListShell.objects.get(pk=shellid)
-
     def get_pretty_name(self):
         """Return the canonical name of the shell"""
         return self.shell.split("/")[-1]
@@ -1236,9 +1224,6 @@ class Whitelist(AclMixin, models.Model):
     def is_active(self):
         return self.date_end > timezone.now()
 
-    def get_instance(whitelistid, *args, **kwargs):
-        return Whitelist.objects.get(pk=whitelistid)
-
     def can_view(self, user_request, *args, **kwargs):
         """Check if an user can view a Whitelist object.