from django.db import models from django.forms import ModelForm, Form from django.contrib.contenttypes.models import ContentType from django.contrib.contenttypes.fields import GenericForeignKey from django.core.exceptions import ValidationError import reversion from re2o.settings import VLAN_ID_LIST def make_port_related(port): related_port = port.related related_port.related = port related_port.save() def clean_port_related(port): related_port = port.related_port related_port.related = None related_port.save() class Switch(models.Model): PRETTY_NAME = "Switch / Commutateur" switch_interface = models.OneToOneField('machines.Interface', on_delete=models.CASCADE) location = models.CharField(max_length=255) number = models.IntegerField() details = models.CharField(max_length=255, blank=True) def __str__(self): return str(self.location) + ' ' + str(self.switch_interface) class Port(models.Model): PRETTY_NAME = "Port de switch" STATES_BASE = ( ('NO', 'NO'), ('STRICT', 'STRICT'), ('BLOQ', 'BLOQ'), ('COMMON', 'COMMON'), ) STATES = STATES_BASE + tuple([(str(id), str(id)) for id in VLAN_ID_LIST]) switch = models.ForeignKey('Switch', related_name="ports") port = models.IntegerField() room = models.ForeignKey('Room', on_delete=models.PROTECT, blank=True, null=True) machine_interface = models.ForeignKey('machines.Interface', on_delete=models.SET_NULL, blank=True, null=True) related = models.OneToOneField('self', null=True, blank=True, related_name='related_port') radius = models.CharField(max_length=32, choices=STATES, default='NO') details = models.CharField(max_length=255, blank=True) class Meta: unique_together = ('switch', 'port') def clean(self): if hasattr(self, 'switch'): if self.port > self.switch.number: raise ValidationError("Ce port ne peut exister, numero trop élevé") if self.room and self.machine_interface or self.room and self.related or self.machine_interface and self.related: raise ValidationError("Chambre, interface et related_port sont mutuellement exclusifs") if self.related==self: raise ValidationError("On ne peut relier un port à lui même") if self.related and not self.related.related: if self.related.machine_interface or self.related.room: raise ValidationError("Le port relié est déjà occupé, veuillez le libérer avant de créer une relation") else: make_port_related(self) elif hasattr(self, 'related_port'): clean_port_related(self) def __str__(self): return str(self.switch) + " - " + str(self.port) class Room(models.Model): PRETTY_NAME = "Chambre/ Prise murale" name = models.CharField(max_length=255, unique=True) details = models.CharField(max_length=255, blank=True) def __str__(self): return str(self.name)