From 0f7b481633a56aa2c865262db5affcd35a84024d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ma=C3=ABl=20Kervella?= Date: Sun, 3 Dec 2017 15:32:43 +0000 Subject: [PATCH] Can_delete for machines.models --- machines/models.py | 122 ++++++++++++++++++++++++++++++++++++--------- 1 file changed, 99 insertions(+), 23 deletions(-) diff --git a/machines/models.py b/machines/models.py index 2ba0c652..e7e92261 100644 --- a/machines/models.py +++ b/machines/models.py @@ -78,6 +78,12 @@ class Machine(models.Model): def can_edit(self, user_request, *args, **kwargs): return True, None + def can_delete(self, user_request, *args, **kwargs): + if not user_request.has_perms(('cableur',)) and self.user != user_request: + return False, u"Vous ne pouvez pas éditer une machine d'un autre user\ + que vous sans droit" + return True, None + def __str__(self): return str(self.user) + ' - ' + str(self.id) + ' - ' + str(self.name) @@ -103,14 +109,19 @@ class MachineType(models.Model): return MachineType.objects.get(pk=machinetypeid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un type de machine" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un type de machine" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des types de machine" return True, None + def can_delete(self, user_request, *args, **kwargs): + if not user_request.has_perms(('infra',)): + return False, u"Vous n'avez pas le droit de supprimer des types de machines" + return True, None + def __str__(self): return self.type @@ -225,14 +236,18 @@ class IpType(models.Model): return IpType.objects.get(pk=iptypeid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un type d'ip" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un type d'ip" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des types d'ip" return True, None + def can_delete(user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer un type d'ip" + def __str__(self): return self.type @@ -250,14 +265,18 @@ class Vlan(models.Model): return Vlan.objects.get(pk=vlanid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un vlan" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un vlan" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des vlans" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de suprimer un vlan" + def __str__(self): return self.name @@ -296,14 +315,18 @@ class Nas(models.Model): return Nas.objects.get(pk=nasid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un nas" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un nas" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des nas" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer un nas" + def __str__(self): return self.name @@ -344,14 +367,18 @@ class SOA(models.Model): return SOA.objects.get(pk=soaid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un enregistrement SOA" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un enregistrement SOA" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des enregistrements SOA" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer des enregistrements SOA" + def __str__(self): return str(self.name) @@ -438,14 +465,18 @@ class Extension(models.Model): return Extension.objects.get(pk=extensionid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer une extension" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer une extension" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des extensions" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer des extension" + def __str__(self): return self.name @@ -475,14 +506,18 @@ class Mx(models.Model): return Mx.objects.get(pk=mxid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un enregistrement MX" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un enregistrement MX" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des enregstrements MX" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer un enregistrement MX" + def __str__(self): return str(self.zone) + ' ' + str(self.priority) + ' ' + str(self.name) @@ -503,14 +538,18 @@ class Ns(models.Model): return Ns.objects.get(pk=nsid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un enregistrement NS" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un enregistrement NS" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des enregistrements NS" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer un enregistrement NS" + def __str__(self): return str(self.zone) + ' ' + str(self.ns) @@ -527,14 +566,18 @@ class Txt(models.Model): return Txt.objects.get(pk=txtid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un enregistrement TXT" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un enregistrement TXT" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des enregistrement TXT" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer des enregistrements TXT" + def __str__(self): return str(self.zone) + " : " + str(self.field1) + " " +\ str(self.field2) @@ -592,14 +635,18 @@ class Srv(models.Model): return Srv.objects.get(pk=srvid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un enregistrement SRV" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un enregistrement SRV" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des enregistrements SRV" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer un enregistrement SRV" + def __str__(self): return str(self.service) + ' ' + str(self.protocole) + ' ' +\ str(self.extension) + ' ' + str(self.priority) +\ @@ -745,6 +792,12 @@ class Interface(models.Model): d'un autre user que vous sans droit" return True, None + def can_delete(self, user_resquest, *args, **kwargs): + if not user_request.has_perms(('cableur',)) and self.machine.user != user_request: + return False, u"Vous ne pouvez pas éditer une machine d'un autre\ + user que vous sans droit" + return True, None + def __str__(self): try: domain = self.domain @@ -877,6 +930,12 @@ class Domain(models.Model): d'un autre user que vous sans droit" return True, None + def can_delete(self, user_request, *args, **kwargs): + if not user_request.has_perms(('cableur',)) and self.machine.user != user_request: + return False, u"Vous ne pouvez pas supprimer un alias à une machine\ + d'un autre user que vous sans droit" + return True, None + def __str__(self): return str(self.name) + str(self.extension) @@ -913,6 +972,9 @@ class IpList(models.Model): def can_edit(self, user_request, *args, **kwargs): return True, None + def can_delete(self, user_request, *args, **kwargs): + return True, None + def __str__(self): return self.ipv4 @@ -957,14 +1019,18 @@ class Service(models.Model): return Service.objects.get(pk=serviceid) def can_create(user_request, *args, **kwargs): - return user_request.has_perms(('infra',)) , u"Vous n'avez pas le droit\ - de créer un service" + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de créer un service" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('infra',)): return False, u"Vous n'avez pas le droit d'éditer des services" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('infra',)), u"Vous n'avez pas le droit\ + de supprimer un service" + def __str__(self): return str(self.service_type) @@ -1014,6 +1080,9 @@ class Service_link(models.Model): def can_edit(self, user_request, *args, **kwargs): return True, None + def can_delete(self, user_request, *args, **kwargs): + return True, None + def __str__(self): return str(self.server) + " " + str(self.service) @@ -1032,13 +1101,17 @@ class OuverturePortList(models.Model): def can_create(user_request, *args, **kwargs): return user_request.has_perms(('bureau',)) , u"Vous n'avez pas le droit\ - d'ouvrir un port" + d'ouvrir un port" def can_edit(self, user_request, *args, **kwargs): if not user_request.has_perms(('bureau',)): return False, u"Vous n'avez pas le droit d'éditer des ouvertures de port" return True, None + def can_delete(self, user_request, *args, **kwargs): + return user_request.has_perms(('bureau',)), u"Vous n'avez pas le droit\ + de supprimer une ouverture de port" + def __str__(self): return self.name @@ -1118,6 +1191,9 @@ class OuverturePort(models.Model): def can_edit(self, user_request, *args, **kwargs): return True, None + def can_delete(self, user_request, *args, **kwargs): + return True, None + def __str__(self): if self.begin == self.end: return str(self.begin)