Coverage for netbox/api/features.py: 91%

1from django.shortcuts import get_object_or_404 

2from rest_framework.decorators import action 

3from rest_framework.response import Response 

4 

5from netbox.api.authentication import TokenSyncPermission 

6 

7__all__ = ( 

8 'SyncedDataMixin', 

9) 

10 

11 

12class SyncedDataMixin: 

13 

14 def get_permissions(self): 

15 if self.action == 'sync': 

16 return [TokenSyncPermission()] 

17 return super().get_permissions() 

18 

19 @action(detail=True, methods=['post']) 

20 def sync(self, request, pk): 

21 """ 

22 Provide a /sync API endpoint to synchronize an object's data from its associated DataFile (if any). 

23 """ 

24 # Rebuild from the default manager: initial() has narrowed self.queryset to the add permission, 

25 # which does not govern this action. 

26 queryset = self.queryset.model.objects.restrict(request.user, 'sync') 

27 obj = get_object_or_404(queryset, pk=pk) 

28 if obj.data_file: 28 ↛ 29line 28 didn't jump to line 29 because the condition on line 28 was never true

29 obj.sync(save=True) 

30 serializer = self.serializer_class(obj, context={'request': request}) 

31 

32 return Response(serializer.data)