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

11432 DRY

Browse Source
This commit is contained in:
Arthur 2023-04-07 11:14:50 -07:00
parent 36d3c84803
commit a72349448d
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/dcim/api/serializers.py
View File

@ -32,24 +32,28 @@ from .nested_serializers import *
class BaseReadonlyDeviceMixin: class BaseReadonlyDeviceMixin:
field_name = 'device'
def get_fields(self, *args, **kwargs): def get_fields(self, *args, **kwargs):
fields = super().get_fields(*args, **kwargs) fields = super().get_fields(*args, **kwargs)
# get_extra_kwargs doesn't work if field explicitly declared on serializer... # get_extra_kwargs doesn't work if field explicitly declared on serializer...
if (self.instance): if (self.instance):
fields[self.field_name].read_only = True fields[self.get_field_name()].read_only = True
return fields return fields
class ReadonlyDeviceMixin(BaseReadonlyDeviceMixin): class ReadonlyDeviceMixin(BaseReadonlyDeviceMixin):
field_name = 'device'
# can't be a field as mucks up serializer
def get_field_name(self):
return "device"
class ReadonlyDeviceTypeMixin(BaseReadonlyDeviceMixin): class ReadonlyDeviceTypeMixin(BaseReadonlyDeviceMixin):
field_name = 'device_type'
def get_field_name(self):
return "device_type"
class CabledObjectSerializer(serializers.ModelSerializer): class CabledObjectSerializer(serializers.ModelSerializer):