8
0
Fork 0
mirror of https://gitlab2.federez.net/re2o/re2o synced 2024-12-23 07:23:46 +00:00

Factorise + fonctions pour recup switch et building d'une machine

This commit is contained in:
Gabriel Detraz 2018-04-25 08:36:36 +02:00 committed by chirac
parent e2f969f831
commit 2c02a9fb6b
2 changed files with 38 additions and 17 deletions

View file

@ -24,6 +24,7 @@ A set of mixins used all over the project to avoid duplicating code
"""
from reversion import revisions as reversion
from django.utils.functional import cached_property
class RevMixin(object):
@ -157,3 +158,23 @@ class AclMixin(object):
),
u"Vous n'avez pas le droit de voir des " + self.get_classname()
)
class SwitchPluggedMixin(object):
"""Mixin to get building.switch where the device is plugged"""
def switch(self):
"""Return the switch where this is plugged"""
return Switch.objects.filter(
ports__machine_interface__machine=self
)
def building(self):
"""Return the building of the AP/Server (building of the switchs connected to...)"""
return Building.objects.filter(
switchbay__switch=self.switch()
)
@cached_property
def short_name(self):
return str(self.interface_set.first().domain.name)

View file

@ -49,7 +49,7 @@ from django.db import transaction
from reversion import revisions as reversion
from machines.models import Machine, regen
from re2o.mixins import AclMixin, RevMixin
from re2o.mixins import AclMixin, RevMixin, SwitchPluggedMixin
class Stack(AclMixin, RevMixin, models.Model):
@ -85,7 +85,7 @@ class Stack(AclMixin, RevMixin, models.Model):
inférieure à l'id minimale"})
class AccessPoint(AclMixin, Machine):
class AccessPoint(AclMixin, SwitchPluggedMixin, Machine):
"""Define a wireless AP. Inherit from machines.interfaces
Definition pour une borne wifi , hérite de machines.interfaces
@ -104,18 +104,6 @@ class AccessPoint(AclMixin, Machine):
("view_accesspoint", "Peut voir une borne"),
)
def switch(self):
"""Return the switch where this is plugged"""
return Switch.objects.filter(
ports__machine_interface__machine=self
)
def building(self):
"""Return the building of the AP (building of the switchs connected to...)"""
return Building.objects.filter(
switchbay__switch=self.switch()
)
@classmethod
def all_ap_in(cls, building_instance):
"""Get a building as argument, returns all ap of a building"""
@ -124,9 +112,21 @@ class AccessPoint(AclMixin, Machine):
def __str__(self):
return str(self.interface_set.first())
@cached_property
def short_name(self):
return str(self.interface_set.first().domain.name)
class Server(Machine, SwitchPluggedMixin):
"""Dummy class, to retrieve servers of a building, or get switch of a server"""
class Meta:
proxy = True
@classmethod
def all_server_in(cls, building_instance):
"""Get a building as argument, returns all server of a building"""
return cls.objects.filter(interface__port__switch__switchbay__building=building_instance).exclude(accesspoint__isnull=False)
def __str__(self):
return str(self.interface_set.first())
class Switch(AclMixin, Machine):
""" Definition d'un switch. Contient un nombre de ports (number),