2424from barbequeue .common .classes import State
2525
2626from .permissions import IsDeviceOwnerOnly
27- from .apps import client
28-
27+ from .client import get_client
2928
3029logging = logger .getLogger (__name__ )
3130
@@ -37,15 +36,15 @@ class TasksViewSet(viewsets.ViewSet):
3736 permission_classes = (IsDeviceOwnerOnly ,)
3837
3938 def list (self , request ):
40- jobs_response = [_job_to_response (j ) for j in client .all_jobs ()]
39+ jobs_response = [_job_to_response (j ) for j in get_client () .all_jobs ()]
4140 return Response (jobs_response )
4241
4342 def create (self , request ):
4443 # unimplemented. Call out to the task-specific APIs for now.
4544 pass
4645
4746 def retrieve (self , request , pk = None ):
48- task = _job_to_response (client .status (pk ))
47+ task = _job_to_response (get_client () .status (pk ))
4948 return Response (task )
5049
5150 def destroy (self , request , pk = None ):
@@ -73,11 +72,11 @@ def startremoteimport(self, request):
7372 _ ("The requested channel does not exist on the content server" )
7473 )
7574
76- task_id = client .schedule (
75+ task_id = get_client () .schedule (
7776 _networkimport , channel_id , track_progress = True )
7877
7978 # attempt to get the created Task, otherwise return pending status
80- resp = _job_to_response (client .status (task_id ))
79+ resp = _job_to_response (get_client () .status (task_id ))
8180
8281 return Response (resp )
8382
@@ -92,11 +91,11 @@ def startlocalimport(self, request):
9291 raise serializers .ValidationError (
9392 "The 'drive_id' field is required." )
9493
95- job_id = client .schedule (
94+ job_id = get_client () .schedule (
9695 _localimport , request .data ['drive_id' ], track_progress = True )
9796
9897 # attempt to get the created Task, otherwise return pending status
99- resp = _job_to_response (client .status (job_id ))
98+ resp = _job_to_response (get_client () .status (job_id ))
10099
101100 return Response (resp )
102101
@@ -111,11 +110,11 @@ def startlocalexport(self, request):
111110 raise serializers .ValidationError (
112111 "The 'drive_id' field is required." )
113112
114- job_id = client .schedule (
113+ job_id = get_client () .schedule (
115114 _localexport , request .data ['drive_id' ], track_progress = True )
116115
117116 # attempt to get the created Task, otherwise return pending status
118- resp = _job_to_response (client .status (job_id ))
117+ resp = _job_to_response (get_client () .status (job_id ))
119118
120119 return Response (resp )
121120
@@ -129,7 +128,7 @@ def cleartask(self, request):
129128 raise serializers .ValidationError (
130129 "The 'task_id' field is required." )
131130
132- client .clear (force = True )
131+ get_client () .clear (force = True )
133132 return Response ({})
134133
135134 @list_route (methods = ['get' ])
0 commit comments