network_inventory/devices/models.py

72 lines
2.4 KiB
Python
Raw Normal View History

2020-01-09 18:01:14 +01:00
from django.db import models
2020-01-09 21:59:37 +01:00
from core.models import Category, Company
from customer.models import Customer, Owner
2020-01-09 21:59:49 +01:00
from customer.models import Location
from inventory.models import User, Net, IpStatus
2020-01-09 18:01:14 +01:00
2020-01-09 21:59:37 +01:00
class DeviceManufacturer(Company):
class Meta:
verbose_name_plural = "Device Manufacturers"
class DeviceCategory(Category):
class Meta:
verbose_name_plural = "Device Categories"
class Device(models.Model):
name = models.CharField(max_length=50)
description = models.TextField(blank=True)
serialnumber = models.CharField(max_length=50, blank=True)
category = models.ForeignKey(DeviceCategory, on_delete=models.SET_NULL,
null=True, blank=True)
owner = models.ForeignKey(Owner, on_delete=models.SET_NULL, null=True, blank=True)
customer = models.ForeignKey(Customer, on_delete=models.CASCADE)
manufacturer = models.ForeignKey(DeviceManufacturer, models.SET_NULL,
null=True, blank=True)
location = models.ForeignKey(Location, models.SET_NULL, null=True,
blank=True)
user = models.ForeignKey(User, models.SET_NULL, null=True, blank=True)
installation_date = models.DateField(null=True, blank=True)
def __str__(self):
return self.name
def get_absolute_url(self):
from django.urls import reverse
return reverse('device', args=[str(self.id)])
class ConnectedDevice(Device):
net = models.ManyToManyField(Net, through='DeviceInNet')
@property
def ips(self):
nets = DeviceInNet.objects.filter(device=self.id)
ip_addresses = {}
for net in nets:
ip_addresses[net.net_id] = net.ip
return ip_addresses
class Meta:
verbose_name_plural = "Connected Devices"
class DeviceInNet(models.Model):
device = models.ForeignKey(ConnectedDevice, on_delete=models.CASCADE)
net = models.ForeignKey(Net, on_delete=models.CASCADE)
ip = models.GenericIPAddressField(verbose_name="IP")
nic = models.CharField(max_length=50, blank=True, verbose_name="NIC")
mac_address = models.CharField(max_length=50, blank=True, verbose_name="MAC Address")
ip_status = models.ForeignKey(IpStatus, models.SET_NULL, null=True,
blank=True, verbose_name="IP Status")
def __str__(self):
return self.ip
class Meta:
verbose_name_plural = "Devices in Net"