230 lines
5.0 KiB
Python
230 lines
5.0 KiB
Python
from django.apps import apps
|
|
from django.conf import settings
|
|
from django.db import models
|
|
|
|
from rest_framework.reverse import reverse
|
|
|
|
from accounting.models.asset_base import AssetBase
|
|
|
|
|
|
|
|
class ITAMAssetBase(
|
|
AssetBase,
|
|
):
|
|
"""IT Asset Base Model
|
|
|
|
This model forms the base of ALL IT asset models and contains the core
|
|
features for all sub-models.
|
|
|
|
**Don't** use this model directly, it should be used via a sub-model.
|
|
"""
|
|
|
|
_is_submodel = True
|
|
|
|
app_namespace = None
|
|
|
|
model_tag = 'it_asset'
|
|
|
|
url_model_name = 'itamassetbase'
|
|
|
|
|
|
class Meta:
|
|
|
|
ordering = [
|
|
'id'
|
|
]
|
|
|
|
sub_model_type = 'it_asset'
|
|
|
|
itam_sub_model_type = 'itam_base'
|
|
|
|
verbose_name = "IT Asset"
|
|
|
|
verbose_name_plural = "IT Assets"
|
|
|
|
|
|
|
|
@property
|
|
def get_itam_model_type(self):
|
|
"""Fetch the ITAM Asset Type
|
|
|
|
Returns:
|
|
str: The models `Meta.m` in lowercase and without spaces
|
|
None: The ticket is for the Base class. Used to prevent creating a base ticket.
|
|
"""
|
|
|
|
model_type = str(self._meta.itam_sub_model_type).lower().replace(' ', '_')
|
|
|
|
if model_type == 'itam_base':
|
|
|
|
return None
|
|
|
|
return model_type
|
|
|
|
|
|
def get_itam_model_type_choices():
|
|
|
|
choices = []
|
|
|
|
if apps.ready:
|
|
|
|
all_models = apps.get_models()
|
|
|
|
for model in all_models:
|
|
|
|
if(
|
|
( isinstance(model, ITAMAssetBase) or issubclass(model, ITAMAssetBase) )
|
|
# and ITAMAssetBase._meta.itam_sub_model_type != 'itam_base'
|
|
|
|
):
|
|
|
|
choices += [ (model._meta.itam_sub_model_type, model._meta.verbose_name) ]
|
|
|
|
|
|
return choices
|
|
|
|
itam_type = models.CharField(
|
|
blank = True,
|
|
choices = get_itam_model_type_choices,
|
|
default = Meta.itam_sub_model_type,
|
|
help_text = 'IT Asset Type. (derived from IT asset model)',
|
|
max_length = 30,
|
|
null = False,
|
|
verbose_name = 'IT Asset Type',
|
|
)
|
|
|
|
|
|
|
|
page_layout: list = [
|
|
{
|
|
"name": "Details",
|
|
"slug": "details",
|
|
"sections": [
|
|
{
|
|
"layout": "double",
|
|
"left": [
|
|
'organization',
|
|
'asset_type',
|
|
'itam_type',
|
|
'asset_number',
|
|
'serial_number',
|
|
],
|
|
"right": [
|
|
'model_notes',
|
|
'created',
|
|
'modified',
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Knowledge Base",
|
|
"slug": "kb_articles",
|
|
"sections": [
|
|
{
|
|
"layout": "table",
|
|
"field": "knowledge_base",
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Tickets",
|
|
"slug": "tickets",
|
|
"sections": [
|
|
{
|
|
"layout": "table",
|
|
"field": "tickets",
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"name": "Notes",
|
|
"slug": "notes",
|
|
"sections": []
|
|
},
|
|
]
|
|
|
|
|
|
table_fields: list = [
|
|
'id',
|
|
{
|
|
"field": "display_name",
|
|
"type": "link",
|
|
"key": "_self"
|
|
},
|
|
'itam_type',
|
|
'asset_number',
|
|
'serial_number',
|
|
'organization',
|
|
'created'
|
|
]
|
|
|
|
|
|
def __str__(self):
|
|
|
|
return self._meta.verbose_name + ' - ' + self.asset_number
|
|
|
|
|
|
|
|
def clean_fields(self, exclude = None):
|
|
|
|
related_model = self.get_related_model()
|
|
|
|
if related_model is None:
|
|
|
|
related_model = self
|
|
|
|
if(
|
|
self.itam_type != str(related_model._meta.itam_sub_model_type).lower().replace(' ', '_')
|
|
and str(related_model._meta.sub_model_type).lower().replace(' ', '_') != 'itam_base'
|
|
):
|
|
|
|
self.itam_type = str(related_model._meta.itam_sub_model_type).lower().replace(' ', '_')
|
|
|
|
|
|
super().clean_fields(exclude = exclude)
|
|
|
|
|
|
|
|
def get_url(
|
|
self, relative: bool = False, api_version: int = 2, many = False, request: any = None
|
|
) -> str:
|
|
|
|
namespace = f'v{api_version}'
|
|
|
|
if self.get_app_namespace():
|
|
namespace = namespace + ':' + self.get_app_namespace()
|
|
|
|
|
|
url_basename = f'{namespace}:_api_{self._meta.model_name}'
|
|
|
|
if self.url_model_name:
|
|
|
|
url_basename = f'{namespace}:_api_{self.url_model_name}'
|
|
|
|
if (
|
|
self._is_submodel
|
|
and self._meta.sub_model_type != 'it_asset'
|
|
):
|
|
|
|
url_basename += '_sub'
|
|
|
|
|
|
if many:
|
|
|
|
url_basename += '-list'
|
|
|
|
else:
|
|
|
|
url_basename += '-detail'
|
|
|
|
|
|
url = reverse( viewname = url_basename, kwargs = self.get_url_kwargs( many = many ) )
|
|
|
|
if not relative:
|
|
|
|
url = settings.SITE_URL + url
|
|
|
|
|
|
return url
|