#i+ SrSSKrSSKrSSKJr SSKJrJr SSKJr SSK J r SSK J r SSK JrJr \ "\5r"S S \ 5r"S S \ 5r"S S\ 5r"SS\ 5r"SS5r\"5rS\S\\S\4SjrS\SS4Sjrg)a WebSocket connection manager and live race updates for TipSharks API. Provides a ConnectionManager for race-room-scoped WebSocket connections and a background simulation task that generates mock odds/result updates as a placeholder for future live data integration. N) defaultdict)UTCdatetime) WebSocket) BaseModel) get_logger)RaceStartercB\rSrSr%Sr\\S'\\S'Sr\ S-\S'Sr g) WSMessagez4Schema for incoming WebSocket messages from clients.typerace_idNdata) __name__ __module__ __qualname____firstlineno____doc__str__annotations__intrdict__static_attributes__r?/root/tipsharks/tipsharks-elo-api/apps/backend/api/websocket.pyr r s> I LD$+rr cL\rSrSr%SrSr\\S'\\S'\\S'\ \ \S'Sr g ) OddsUpdateMessage!zOdds update broadcast message. odds_updaterr timestampoddsrN rrrrrrrrrlistrrrrrrr!s#(D# LN t*rrcL\rSrSr%SrSr\\S'\\S'\\S'\ \ \S'Sr g ) ResultUpdateMessage*z Result update broadcast message. result_updaterrr"resultsrNr$rrrr'r'*s#*D# LN $Zrr'c<\rSrSr%SrSr\\S'\\S'\ \S'Sr g) InitialStateMessage3z&Initial race state sent on connection. initial_staterrrrN) rrrrrrrrrrrrrrr,r,3s0D# L Jrr,c\rSrSrSrSSjrS\S\SS4SjrS\S\SS4S jr S \ S\SS4S jr S \ S\SS4S jr S\S\4S jr S\S\4SjrS\SS4SjrS\SS4SjrSSjrSrg)ConnectionManager>zManages WebSocket connections grouped by race room. Each race has its own "room" identified by race_id. Operations are thread-safe via an asyncio lock. returnNcn[[5Ul[R"5Ul0Ulg)N)rset _connectionsasyncioLock_lock_simulation_tasks)selfs r__init__ConnectionManager.__init__Es$7B37G\\^ :<r websocketrcN# UR5IShvN URIShvN URURU5 SSS5IShvN [R SUUR U5S.S9 gNnN[N/!,IShvN (df  ND=f7f)zAccept a WebSocket and add it to the race room. Args: websocket: The WebSocket connection to register. race_id: Race room identifier. NzWebSocket connected for race)rconnection_countextra)acceptr8r5addloggerinfoget_connection_countr:r=rs rconnectConnectionManager.connectJsz   :::   g & * *9 5: *"$($=$=g$F   !:::sTB%BB%BB%B  B%B ,B%B% B% B"B  B"B%cZ# URIShvN URURU5 URU(d URU SSS5IShvN [R SX R U5S.S9 gNyN,!,IShvN (df  NA=f7f)zRemove a WebSocket from the race room. Cleans up the room entry when the last client disconnects. Args: websocket: The WebSocket connection to remove. race_id: Race room identifier. Nz WebSocket disconnected from race)r remainingr@)r8r5discardrDrErFrGs r disconnectConnectionManager.disconnect\s~:::   g & . .y 9$$W-%%g.:  .%4M4Mg4VW  :::sEB+B B+AB B+"B#+B+B+B(B  B($B+messagec# URU5IShvN gN![a [RSSS9 gf=f7f)zSend a JSON message to a single WebSocket client. Args: message: JSON-encoded message string. websocket: The target WebSocket connection. NzFailed to send personal messageT)exc_info) send_text ExceptionrDwarning)r:rOr=s rsend_personal_message'ConnectionManager.send_personal_messagens@ M%%g. . . M NNC3B CB C B4B5B9CCBB BCB!C?CCCc\[URRU[555$)z2Return the number of connections for a given race.)lenr5rXr4r:rs rrF&ConnectionManager.get_connection_counts"4$$((#%899rcXR;$)z9Check if a simulation task is already running for a race.)r9r`s ris_simulation_running'ConnectionManager.is_simulation_runnings0000rcXR;aD[R"[U55nX RU'[R SSU0S9 gg)ztStart the background simulation for a race if not already running. Args: race_id: Race identifier to simulate. zStarted race simulationrr@N)r9r6 create_tasksimulate_race_updatesrDrEr:rtasks rstart_simulation"ConnectionManager.start_simulationsO 00 0&&'s =  y 3 4 $ ) c d $ M3 M9 MQU MsST&:C:C:1S1T1 O O O Os Ot O+rr0racestartersr2cn/nUHnUR(aURROSnUR(aURROSnUR(aURROSnUR UR UR UURUURUURURURURURS. 5 M UR(aURROSnUR(a?URR (a$URR R#5OSnUR$n U (aU R#5OSn UR UR&UR(UR*UR,UR.UR0UR2U UUS. U[5U5S.n [7[9UR 5U S9n U R;5$)zBuild the initial race state JSON string. Args: race: The Race ORM instance. starters: List of Starter ORM instances. Returns: JSON-encoded InitialStateMessage. N) idhorse_id horse_name driver_id driver_name trainer_id trainer_name runner_numberbarrier handicap_mplacingdid_not_finish) rz meeting_id race_number distance_m start_typegaitweathertrack_condition race_datetimevenue meeting_date)rwrx starter_count)rr)horsenamedrivertrainerappendrzr{r}rrrrrrmeetingrr isoformatrrrrrrrrr_r,rmodel_dump_json) rwrx starter_listsr|r~rrrrace_dt race_dt_isormsgs r_build_initial_statersL %&WWQWW\\$ '(xxahhmmT )*qyy~~ ddJJ([[*ll ,!"99ll99"#"2"2  *#',,DLL  DE<@LLT\\MfMf4<<,,668lpL  G)0'##%dK''//++////II||#33((  !X D$ c$''l >C     rrc p# [RSSU0S9 [R"[5nSn[ R "SS5nUS:GaN[RU5S:Xa[RS SU0S9 g [ R "SS5n[U5Vs/sH*nUS -[[ R"S S 5S5S.PM, nn[SU[R"[5R5US9n[RUR5U5IS hvN [ R"SS5n[ R""U5IS hvN [R"[5U- R%5nUS:aGMN[RU5S:a['SU[R"[5R5[U5Vs/sHnUS -US -SS.PM snS9n[RUR5U5IS hvN [RSSU0S9 g s snfGN%Ns snfN*![ R(a [RSSU0S9 g [*a [R-SSU0S9 g f=f7f)aiSimulate live race updates for a race room. Broadcasts mock odds updates every 5-10 seconds for 60 seconds, then sends a final result update. This is a placeholder until real live data integration is built. The function checks for active connections before each broadcast and exits early if the room is empty. Args: race_id: Race identifier to simulate. zRace simulation task startedrr@g <rz0No more connections for race, simulation exitingNg?gI@)r{r#r!)rrr"r#g@g$@r)T)r{rfinished)rrr"r*zRace simulation completedzRace simulation cancelledzRace simulation error)rDrErnowrrandomrandintmanagerrFrangerounduniformrrr\rr6sleep total_secondsr'CancelledErrorrS exception) r start_timeelapsed num_horsesi odds_listrOdelayresult_messages rrgrgsv KK.y'6JKKc"JG2&J.Nl++G49 F$g. 2.Jz**AUE&..d2KQ,OP*  ("",,s+557 G ++G,C,C,EwO O ONN3-E--& & &||C(:5DDFG5l:  ' ' 01 40$",,s+557#:.."#Q1q5dK. N++N,J,J,LgV V V / 77K L? P ' W  ! !M / 77K L N0G8LMNsA J6 6I%J6$I%(1IAI%4I54I%)I*5I%"AI%6I ,I%8I#9I%J6I%I%I%%*J3J6J30J62J33J6)rr6r collectionsrrrfastapirpydanticrpackages.core.common.loggingrpackages.core.storage.modelsr r rrDr rr'r,r0rr%rrrrgrrrrs #"36 H    ))|+|+@   9!t9!tG}9!9!~@N@N@Nr