zj(UddlmZddlmZmZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZd Zd ed <d d ddddZded<ddddd Zded<d,dZd-dZd.dZd/dZd0dZd1dZd2d Z d3d!Z d4d"Z!d-d#Z"d5d$Z#d5d%Z$d6d&Z% d7d'Z& d8d(Z' d9d)Z(d:d*Z)y+);) annotations) HTTPExceptionstatus)Session)UserUserRegisterStep)ProgrammeSession) seed_prior_programme_completions) selectinload)RegisterStepItemOutRegisterStepStatusUserRegisterStepAcceptRequestUserRegisterStepSaveRequestUserRegisterStepStateResponselevelmoduleweeksessionztuple[str, ...]REGISTER_STEP_ORDERLevelModuleWeekr Completed)rrrr completedzdict[str, str]REGISTER_STEP_LABELS)level_id level_name) module_id module_name)week_id week_name) session_id session_namezdict[str, tuple[str, str]] _STEP_FIELDSc|jtjtj|jk(j SN)queryrfilteruser_ididfirst)dbusers L/var/www/html/kelly/kelly-backend/app/services/user_register_step_service.pyget_user_register_stepr0#s6 88$ % , ,-=-E-E-P Q W W YYct||}|0|j|d|_d|_|j yyr')r0deleteprogramme_session_idprogramme_assessment_idcommitr-r.rows r/delete_user_register_stepr9's< T *C  #$(!'+$ r1c|jj}|dk(rttS tj |S#t $r!}t tjd|d}~wwxYw)NrzDcurrent_step must be one of: level, module, week, session, completed status_codedetail) striplowerlenrindex ValueErrorrrHTTP_400_BAD_REQUEST)step normalizedexcs r/ _step_indexrG0sn##%J[ &''"((44 33Y  sA A3A..A3c|y|j|jf|j|jf|j|j f|j |jfd}|j|dS)N)NNr) rrrr r!r"r#r$get)r8rDmappings r/_selected_for_steprK=sf {,,/==#//2cmm,NNC$4$45 G ;;t\ **r1cft||\}}|duxst|xr|jSr')rKboolr>)r8rD selected_id selected_names r/_step_has_selectionrPIs6!3C!>K d " Sd=+R]=P=P=R&SSr1cZt||sttjd|dy)Nz Select a z before continuing.r;)rPrrrC)r8rDs r/_validate_step_selectionrRNs4 sD )33tf$78   *r1c V|d}d}d}nd|js|jdk(rd}tt}d}n5|jj j }t |}d}g}ttD]x\}}|s||krtj}n&||k(rtj}ntj}t||\}} |jt|t|||| z| xr|duxrt!||tvr|nd} t#||| ||S) NrrFrT)rDlabelrrNrOr) current_step is_completed can_proceedsteps selection)rVrUr@rr>r?rG enumerater ropenlockedrKappendr rrPr) r8rU current_idxrVrXidxrD step_statusrNrOrWs r/build_register_step_stateraVsE {    S--<" -.  ''--/557 !,/  ')E23 T 3,,66K K ,11K,33K%7T%B" ] *40"'+   4(  i tO i \EX5X\^g h )!!  r1ct|\}}|j}||vrt||dy||vr3t||d}t|xrt |j Sy)NTF)r%model_fields_setgetattrrMstrr>)bodyrDid_field name_field sent_fieldsvalues r/_step_data_in_bodyrksm(-Hj''K;744#@#L[ j$/E0c%j..011 r1c"t|\}}|j}|r||vr"t||d}||r||vr t||||r||vrIt||d}t |t r|j xsd}||r||vrt|||yyyy)zTUpdate id/name fields for one step only; never touches current_step or is_completed.N)r%rcrdsetattr isinstancerer>)r8rfrDonly_sent_fieldsrgrhrirjs r/_apply_step_fieldsrps(-Hj''K x;6h-  !1h+6M C5 ) z[8j$/ eS !KKM)TE  !1jK6O CU +7P!1 9r1ct||}||jr |jy|jtj t tjjtj|jk(j}|yt|jd}|r|dnd}t||j|||j|_ |r |jnd|_|jy)zCMark everything before the user's registration anchor as completed.Nc|jSr') sort_order)items r/z1reseed_track_from_register_step..str1)keyr)r0rVr#r(r optionsr assessmentsr)r+r,sortedr r4r5flush)r-r. register_steprrxfirst_assessments r/reseed_track_from_register_stepr}s*2t4MM$>$>-BZBZBb !" .::; <  ##}'?'?? @   ,,2NOK)4{1~$$R';KL ' D:J#3#6#6PTD HHJr1c6|jyt||yr')r#r}r7s r/_sync_user_programme_pointerrs ~~#B-r1ct||jt|j}|ttdz k\rd|_d|_t |||yt|dz|_d|_y)NrTF)rRrUrGr@rrVr)r-r.r8r^s r/_advance_register_steprsmS#"2"23c../Kc-.22&$Rs3*;?;CCr1ct||}|9t|jdd}|j||j |S)NrF)r*rUrV)r0rr+addrzr7s r/_get_or_create_rowrs? T *C {twwWSXY s   Jr1c tD]}t||st|||d tD]}t||d|_d|_t ||||j|j|t|S)uVSession submit with level/module/week/session ids — complete onboarding in one call.Tror) rrkrprRrUrVrr6refreshra)r-r.r8rfrDs r/_finish_register_step_from_bodyrsy$ dD ) sD$ F$$ d+$#CC T3/IIKJJsO $S ))r1c t||}|js|jdk(rttj dt D]}t||st|||d t|dr/t|||tj|jS|j|j|t|S)Nrz"Registration is already completed.r;Trr)rrVrUrrrCrrkrprrmodel_validate model_dumpr6rrar-r.rfr8rDs r/save_user_register_steprs R &C 3++{:337  $ dD ) sD$ F$$ *. c8GGHYZ  IIKJJsO $S ))r1ct||}|js|jdk(rttj dt |drt||||S|jjj}|tvrttj dt|||dt||t||||j|j|t!|S)Nr)Registration steps are already completed.r;rzInvalid current step.Fr)rrVrUrrrCrkrr>r?r%rprRrr6rrars r/accept_user_register_stepr s R &C 3++{:33>  $ *.r4dCC    ! ! # ) ) +D <(C(CLcddsD$?S$'2tS)IIKJJsO $S ))r1ct||}|js|jdk(rttj dt ||||j|j|t|S)Nrrr;) rrVrUrrrCrr6rrar7s r/complete_user_register_stepr(sj R &C 3++{:33>  2tS)IIKJJsO $S ))r1N)r-rr.rreturnUserRegisterStep | None)r-rr.rrNone)rDrerint)r8rrDrerztuple[int | None, str | None])r8rrDrerrM)r8rrDrerr)r8rrr)rf;UserRegisterStepSaveRequest | UserRegisterStepAcceptRequestrDrerrM) r8rrfrrDrerorMrr)r-rr.rr8rrr)r-rr.rrr) r-rr.rr8rrfrrr)r-rr.rrfrrr)r-rr.rrfrrr)r-rr.rrr)* __future__rfastapirrsqlalchemy.ormr app.modelsrrapp.programme.models_programmer (app.programme.services.programme_servicer r app.schemasr r rrrr__annotations__rr%r0r9rGrKrPrRrarkrpr}rrrrrrrr1r/rsu")"-;U'(N_M   (n(* $- , (Z  +T  .b  E     , , E, ,  ,  ,0.. ***"2*:W*"*&***#>*"*2***#@*"*6 *r1