koidZddlZddlZddlZddlmZddlmZeje Z ejdd ddee d ee d e d e d e d eede de fdZejddddee fdZy)z Hold Task Poller ================ Celery task that polls for HOLD tasks and starts them when dependencies are met. Uses non-blocking self-rescheduling to avoid blocking the worker. N)Optional) celery_appT:src.smart_inventory.tasks.hold_task_poller.poll_hold_tasks)bindname company_idbatch_idmax_duration_secondspoll_interval_secondsmain_poll_interval_seconds _start_time_total_started _iterationsc   ddlm} ddlm} m} ddlm} m} |tj}tj|z }||k\r!tjd |d d |||d d S| } | ||}| ||}|s/|s-tjdd |||dd |jSd}|r || |||z }|r || ||z }||z }|dz }|dkDrtjd|d|dzdk(rtjd|dd|d|j|||||||d}t"j$dk(rt&j)||nt+j,d||d |||dd S#t$r0}tjd|dt|dcYd}~Sd}~wwxYw#t $r"}tjd|Yd}~d}~wwxYw#|jwxYw)a Poll for HOLD tasks and start them when dependencies are met. This task handles BOTH: - Snapshot tasks (daily_sales, inventory_snapshot, etc.) - CSV processing tasks (product, purchase_order, sales_order, etc.) This task uses non-blocking self-rescheduling: - Checks if HOLD tasks can be started - Reschedules itself with countdown to avoid blocking the worker - Stops when no HOLD tasks remain or max duration is reached Args: company_id: Optional filter by company batch_id: Optional filter by batch max_duration_seconds: Maximum time to poll (default 5 minutes) poll_interval_seconds: How often to check for HOLD tasks (default 2 seconds) rget_db_sessionprocess_hold_taskshas_hold_tasksprocess_hold_csv_taskshas_hold_csv_tasks#Failed to import required modules: FsuccesserrorNzHold poller timeout (zs)Ttimeout)r total_startedduration_seconds iterationsreasonz)No HOLD tasks remaining (snapshot or CSV) no_hold_taskszStarted z HOLD tasks (snapshot + CSV) z Hold poller: z.0fzs, z startedzError in hold task poller: )rr r r r rrwin32)kwargs countdownr rescheduled) src.utils.dbr+src.smart_inventory.utils.task_orchestratorrr/src.smart_inventory.utils.csv_task_orchestratorrr ImportErrorloggerrstrtimedebugcloseinfo Exceptionsysplatformpoll_hold_tasks apply_asyncr send_task)selfrr r r r r rrrrrrreelapseddbhas_snapshot_hold has_csv_holdstarted task_kwargss I/var/www/html/hubwallet-dev/src/smart_inventory/tasks/hold_task_poller.pyr6r6sE<3/bn iik iikK'G&& ,-A,B"EF+ '%     B&*2z:)"j9   LLD E!/$+)) >  -  )"j(C CG  -b*= =G'!q Q; KK(7)+GH I  q LL= S8HQ R  ! 4!6"("K ||w##;BW#X  H+ '#!  m 3 :1#>? 3q6223t 8 21#6778  sNF32G6A,G G%F;5G;G G. G)$G1)G..G11HzGsrc.smart_inventory.tasks.hold_task_poller.check_and_process_hold_tasksc@ ddlm}ddlm}m}ddlm}m}|} |||} |||} | s| sd d dd |jSd} | r | |||z } | r | |||z } d | d | d |jS#t$r0}tjd|dt|dcYd}~Sd}~wwxYw#t$r@}tjd|dt|dcYd}~|jSd}~wwxYw#|jwxYw)z One-time check to process any HOLD tasks (both snapshot and CSV). Use this for manual triggering or scheduled checks. Args: company_id: Optional filter by company rrrrrFrNTzNo HOLD tasks to process)rmessager?z/Processed HOLD tasks (snapshot + CSV), started )rr?rCzError checking HOLD tasks: )r)rr*rrr+rrr,r-rr.r1r3) r9rrrrrrr:r<r=r>r?s rAcheck_and_process_hold_tasksrDs,3/bn  B*2z:)"j9  #0JWXY.  +  )"j9 9G  -b*= =GH R   E 3 :1#>? 3q6223: 3 21#67 3q622  3  sRBB? &B? B< %B71B<7B<? D%D-D.D DD D)NNi,Nrr)N)__doc__loggingr4r/typingrsrc.utils.celery_workerr getLogger__name__r-taskintr.floatr6rDrArRs .   8 $d!]^!%" #!"&(#'} }sm} }  } !$ }%}}}_}@d!jk.8C=.l.rQ