j#dZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z mZmZmZddlmZdd lmZmZmZdd lmZmZmZmZed g ZGd de ZGdde ZGdde ZGdde Z ejCdeeeefd-dZ"ejGdeeeefd.dZ$ejCdeeeefd/dZ%ejGdeeeefd0dZ&ejCdeeeefd/dZ'ejCdeeeefd/dZ(ejCd eeeefd/d!Z)ejCd"eeeefd1d#Z*d2d$Z+ejGd eeeefd3d%Z,ejGd&eeeefd1d'Z-ejGd(eeeefd4d)Z.ejCd*d+eeeefd5d,Z/y+)6z5Centers, zones, members and versioned configurations.) annotations) APIRouterDepends HTTPException) Connection) BaseModelField)User current_user require_adminrequire_center) replay_config) active_config config_by_id seed_default)Jsonbfetchallfetchoneget_conncenters)tagsceZdZUedZded<eddZded<d Zd ed <d Zded <dZ ded<eddZ ded<y )CenterInz^[a-z0-9-]{2,40}$)patternstrslugr x min_length max_lengthnameN str | None afp_center_idzAmerica/Los_AngelestimezoneenlocaleNACz ^(NAC|SAC)$)defaultr danger_scale) __name__ __module__ __qualname__r r__annotations__r"r$r%r'r*0/var/www/avy-guidance/api/app/routers/centers.pyrrsR12D#2s3D#3 $M:$)Hc)FCe]CL#Cr0rcHeZdZUeddZded<dZded<dZd ed <y) ZoneInrrrr"Nr# afp_zone_iddict[str, str] | None band_labels)r+r,r-r r"r.r5r7r/r0r1r3r3s(s3D#3"K")-K&-r0r3c~eZdZUdZded<dZded<dZded<dZded <dZd ed <dZ d ed <e dZ ded<y) ConfigDraftInNz int | None parent_idzdict[str, list[int]] | None danger_tablez!dict[str, list[list[int]]] | Nonegradient_tablez dict[str, dict[str, int]] | Nonelikelihood_matrixz bool | Noneshow_danger_suggestionr6r7 )r r rationale) r+r,r-r:r.r;r<r=r>r7r r@r/r0r1r9r9 sS Iz 04L-487>*.K.)-K&-b)Is)r0r9ceZdZUdZded<y) ApproveInNr#r@)r+r,r-r@r.r/r0r1rBrB+s  Iz r0rBz/centerscb|jr t|dSt|d|jfS)Nz#SELECT * FROM centers ORDER BY namezoSELECT c.*, m.role FROM centers c JOIN memberships m ON m.center_id = c.id WHERE m.user_id = %s ORDER BY c.name)is_adminrid)connusers r1 list_centersrH/s3 }}CDD  y   r0ct|t|d|j}t||d|j|S)NzINSERT INTO centers (slug, name, afp_center_id, timezone, locale, danger_scale) VALUES (%(slug)s, %(name)s, %(afp_center_id)s, %(timezone)s, %(locale)s, %(danger_scale)s) RETURNING *rE)r r model_dumprrE)bodyrFrGrows r1 create_centerrM:sB$   u   C s4y$''* Jr0z/centers/{center_id}ct||t|d|f}|s tddt|d|f}i|||j |dS)Nz#SELECT * FROM centers WHERE id = %sizCenter not foundz6SELECT * FROM zones WHERE center_id = %s ORDER BY name)zonesrole)rrrrrole_in) center_idrFrGcenterrOs r1 get_centerrTGs\4# dAI< PF C!344 TSV_Ua bE Ff Fudll9.E FFr0z/centers/{center_id}/zonesc t||dt|d||j|j|jrt |jfSdfS)NTadminzaINSERT INTO zones (center_id, name, afp_zone_id, band_labels) VALUES (%s, %s, %s, %s) RETURNING *)rrr"r5r7r)rRrKrFrGs r1 create_zonerXQs^4$/  k DIIt//DL\L\t7G7G1Hg cgg r0z/centers/{center_id}/membersc:t||dt|d|fS)NTrVzSELECT u.id, u.email, u.name, u.last_login_at, m.role FROM memberships m JOIN users u ON u.id = m.user_id WHERE m.center_id = %s ORDER BY u.namerrrRrFrGs r1 list_membersr\[s(4$/   5   r0z/centers/{center_id}/configc2t||t||S)N)rrr[s r1get_active_configr^fs4# y ))r0z/centers/{center_id}/configsc6t||t|d|fS)Na{SELECT cc.id, cc.version, cc.status, cc.rationale, cc.parent_id, cc.created_at, cc.proposed_at, cc.approved_at, cu.name AS created_by_name, au.name AS approved_by_name FROM center_configs cc LEFT JOIN users cu ON cu.id = cc.created_by LEFT JOIN users au ON au.id = cc.approved_by WHERE cc.center_id = %s ORDER BY cc.version DESCrZr[s r1 list_configsr`ls(4#   ?   r0z/configs/{config_id}c<t||}t||d|S)NrR)rr config_idrFrGrLs r1 get_configrdzs" tY 'C4[)* Jr0c |d}t|dxrt|jdg}dD]V}|j|}t|tr!t||k7st d|DsEt dd|d|d |d j D]5\}}t d |jDs't dd |d y)N dangerTablesizes1)rh2345c3dK|](}t|t xsd|cxkxrdknc *yw)rN isinstanceint.0vs r1 z#_validate_tables..s4?wnoJqRUDV@V@i^_cd^ihi^iZi@i?w.0zDanger table row z must have z levels between 0 and 5.likelihoodMatrixc3dK|](}t|t xsd|cxkxrdknc *yw)r4rnNrorrs r1ruz#_validate_tables..s,OQ:a%%8Q![q[8OrvzLikelihood matrix row z must contain levels 1-5.)lengetrplistanyritemsvalues)cfgtablecolslkrLdists r1_validate_tablesrs  E s7|  8UYYsB%7!8D'hiim#t$CD(8C?wsv?wr7rrrr@rE)rRrKrFrGparentrversions r1 create_draftrsP4$/37>>\$ /}UY[dGeF ~~&-:C!CDD vh C $!..M &#22O )"&"8"8  "".&*&A&A "# # ,,LStqt}sABEFG   U GU3ZtwwO  r0z/configs/{config_id}/proposect||}t||dd|ddk7r tdd|jd|fd diS) NrRTrVstatusdraftzOnly drafts can be proposedzPUPDATE center_configs SET status = 'proposed', proposed_at = now() WHERE id = %sok)rrrexecuterbs r1proposersV tY 'C4[)6 8}C!>??LLcfoeqr $<r0z/configs/{config_id}/approvec`t||}t||dd|ddk7r tdd|d|jk(r|js td d |j d |df|j d |j|j rd |j nd|fddiS)z^Activate a proposed version. Needs a DIFFERENT center admin (or a site admin) than its author.rRTrVrproposedrz&Only proposed versions can be approved created_byiz2A different center admin must approve this versionzWUPDATE center_configs SET status = 'retired' WHERE center_id = %s AND status = 'active'zUPDATE center_configs SET status = 'active', approved_by = %s, approved_at = now(), rationale = coalesce(rationale, '') || coalesce(%s, '') WHERE id = %sz Approval note: Nr)rrrrErDrr@)rcrKrFrGrLs r1approvers tY 'C4[)6 8} "C!IJJ <DGG#DMMC!UVVLLjmpq|m}lALL X DNN''78PTV_` $<r0z/configs/{config_id}/replayNct||}t||dt||d}t||d|d|d|S)zjBacktest a config against imported forecasts: agreement vs the active config and which suggestions change.rRr)rrrr)rcholdout_seasonrFrGrLactives r1replayrsR tY 'C4[)* 4[!1 2F s;/Xx@PR` aar0)rFrrGr )rKrrFrrGr )rRrqrFrrGr )rRrqrKr3rFrrGr )rcrqrFrrGr )rrreturnNone)rRrqrKr9rFrrGr )rcrqrKrBrFrrGr )rcrqrr#rFrrGr )0__doc__ __future__rfastapirrrpsycopgrpydanticrr authr r r r calibrationrconfigsrrrdbrrrrrouterrr3r9rBr{rHpostrMrTrXr\r^r`rdrrrrrr/r0r1rs;"55%DD'??44  $DyD.Y. *I*! !J$+H$5GLDYZ5