7>hAzddlmZddlmZmZddlmZmZmZmZddl m Z m Z m Z m Z ddlZddlZddlmZmZmZddlmZej,eZded efd Zeej6d d eej6d dddeej6ddeej6ddddeej6ddeej6dd dddZdeded efdZdedej@ded efdZ!dededejDded ef d Z#deded e ee ffd!Z$ d/dede ed"e ed#e ed e e ee ff d$Z%deded efd%Z&ded e ee ffd&Z'ded'e ed(e ee fded e ee ff d)Z( d0ded*ed+e ed e e ee ffd,Z)ded-ed e e ee ffd.Z*y)1)Session) HTTPExceptionstatus)datetimetime timedeltadate)ListDictAnyOptionalN)AdvisorAdvisorScheduleUser)schemastime_strreturnc^tt|jd\}}t||S)z0Parse time string in format HH:MM to time object:)mapintsplitr)rhoursminutess O/var/www/html/DP/alpha_backend/app/advisor_service/services/schedule_service.py parse_timer s(hnnS12NE7 w  SHIFT_1_STARTz09:00 SHIFT_1_ENDz17:00z Morning Shift)startendname SHIFT_2_START SHIFT_2_ENDz01:00z Evening Shift SHIFT_3_START SHIFT_3_ENDz Night Shift)SHIFT_1SHIFT_2SHIFT_3 advisor_iddbcH tj}|j}|j}|j t j t j|k(t j|kt j|k\j}|sytj|j}|sy|d}|d}||kDr ||k\xs||kS||cxkxr|kScS#t$r+} tj!dt#| Yd} ~ yd} ~ wwxYw)z Check if an advisor is currently in their scheduled shift Args: advisor_id: ID of the advisor to check db: Database session Returns: bool: True if advisor is in shift, False otherwise Fr r!z'Error checking if advisor is in shift: N)rnowrr queryrfilterr*shift_period_startshift_period_endfirstSHIFTSgetshift Exceptionloggererrorstr) r*r+r- current_time current_dateschedule shift_info shift_start shift_endes ris_advisor_in_shiftrA$s llnxxz xxz 88O,33  & &* 4  . ., >  , , <  %'  ZZ/  ) u%   ";.K,)2K K,;); ;; ;  >s1vhGHs0BC- !C-C- C-*C-- D!6!DD! schedule_data manager_idc$|jtjtj|jk(j }|s t dd|jtjtj|jk(tj|jktj|jk\j }|r!|j||jt|j||j|j|j|j}|j||j|j|t!||j|S)a( Create a new schedule for an advisor If there's an overlapping schedule, it will be overridden Args: db: Database session schedule_data: Schedule data manager_id: ID of the manager creating the schedule Returns: Created schedule zAdvisor not found status_codedetail)r*shift_setup_byr0r1r5advisor_status)r.rr/idr*r2rrr0r1deletecommitr5rJaddrefreshupdate_advisor_status)r+rBrCadvisoroverlapping_schedule db_schedules rcreate_advisor_schedulerTQs;$hhw&&wzz]5M5M'MNTTVG 4GHH88O4;;""m&>&>>**m.L.LL((M,L,LL eg  &' " ++!(;;&77!!$33 KFF;IIKJJ{"m667 r schedule_idc@|jtjtj|k(j }|s t dd|j |j |_|j|j|_|j|j|_|j|j|_ ||_ tj|_ |j|j|t!||j"|S)a# Update an existing advisor schedule Args: db: Database session schedule_id: ID of the schedule to update schedule_data: Updated schedule data manager_id: ID of the manager updating the schedule Returns: Updated schedule rESchedule not foundrF)r.rr/rKr2rr0r1r5rJrIrutcnow updated_onrMrOrPr*)r+rUrBrCrSs rupdate_advisor_schedulerZs&((?+22?3E3E3TU[[]K 4HII''3)6)I)I &%%1'4'E'E $&)// ##/%2%A%A "",K%__.KIIKJJ{"k445 rc |jtjtj|k(j }|s t dd|j }|j||jt||ddiS)z Delete an advisor schedule Args: db: Database session schedule_id: ID of the schedule to delete Returns: Success message rErWrFmessagezSchedule deleted successfully) r.rr/rKr2rr*rLrMrP)r+rUrSr*s rdelete_advisor_scheduler]s|((?+22?3E3E3TU[[]K 4HII''JIIkIIK"j) 6 77r start_dateend_datec|jtjttjtj k(dj tjjdtjjd}|r"|jtj|k(}|r"|jtj|k\}|r"|jtj|k}|jtj}g}|jD]}|d}|j}|j } |j#|j |j|| |j|j|j$|j&|j(|j*d |S)af Get advisor schedules with optional filtering, including advisor details Args: db: Database session advisor_id: Optional advisor ID to filter by start_date: Optional start date to filter by end_date: Optional end date to filter by Returns: List of advisor schedules with advisor details T)isouter advisor_name advisor_typer) rKr*rbrcr0r1rJr5 created_atrY)r.rjoinrr*rK add_columns full_namelabeltyper/r1r0order_byallrbrcappendrJr5rdrY) r+r*r^r_r.resultsrowr<rbrcs rget_advisor_schedulesrosr& !    & &'** 4      # #N 3 LL  ~ .   _77:EF _==KL _??8KL NN?== >EGyy{q6'' '' ++"--(("*"="= ( 9 9&55^^"--"--  $ Nrc t||}|jtjtj|k(j }|st jd|dy|r|jtjtj|k(tjtjjktjtjjk\j }|r|jrd|_nd|_nd|_|j"r`|jt$jt$j|j"k(j }|r|j |_|j'|S#t($r;}t jdt+||j-Yd}~yd}~wwxYw)z Update an advisor's status based on their current schedule Args: db: Database session advisor_id: ID of the advisor to update Returns: bool: True if advisor is in shift, False otherwise zAdvisor with ID z not foundF10zError updating advisor status: N)rAr.rr/rKr2r7r8rr*r0rr-r r1rJruser_idrrMr6r9rollback)r+r*in_shiftrQr<userr@s rrPrPsw&&z26((7#**7::+CDJJL LL+J((GOO)CDJJLD%nn    6s1vh?@ sA*F9-E F99 G=1G88G=cx |jtj}d}d}d}|D]*}t||j}|dz }|r|dz }&|dz },d|||dS#t $rO}t jdt||jtddt|d }~wwxYw) z Refresh the status of all advisors based on their schedules Args: db: Database session Returns: Summary of updates rzAll advisor statuses refreshed)r\ total_updatedonlineofflinez'Error refreshing all advisor statuses: iz#Error refreshing advisor statuses: rFN) r.rrkrPrKr6r7r8r9rtr)r+advisors updated_count online_count offline_countrQrur@s rrefresh_all_advisor_statusesrFs 88G$((*   G,Rs1vhGH 8QA   sAA!! B9*A B44B9 advisor_ids base_schedulec"ggd}|D]N} i|d|i}tjdi|}t|||}|dj||jdP|S#t $r*} |dj|t | dYd} ~ d} ~ wwxYw) aD Create schedules for multiple advisors at once Args: db: Database session advisor_ids: List of advisor IDs base_schedule: Base schedule data (without advisor_id) manager_id: ID of the manager creating the schedules Returns: Summary of created schedules )successfailedr*r)r*rUr)r*r8N)rAdvisorScheduleCreaterTrlrKr6r9) r+rrrCrmr*rBschedule_creater<r@s rcreate_bulk_advisor_schedulesrps&G "  G}GlJGM%;;LmLO/r?JOH I  % %('{{' ", N  H  $ $(Q&   sA A B$ B  Br5 check_datec @|"tjj}tj|tjj }|j tjtj|k(tj|ktj|k\tjdk(j}g}|D]}|j tjtj|j k(j#}|sS|j%|j|j&|j(|j*|j,|j|j|j|jd |S)a( Get all advisors scheduled for a specific shift on a given date Args: db: Database session shift: Shift identifier (e.g., "SHIFT_1") check_date: Date to check (defaults to today) Returns: List of advisors with their schedule details T) r*rgemail phone_numberrrUr5r0r1)rr-r combineminrr.rr/r5r0r1rJrkrrKr*r2rlrgrrr)r+r5rcheck_datetime schedulesresultr<rQs rget_advisors_by_shiftrsH"\\^((* %%j(,,2C2C2EFN)00&**n<((N:&&$.   ce F((7#** JJ(-- - %'   MM%jj$..  ' 4 4!..'{{!&.&A&A$,$=$=   $ Mrrc 6|j}|j}d}tjD]5\}}|d}|d}||kr||k\s||ks |}n||cxkr|ks0n3|}n|sgS|j t j t j|k(t j|kt j|k\t jdk(j} g} | D]} |j tj tj| jk(j} | sS| j!| j| j"| j$| j&| j| jd| S)z Get advisors available at a specific date and time Args: db: Database session check_datetime: Datetime to check availability Returns: List of advisors with their IDs and statuses Nr r!T)r*rgrcurrent_statusr5rU)r rr3itemsr.rr/r5r0r1rJrkrrKr*r2rlrgrr) r+rr check_time target_shift shift_keyr=r>r?rrr<rQs r"get_available_advisors_at_datetimers $$&J$$&JL!' : ) u%  { "[(J),C( j5I5( "0  )00-**j8((J6&&$.   ce F((7#** JJ(-- - %'   MM%jj$.. ")..!'{{    Mr)NNN)N)+sqlalchemy.ormrfastapirrrrrr typingr r r r osloggingapp.common.modelsrrr app.commonr getLogger__name__r7r9rgetenvr3rboolrArrTAdvisorScheduleUpdaterZr]rorPrrrrrrrrs")44,, <<   8 $   IBIIow?@)"))M7;< IBIIow?@)"))M7;< IBIIow?@)"))M7;< $+C+W++Z330033 3j...00. .  .`88c8d38n8:!%%)#' @@ @"@x @  $sCx. @D1g13141f( W( c3h( T--c-S>- -  #s(^ -d"&33 33 $sCx. 3j@@@ $sCx.@r