08i< .dZddlZddlZddlmZmZmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZddlmZej(j+ej(j-ej(j-ej(j-ej(j-edd lmZmZe Gd d Ze Gd d ZdZddeedefdZdedededefdZ ddeefdZ!ddeefdZ"eZ#y)a8 Smart Inventory - Daily Snapshot Task ==================================== Celery task for computing daily inventory snapshots. This task computes inventory levels by: 1. Taking previous day's snapshot as starting point 2. Applying all inventory movements for the current day 3. Computing final on-hand, inbound, outbound quantities 4. Storing results in inventory_snapshot_daily table Formula: -------- today_on_hand = previous_day_on_hand + sum(today_movements) today_inbound = sum(positive_movements_today) today_outbound = sum(abs(negative_movements_today)) N)datetimedate timedelta)OptionalDictListTuple) dataclass)Session)textand_)IntegrityError) SessionLocalenginec<eZdZUdZdZeed<dZeed<dZeed<y)MovementSummaryz&Summary of movements for a combination inbound_qty outbound_qty net_movementN) __name__ __module__ __qualname____doc__rfloat__annotations__rrP/var/www/html/hubwallet-dev/src/smart_inventory/tasks/inventory_snapshot_task.pyrr$s#0KL%L%rrcleZdZUdZdZeed<dZeed<dZeed<dZ e ed<dZ e ed <d Z eeed <y ) SnapshotComputationResultzResult of snapshot computationrrecords_processedrecords_createdrecords_updatedrcomputation_timeFsuccessN error_message)rrrrr"intrr#r$r%rr&boolr'rstrrrrr!r!+sG(sOSOS!e!GT#'M8C='rr!c ddlm}|jddd dttdt fd }|S#t $r }Yd}~yd}~wwxYw) z.compute_inventory_snapshot_task:s24E Err2)src.utils.celery_workerr,taskrr*r ImportError)r,r5es rcreate_celery_taskr:5s^ 6 bim n Fx} FX\ F o F/. s58 A A r/r0ctj}|rt|dijddnd}t d|d|r& tj |dj }n!t jtd z }|td z }t d|d|t d|d|t} t|||}tj|z j} | |_ |jrt d|d| ddt d|jt d|j t d|j"d|j|j |j"| |j%d|j'St d|d|j(d |j(| d|j'S#t$rd|d}t d|d |d |d cYSwxYw#t*$r>} dt-| }t d|d |d |d cYd} ~ |j'Sd} ~ wwxYw#|j'wxYw)a3 Celery task to compute daily inventory snapshots for a given date. Args: target_date: Date string in YYYY-MM-DD format. If None, uses yesterday. Returns: Dict with computation results and statistics Raises: Exception: If computation fails requestidmanualz[Task z,] Starting inventory snapshot computation...%Y-%m-%dzInvalid date format: . Expected YYYY-MM-DDz ] ERROR: F)r&errordaysz] Target Date: z] Previous Date: z] Computation completed in .2fs Records processed:  Records created:  Records updated: T)r&r"r#r$r%r/z] Computation failed: )r&rAr%z.Unexpected error during snapshot computation: N)rnowgetattrgetprintstrptimer ValueErrortodayrr_compute_snapshots_for_date total_secondsr%r&r"r#r$ isoformatcloser' Exceptionr*) r4r/ start_timetask_idtarget_date_obj error_msg previous_datedbresultr%r9s rr3r3CstJBFgdIr*..tX>HG F7)G HI :&// ZHMMOO **,)::#iQ&77M F7)??*; <= F7),]O <= B",R-P$LLNZ7FFH"2 >> F7)#>?OPS>TTUV W *6+C+C*DE F ()?)?(@A B ()?)?(@A B %+%=%=#)#9#9#)#9#9$4.88: ,   F7)#9&:N:N9OP Q --$4  a :/ }Q>QRC$X%9%9:" #"M#   }  " 4]OCA B$&88,=#>#E#E +667+E$ce ! F33455RS^R_` a1 T**H,@,@(BUBUVm+'*'*(+*M#&c"=1U8;R;R5SS1 T *3}+=*>>ij k M*++ABC &{m378HH./66 II'22 3{ B #%  s9~&ok]CD ! (H&&(<(Q>QRC,,*9*;"3'(-G//0CQw##s*#$$C0$  C '  (""89@@ " 0 0K ? %'  A  In--Ek]Rcd e HH+ , 3 3&44 C fh IIK||~ &8%=%=%? [ !C!25 /J Z%(("" #/H#=14D4Q4QQK+)%'%',88-::' F  $ $V , OJ Direct test of core logic without any Celery wrapper z!Running direct core logic test...r?zERROR: Invalid date format: r@NrBrCz Target Date: zPrevious Date: rrErFrGrHrIrzERROR: Unexpected error: )rMrrNrrOrPrrrJrQrRr&r"r#r$r'rUrT)r/rXrZr[rVr\r%r9s rmanual_test_core_logicras -. &// ZHMMOO **,)::#iQ&77M M/* +, OM? +, B\\^ ,R-P$LLNZ7FFH >> 67G6LAN O *6+C+C*DE F ()?)?(@A B ()?)?(@A B /0D0D/EF G  =  0 =RS T  4 / )!-../  s<$E CE=E:9E:= FFF!FF!!F3r2)$rossysrrrtypingrrrr dataclassesr sqlalchemy.ormr rir r sqlalchemy.excrpathrtdirname__file__ src.utils.dbrrrr!r:r*r3rQrrcompute_inventory_snapshotrrrrs$& ....!"!)PX@Y0Z [\]-    (( ( Gx}GPTGZvvvv vxY8C=Y*( (X01r