From 9f153c49d80757df31af0184a06520953c0b780a Mon Sep 17 00:00:00 2001
From: Gabriel Detraz <detraz@crans.org>
Date: Wed, 25 Apr 2018 18:38:40 +0200
Subject: [PATCH] Ca me broute donc je fais simple

---
 re2o/mixins.py      | 18 ------------------
 topologie/models.py | 40 +++++++++++++++++++++++++++++++++++++---
 2 files changed, 37 insertions(+), 21 deletions(-)

diff --git a/re2o/mixins.py b/re2o/mixins.py
index 036b4f72..5fb3e468 100644
--- a/re2o/mixins.py
+++ b/re2o/mixins.py
@@ -160,21 +160,3 @@ class AclMixin(object):
         )
 
 
-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)
diff --git a/topologie/models.py b/topologie/models.py
index 6f4999d1..57f66901 100644
--- a/topologie/models.py
+++ b/topologie/models.py
@@ -49,7 +49,9 @@ from django.db import transaction
 from reversion import revisions as reversion
 
 from machines.models import Machine, regen
-from re2o.mixins import AclMixin, RevMixin, SwitchPluggedMixin
+from re2o.mixins import AclMixin, RevMixin
+
+
 
 
 class Stack(AclMixin, RevMixin, models.Model):
@@ -85,7 +87,7 @@ class Stack(AclMixin, RevMixin, models.Model):
                 inférieure à l'id minimale"})
 
 
-class AccessPoint(AclMixin, SwitchPluggedMixin, Machine):
+class AccessPoint(AclMixin, Machine):
     """Define a wireless AP. Inherit from machines.interfaces
 
     Definition pour une borne wifi , hérite de machines.interfaces
@@ -104,6 +106,22 @@ class AccessPoint(AclMixin, SwitchPluggedMixin, 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/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)
+
     @classmethod
     def all_ap_in(cls, building_instance):
         """Get a building as argument, returns all ap of a building"""
@@ -113,12 +131,28 @@ class AccessPoint(AclMixin, SwitchPluggedMixin, Machine):
         return str(self.interface_set.first())
 
 
-class Server(Machine, SwitchPluggedMixin):
+class Server(Machine):
     """Dummy class, to retrieve servers of a building, or get switch of a server"""
 
     class Meta:
         proxy = True
 
+    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)
+
     @classmethod
     def all_server_in(cls, building_instance):
         """Get a building as argument, returns all server of a building"""