22import logging
33
44from celery .exceptions import TimeoutError
5- from celery .result import AsyncResult
5+ # from celery.result import AsyncResult
66from celery .states import FAILURE , PENDING , SUCCESS
77from django .contrib .auth .decorators import login_required
88from django .core import serializers
2626from app .worker .tasks import receiptor
2727from app .worker .tasks .exporter import exporter
2828from app .worker .tasks .importers import historical_data_importer
29+ from reboot .celery import app
2930
3031logger = logging .getLogger (__name__ )
3132
@@ -121,8 +122,9 @@ def poll_state(request: HttpRequest):
121122 request = request ,
122123 err_msg = "The task_id query parameter of the request was omitted." )
123124
124- task = AsyncResult (task_id )
125+ task = app . AsyncResult (task_id )
125126 res = JsonResponse (_poll_state (PENDING , 0 , 200 ))
127+ print (f"!!! task id={ task_id } ,state={ task .state } ,successful={ task .successful ()} ,ready={ task .ready ()} ,failed={ task .failed ()} " )
126128 if task .state == FAILURE or task .failed ():
127129 res = JsonResponse (_poll_state (FAILURE , 0 , 400 ))
128130 elif task .state == PROGRESS :
@@ -146,8 +148,8 @@ def download_file(request: HttpRequest):
146148 while (attempts < ATTEMPT_LIMIT ):
147149 try :
148150 attempts += 1
149- task = AsyncResult (task_id )
150- result = task .get (timeout = 0.5 * attempts )
151+ task = app . AsyncResult (task_id )
152+ result = task .get (timeout = 1.0 * attempts )
151153 print (f"{ task } { task_name } success #{ attempts } : { result } " )
152154 break
153155 except TimeoutError :
0 commit comments