di!dZddlmZddlmZddlmZmZddlmZddl m Z ejdd  d d d Z y )z Celery task for running monthly demand prediction job for all product/location combinations. Uses the generate_monthly_forecasts function from controller to generate and save forecasts. ) annotations)Dict)datetimedate) celery_app) dispatch_taskTz^src.smart_inventory.tasks.monthly_demand_prediction_job_task.run_monthly_demand_prediction_job)bindnamec6 ddlm}ddlm}m}m}ddlm}ddlm }ddl m } t|} d} |jj} td| d | | j!|j#|j$| k(j'} td | | rF|j(| _t-j.| _| j3td ntd | | j!||j4|j6j#|j8|k(j;|j4|j6j=} | sb| rD|j>| _d || _ t-j.| _!| j3dd ||dddd| jEStGjH}tG|jJ|jLd}|| |dz z}tOd| D}tOd| D}tQ| }d}d}g}td|d|tdtQ|dtQ|td|d|d|d| D]3\}} || ||||||dz }|dzdk(rtd|d|d 5| r:|jX| _t-j.| _!| j3 tGjHj[d%}td&|t]d'|f(}|j}||d)|j^|t-j.*}| ja|| j3td+|d-d.||||j[d/|j[d/tQ|tQ|||||r|dd0ngd1 }td2|d3|d4|| jES#tR$rC}|dz }d!|d"|d#tU|}|jW|td$|Yd}~d}~wwxYw#tR$r}td,|Yd}~d}~wwxYw#tR$r}tU|} | s;| j!|j#|j$| k(j'} | rD|j>| _| dd5| _ t-j.| _!| j3n$#tR$r}!td6|!Yd}!~!nd}!~!wwxYwtd7| dd8| || d9cYd}~| jESd}~wwxYw#| jEwxYw):a Run monthly demand prediction for all product/location combinations for a company. This task: - Finds all unique product_id and location_id combinations for the company - Generates monthly forecasts for each combination - Saves predictions to the MonthlyForecast table Args: company_id: Company ID to run predictions for months_ahead: Number of months to predict ahead (default: 6) Returns: Dict with success status, counts, and any errors r)get_db)CeleryTaskTrackerCeleryTaskStatus DailySales)generate_monthly_forecasts)distinct) relativedeltaNz>[CELERY TASK] Starting monthly demand prediction job with ID: z for company_id: z"[CELERY TASK] Found task_tracker: z'[CELERY TASK] Updated status to STARTEDz:[CELERY TASK] WARNING: No task_tracker found for task_id: z3No product/location combinations found for company F)successmessage company_idtotal_combinationssuccessful_predictionsfailed_predictions)monthsc3&K|] }|d yw)rN.0combos [/var/www/html/hubwallet-dev/src/smart_inventory/tasks/monthly_demand_prediction_job_task.py z4run_monthly_demand_prediction_job..es Deq Dc3&K|] }|d yw)rNrrs r r!z4run_monthly_demand_prediction_job..fs!Eu%(!Er"z[CELERY TASK] Processing z combinations for company z[CELERY TASK] Unique products: z, Unique locations: z[CELERY TASK] Predicting from z to z (z months))dbr location_id product_id start_month end_monthdz[CELERY TASK] Processed /z combinationsz product_id=z, location_id=z: z[CELERY TASK] Error: z%Y-%m-%dz9[CELERY TASK] Triggering inventory planning snapshot for zUsrc.smart_inventory.tasks.inventory_planning_task.compute_inventory_planning_snapshot)args#compute_inventory_planning_snapshot)task_id task_namestatusr created_atzB[CELERY TASK] Inventory planning snapshot task triggered with ID: z<[CELERY TASK] Error triggering inventory planning snapshot: Tz4Monthly demand prediction job completed for company z%Y-%m ) rrr months_aheadr'r(unique_product_idsunique_location_idsrrrerrorsz[CELERY TASK] Job completed: z successful, z failediz+[CELERY TASK] Error updating task tracker: z%[CELERY TASK] Job failed with error: z&Monthly demand prediction job failed: )rrrerror)1 src.utils.dbr )src.smart_inventory.apps.inventory.modelsr rr-src.smart_inventory.apps.inventory.controllerr sqlalchemyrdateutil.relativedeltarnextrequestidprintqueryfilterr-firstSTARTEDr/rnow started_atcommitr&r%rgroup_byallFAILURE error_message completed_atclosertodayyearmonthsetlen ExceptionstrappendSUCCESSstrftimerPENDINGadd)"selfrr2r r rrrrrr$ task_trackerr- combinationsrMr'r(r3r4rrrr5r&r%e error_msg today_str planning_taskplanning_task_idplanning_trackerresultrJdb_errors" r !run_monthly_demand_prediction_jobrd s*$ Y#4 fhBLllooG J7)Sdeodp qrbxx 1299  % % 0 %'  2<.AB "2":":L &.llnL # IIK ; = NwiX Yxx Z** +  " "  &  ! !Z / (  ! !  " "  #% &6&>&> #/bcmbn-o *,4LLN ) !PQ[P\](&'*+&' ~  k 5::u{{A6 -|a7G"HH ! D| DD!!E !EE .!" )*<)==WXbWcde /4F0G/HH\]`at]u\vwx .{m4 {"\NZbcd'3  #J  *) +) +' '!+&)C/1445K4LAN`Maanop 0 "2":":L (0 L % IIK V --j9I Mi[Y Z*g\M -//  1(?'//%#<<>    FF# $ IIK VWgVhi j Mj\Z$(&//8"++G4"%&8"9#&':#;"4&<"4%+fSbk   -.D-E]SeRffmno8  i "a'")*^K=PRSVWXSYRZ[  i(-i[9:  J V PQRPST U U V*  A  L!xx(9:AA%--8 %'&6&>&> #-:5D-A *,4LLN )  L ?zJ K K L 5m_EF? O$"     5 4  sE,RC R%-P!>RB#Q04AR! Q-*8Q("R(Q--R0 R9R R RR V U>)BT-,U>- U6U U> UU>(V)V>VVVN))rintr2rfreturnr) __doc__ __future__rtypingrrrsrc.utils.celery_workerr)src.smart_inventory.utils.celery_dispatchrtaskrdrr rosj##.Cd"BCGGG GCGrn