Coverage for dcim/api/serializers_/roles.py: 100%

1from rest_framework import serializers 

2 

3from dcim.models import DeviceRole, InventoryItemRole 

4from extras.api.serializers_.configtemplates import ConfigTemplateSerializer 

5from netbox.api.fields import RelatedObjectCountField 

6from netbox.api.serializers import NestedGroupModelSerializer, OrganizationalModelSerializer 

7 

8from .nested import NestedDeviceRoleSerializer 

9 

10__all__ = ( 

11 'DeviceRoleSerializer', 

12 'InventoryItemRoleSerializer', 

13) 

14 

15 

16class DeviceRoleSerializer(NestedGroupModelSerializer): 

17 parent = NestedDeviceRoleSerializer(required=False, allow_null=True, default=None) 

18 config_template = ConfigTemplateSerializer(nested=True, required=False, allow_null=True, default=None) 

19 device_count = serializers.IntegerField(read_only=True, default=0) 

20 virtualmachine_count = serializers.IntegerField(read_only=True, default=0) 

21 

22 class Meta: 

23 model = DeviceRole 

24 fields = [ 

25 'id', 'url', 'display_url', 'display', 'name', 'slug', 'color', 'vm_role', 'config_template', 'parent', 

26 'description', 'tags', 'custom_fields', 'created', 'last_updated', 'device_count', 'virtualmachine_count', 

27 'owner', 'comments', '_depth', 

28 ] 

29 brief_fields = ( 

30 'id', 'url', 'display', 'name', 'slug', 'description', 'device_count', 'virtualmachine_count', '_depth' 

31 ) 

32 

33 

34class InventoryItemRoleSerializer(OrganizationalModelSerializer): 

35 

36 # Related object counts 

37 inventoryitem_count = RelatedObjectCountField('inventory_items') 

38 

39 class Meta: 

40 model = InventoryItemRole 

41 fields = [ 

42 'id', 'url', 'display_url', 'display', 'name', 'slug', 'color', 'description', 'owner', 'comments', 'tags', 

43 'custom_fields', 'created', 'last_updated', 'inventoryitem_count', 

44 ] 

45 brief_fields = ('id', 'url', 'display', 'name', 'slug', 'description', 'inventoryitem_count')