a #e} @sddlZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl m Z ddlmZGd d d eZGd d d eZGd ddeZdS)N)TestSafeAsyncResult)APIView)Response)status permissions) FileResponse) HttpResponse) FileWrapperc@s.eZdZejfZdddZddZddZdS) CheckTaskNcKst|}|s"tdditjdS|}|dddurR||}td|dS||durx||}td|dStddiSdS)NreadyF)rerrorT)r r )rr rr HTTP_200_OKgeton_error error_check)selfrequestcelery_task_idkwargsresresultmsgr/webodm/app/api/workers.pyrs  z CheckTask.getcCs|dS)Nr rrrrrrr"szCheckTask.on_errorcCsdSNrrrrrr%szCheckTask.error_check)N) __name__ __module__ __qualname__rAllowAnypermission_classesrrrrrrrr s r c@s eZdZdS)TaskResultOutputErrorN)rrrrrrrr!(sr!c@s&eZdZejfZdddZddZdS) GetTaskResultNc KsLt|}|r2|}|dd}|dd}n tddiS|dur|jdtj|}t|j } t |d} | dk} | rt | } nt t | t|dpd d } t|dpd | d <d || d <| | d<| S|durs