7>h= ddlmZmZmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZddlZddlmZddlmZddlmZmZd d lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%eZ&ejNe(Z)d e fd Z*GddeZ+e&jYdejZedeeee%fde e.e fded e fdZ/e&jYde e.e fedeeee%fde e.e fded e fdZ0e&jcdejZedeeee%fde2de e.e fded e fdZ3e&jideeee%fde2ded e fdZ5e&jYde e e.e fee6eeee%fde+ded e fd Z7e&jYd!eeee%fded e fd"Z8e&jYd#eeee%fd$e2ded e fd%Z9e&jud&ee%fd e fd'Z;e&jud(e e e.e fedd)*eeee%fd+e.d,e eded e fd-Z"e&jYd.e e e.e fedeeee%fd/e e.e fded e fd0Zr?r! manager_idschedule_createes r/create_schedulerM;s&|$!!),J !%%fb155d; 44M   !77H-H #2 CC  33,SVH5   s*AB: B5 B: B7 'B22B77B:z/schedules/bulk bulk_datac,Kt||jd}|s!|jdijd}|sttjd|jdg}|rt |t sttjd|jd|jd |jd |jd d d }t||||}dt|ddt|dd|dSw)a& Create schedules for multiple advisors at once (manager only) Expects a JSON object: { "advisor_ids": [1, 2, 3], "shift_period_start": "2023-06-01", "shift_period_end": "2023-06-07", "shift": "SHIFT_1", "advisor_status": true } rAr$rBrCr' advisor_idsz$advisor_ids must be a non-empty listshift_period_startshift_period_endshiftadvisor_statusT)rQrRrSrTzCreated schedules for success advisors, failed failedmessageresults) r0r,rrrEr*listrHrlen)rNr?r!rJrP base_scheduler[s r/create_bulk_schedulesr_hs &|$!!),J !%%fb155d; 44M  -- r2K jd;339  (mm,@A%MM*<=w'#--(8$? M,B ]JWG,C 0B,C+DKPST[\dTePfOggno sDDz/schedules/{schedule_id} schedule_idczKt||jd}|s!|jdijd}|sttjd t j di|}t||||S#t$r,}ttjdt|d}~wwxYww) a Update an existing advisor schedule (manager only) Expects a JSON object with fields to update: { "shift_period_start": "2023-06-01", "shift_period_end": "2023-06-07", "shift": "SHIFT_2", "advisor_status": false } rAr$rBrCr'rDNr<) r0r,rrrErAdvisorScheduleUpdaterGrHrIr)r`r>r?r!rJschedule_updaterLs r/update_schedulerds&|$!!),J !%%fb155d; 44M   !77H-H #2{OZ PP  33,SVH5   s*AB; B5B; B8 'B33B88B;c8Kt|t||Sw)z3 Delete an advisor schedule (manager only) )r0r)r`r?r!s r/delete_schedulerfs|$ #2{ 33sz/all-schedules)default_factoryfilterscKt||jr|jrytj|jtj j }tj|jtjj }n%tj}|tdz }|}t||j||Sw)zI Get advisor schedules with optional JSON filters (manager only) )weeks) r0r5r6rcombinemintimemaxnowrrr4)rhr?r!start_datetime end_datetimetodays r/ get_schedulesrts|$g..!))'*<*O>O>QR''(8(8(,,:K:K:MN  !33  !W%7%7 VVsCCz/schedules/refreshc6Kt|t|Sw)zT Refresh the status of all advisors based on their schedules (manager only) )r0r)r?r!s r/refresh_advisor_statusesrvs|$ ( ++sz'/schedules/advisor/{advisor_id}/refreshr4cDKt|t||}d||dSw)zW Refresh the status of a single advisor based on their schedule (manager only) zAdvisor status refreshed)rZr4in_shift)r0r)r4r?r!rxs r/refresh_single_advisor_statusrys3|$%R4H.  s z/shiftscKi}tjD]5\}}|d|djd|djdd||<7|Sw)z0 Get available shifts and their timings namestartz%H:%Mend)r{r|r})ritemsstrftime)r!formatted_shifts shift_key shift_infos r/ get_shiftsrsf!' :v&(11':e$--g6' #"0 sAAz/shifts/{shift}/advisorsz!Date to check (defaults to today)) descriptionrSrc Kt||xs"tjj}tj|tj j }|jtjjtjj|k(tjj|ktjj|k\tjjdk(j}g}|D]}|jtj jtj j"|j$k(j'} | sg|j)| j"| j*| j,| j.| j0|j"|j|j|jd |Sw)z Get all advisors scheduled for a specific shift on a given date Returns a list of advisors with their schedule details T) r4 full_nameemail phone_numberrr`rSrQrR)r0rrprrlrmrnqueryrAdvisorSchedulefilterrSrQrRrTallAdvisorrBr4firstappendrrrr) rSrr?r! check_datecheck_datetime schedulesresultscheduleadvisors r/rr1s}|$.,,.J%%j(,,2C2C2EFN//077$$-11^C//>A--5   ce F((6>>*11 NN  !4!4 4 %'   MM%jj$..  ' 4 4!..'{{!&.&A&A$,$=$=   $ Ms E9G5fastapirrrrrrsqlalchemy.ormr typingr r r r rrrloggingpydanticrapp.common.databaser app.commonrrservices.schedule_servicerrrrrrrrrr"auth_service.services.auth_servicer router getLoggerr7loggerr0r3postrrIrMr_putr:rddeleterfr+rtrvryr,rrrr<r1r/rsdKJ",,..&&   G    8 $  T  $9$ \'*A*AB$(I&/ !89)DS>)D)D)DC)DX tCH~> $S &/ !896CH~666?6r &w7N7NO%)I&/ !89 )Q)QS>)Q )Q )QP)QX)*&/ !89 4 4 4 4+ 4 d4S>.BC$($>&/ !89W !WWWDW2 !"&/ !89 , , ,# , 67&/ !898*I !89$ &tDcN7KL!3VW&/ !89 0 0 4.0 0 0M0d ,T$sCx.=QR9&/ !89)B sCx.)B)B)BS)BV $T#s(^D%)#Y&/ !89NcNNNNENr1