wkootandManuel Hartung
aa41f8a402
* Updated imports for Netbox 3.6.1 Closes #21 and #23 * Changed Fieldtypes of Start and Expiration Date Closes #22 * Bump to NetBox v3.6.1 * Add pyproject.toml and python workflow --------- Co-authored-by: Manuel Hartung <manuel.hartung@atos.net>
113 lines
3.8 KiB
Python
113 lines
3.8 KiB
Python
from django.db import models
|
|
from django.urls import reverse, reverse_lazy
|
|
from django.utils import safestring
|
|
|
|
from netbox.models import NetBoxModel
|
|
from utilities.querysets import RestrictedQuerySet
|
|
|
|
|
|
class SoftwareProduct(NetBoxModel):
|
|
name = models.CharField(max_length=128)
|
|
description = models.CharField(max_length=255, null=True, blank=True)
|
|
|
|
manufacturer = models.ForeignKey(to="dcim.Manufacturer", on_delete=models.PROTECT, null=True, blank=True)
|
|
|
|
objects = RestrictedQuerySet.as_manager()
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse("plugins:netbox_slm:softwareproduct", kwargs={"pk": self.pk})
|
|
|
|
def get_installation_count(self):
|
|
count = SoftwareProductInstallation.objects.filter(software_product_id=self.pk).count()
|
|
return (
|
|
safestring.mark_safe(
|
|
'<a href="{url}">{count}</a>'.format(
|
|
url=reverse_lazy("plugins:netbox_slm:softwareproductinstallation_list") + f"?q={self.name}",
|
|
count=count,
|
|
)
|
|
)
|
|
if count
|
|
else "0"
|
|
)
|
|
|
|
|
|
class SoftwareProductVersion(NetBoxModel):
|
|
name = models.CharField(max_length=64)
|
|
|
|
software_product = models.ForeignKey(
|
|
to="netbox_slm.SoftwareProduct",
|
|
on_delete=models.PROTECT,
|
|
)
|
|
|
|
objects = RestrictedQuerySet.as_manager()
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse("plugins:netbox_slm:softwareproductversion", kwargs={"pk": self.pk})
|
|
|
|
def get_installation_count(self):
|
|
count = SoftwareProductInstallation.objects.filter(version_id=self.pk).count()
|
|
return (
|
|
safestring.mark_safe(
|
|
'<a href="{url}">{count}</a>'.format(
|
|
url=reverse_lazy("plugins:netbox_slm:softwareproductinstallation_list") + f"?q={self.name}",
|
|
count=count,
|
|
)
|
|
)
|
|
if count
|
|
else "0"
|
|
)
|
|
|
|
|
|
class SoftwareProductInstallation(NetBoxModel):
|
|
device = models.ForeignKey(to="dcim.Device", on_delete=models.PROTECT, null=True, blank=True)
|
|
virtualmachine = models.ForeignKey(
|
|
to="virtualization.VirtualMachine", on_delete=models.PROTECT, null=True, blank=True
|
|
)
|
|
software_product = models.ForeignKey(to="netbox_slm.SoftwareProduct", on_delete=models.PROTECT)
|
|
version = models.ForeignKey(to="netbox_slm.SoftwareProductVersion", on_delete=models.PROTECT)
|
|
|
|
objects = RestrictedQuerySet.as_manager()
|
|
|
|
def __str__(self):
|
|
return f"{self.pk} ({self.platform})"
|
|
|
|
def get_absolute_url(self):
|
|
return reverse("plugins:netbox_slm:softwareproductinstallation", kwargs={"pk": self.pk})
|
|
|
|
@property
|
|
def platform(self):
|
|
return self.device or self.virtualmachine
|
|
|
|
def render_type(self):
|
|
return "device" if self.device else "virtualmachine"
|
|
|
|
|
|
class SoftwareLicense(NetBoxModel):
|
|
name = models.CharField(max_length=128)
|
|
description = models.CharField(max_length=255, null=True, blank=True)
|
|
type = models.CharField(max_length=128)
|
|
|
|
stored_location = models.CharField(max_length=255, null=True, blank=True)
|
|
start_date = models.DateField(null=True, blank=True)
|
|
expiration_date = models.DateField(null=True, blank=True)
|
|
|
|
software_product = models.ForeignKey(to="netbox_slm.SoftwareProduct", on_delete=models.PROTECT)
|
|
version = models.ForeignKey(to="netbox_slm.SoftwareProductVersion", on_delete=models.PROTECT, null=True, blank=True)
|
|
installation = models.ForeignKey(
|
|
to="netbox_slm.SoftwareProductInstallation", on_delete=models.PROTECT, null=True, blank=True
|
|
)
|
|
|
|
objects = RestrictedQuerySet.as_manager()
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse("plugins:netbox_slm:softwarelicense", kwargs={"pk": self.pk})
|