network_inventory/devices/models/device.py

112 lines
3.8 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
2020-01-10 00:01:06 +01:00
from customers.models import Customer, Owner, Location
from users.models import User
from nets.models import 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 HardwareModel(models.Model):
name = models.CharField(max_length=50)
manufacturer = models.ForeignKey(DeviceManufacturer,
on_delete=models.CASCADE)
def __str__(self):
return self.name
2020-01-09 21:59:37 +01:00
class Device(models.Model):
name = models.CharField(max_length=50)
description = models.TextField(blank=True)
serialnumber = models.CharField(max_length=50, blank=True)
2020-02-29 21:01:26 +01:00
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)
2020-02-29 21:01:26 +01:00
customer = models.ForeignKey(Customer,
on_delete=models.CASCADE)
manufacturer = models.ForeignKey(DeviceManufacturer,
models.SET_NULL,
null=True,
blank=True)
model = models.ForeignKey(HardwareModel,
models.SET_NULL,
null=True,
blank=True)
2020-02-29 21:01:26 +01:00
location = models.ForeignKey(Location,
models.SET_NULL,
null=True,
2020-01-09 21:59:37 +01:00
blank=True)
2020-02-29 21:01:26 +01:00
user = models.ForeignKey(User,
models.SET_NULL,
null=True,
blank=True)
2020-01-09 21:59:37 +01:00
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
2020-02-16 16:56:46 +01:00
def nets(self):
2020-02-29 21:10:03 +01:00
nets_result = DeviceInNet.objects.filter(
device=self.id).order_by('net')
2020-02-16 16:56:46 +01:00
nets = []
for net in nets_result:
ip_addresses = {}
ip_addresses['net_id'] = net.net.id
ip_addresses['ip'] = "{}: {}".format(net.ip, net.ip_status)
2020-02-16 16:56:46 +01:00
ip_addresses['name'] = net.net.name
nets.append(ip_addresses)
return nets
2020-01-09 21:59:37 +01:00
class Meta:
verbose_name_plural = "Connected Devices"
2020-02-15 18:50:51 +01:00
def get_absolute_url(self):
from django.urls import reverse
return reverse('connected_device', args=[str(self.id)])
2020-01-09 21:59:37 +01:00
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", blank=True, null=True)
2020-01-09 21:59:37 +01:00
nic = models.CharField(max_length=50, blank=True, verbose_name="NIC")
2020-02-29 21:11:06 +01:00
mac_address = models.CharField(max_length=50,
blank=True,
2020-02-15 18:50:51 +01:00
verbose_name="MAC Address")
2020-02-29 21:11:06 +01:00
ip_status = models.ForeignKey(IpStatus,
models.SET_NULL,
null=True,
blank=True,
verbose_name="IP Status")
2020-01-09 21:59:37 +01:00
def __str__(self):
return self.ip
class Meta:
verbose_name_plural = "Devices in Net"