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

1from ipam.models import Role 

2from netbox.api.fields import RelatedObjectCountField 

3from netbox.api.serializers import OrganizationalModelSerializer 

4 

5__all__ = ( 

6 'RoleSerializer', 

7) 

8 

9 

10class RoleSerializer(OrganizationalModelSerializer): 

11 

12 # Related object counts 

13 prefix_count = RelatedObjectCountField('prefixes') 

14 vlan_count = RelatedObjectCountField('vlans') 

15 asn_count = RelatedObjectCountField('asns') 

16 

17 class Meta: 

18 model = Role 

19 fields = [ 

20 'id', 'url', 'display_url', 'display', 'name', 'slug', 'weight', 'description', 'owner', 'comments', 'tags', 

21 'custom_fields', 'created', 'last_updated', 'prefix_count', 'vlan_count', 'asn_count', 

22 ] 

23 brief_fields = ('id', 'url', 'display', 'name', 'slug', 'description', 'prefix_count', 'vlan_count', 

24 'asn_count')