NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-24 08:25:17 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

9654 add weight fields to devices

Browse Source
This commit is contained in:
Arthur 2022-09-23 08:12:16 -07:00
parent e21f4854b8
commit d0bed830d7
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
netbox/dcim/models/mixins.py
View File

@ -1,3 +1,4 @@
from django.core.exceptions import ValidationError
from django.db import models from django.db import models
from dcim.choices import * from dcim.choices import *
from utilities.utils import to_kilograms from utilities.utils import to_kilograms
@ -35,3 +36,12 @@ class DeviceWeightMixin(models.Model):
self._abs_weight = None self._abs_weight = None
super().save(*args, **kwargs) super().save(*args, **kwargs)
def clean(self):
super().clean()
# Validate weight and weight_unit
if self.weight is not None and not self.weight_unit:
raise ValidationError("Must specify a unit when setting a weight")
elif self.weight is None:
self.weight_unit = ''