o ;i=@sUddlmZmZmZddlmZmZmZddlZddl Z ddl Z ddl Z e Z eeed<iZeeeefed<eZe Z eeed<daee jed<ddhZd hZd Zd Zd d hZedddZedddZdedefddZiZdedefddZdddedefddZ d ede!fd!d"Z"d edefd#d$Z#d ed%efd&d'Z$e%d(defd)d*Z&defd+d,Z'e(d-d.e)fd/d0Z*dS)1)FastAPI WebSocketWebSocketDisconnect)SetOptionalDictNclients email_clientsdb_poolzhttps://b52-9979.onlinez https://football.b52-9979.onlinexxtruc_ntdangstartupc s.tjdddddddddIdHatd dS) Nz 127.0.0.1i rb52Tr )hostportuserpassworddb autocommitminsizemaxsizez? DB pool created)aiomysql create_poolr printrr8/var/www/html/football.b52-9979.online/server_fastapi.py startup_events r shutdowncs4tdurttIdHdatddSdS)Nz? DB pool closed)r close wait_closedrrrrrshutdown_event+s r$ session_tokenreturnc stdur tdt4IdHK}|tj4IdH*}|d|fIdH|IdH}|duWdIdHWdIdHS1IdHsLwYWdIdHdS1IdHsbwYdS)NzDB pool not initializedz4SELECT 1 FROM users WHERE session_token = %s LIMIT 1)r RuntimeErroracquirecursorr DictCursorexecutefetchone)r%conncurrowrrrsession_exists4s .r0wsdatacs.z ||IdHWdStyYdSwN) send_text Exception)r1r2rrr send_safeDs  r6)excluder7csXg}ttD]}|dur||urq|t||q|r*tj|ddiIdHdSdS)Nreturn_exceptionsT)listrappendr6asynciogather)r2r7taskscrrrbroadcast_safeJs r?emailcCs|tvrtStSr3) ADMIN_EMAILS ADMIN_MAXUSER_MAXr@rrr max_sessionsSsrEcCs|pdtvS)N)striplowerNO_KICK_EMAILSrDrrr is_no_kickVsrJnew_wsc st|}|s dS||t|}t||krdSt||d}t|d|}tjd|dkr4dnd|tdddd}|D])}t ||IdHz |j d d IdHWn t y`Ynwt |||qB|svt |ddSdS) Nr kickanother_session_connectedtoo_many_sessions)typereasonr@tsF,: ensure_ascii separatorscode)r getdiscardrElenr9jsondumpstimer6r"r5rpop)r@rKws_setlimit need_kickto_kickpayloadold_wsrrrkick_oldYs<      rhz/banh/wsc s@|IdH|jd}|tvr|jddIdHdS|jd}|jd}|dus1|dur<|jddIdHdS|dkrRt|IdHsR|jddIdHdSt|s^t||IdHt |t  |t |zztt|} |IdH}zt|}Wntjyt|tjdd |d d d IdHYqvwt|tst|tjdd |d d d IdHqv|d}|dkrtjd|ddd dd}t||IdHt|IdH|dkrtjd|ddd dd}t||IdHt|IdHqwty YnwWt ||dSt ||w)NoriginrXrYtokenr@ danimhdmaetTerror invalid_json)rOrlrawF)rVjson_not_objectrOgetdatamatchesr2)rOr2rRrU getdatatest matches_test)acceptheadersr[ALLOWED_ORIGINSr" query_paramsr0rJrhr setdefaultsetaddrr; create_taskticker receive_textr^loadsJSONDecodeErrorr6r_ isinstancedictr?rr\cancel) r1rirjr@ tick_taskmsgobjmsg_typetextrrr ws_endpoints      &   rcs>d} t|td|dIdH|d7}tdIdHq)NrTping)rOnr )r6r^r_r;sleep)r1irrrr|sr|z /broadcastrfcsBtd|dtjfddttDIdHdttdS)N broadcast)rOrfc3s|]}t|VqdSr3)r6).0r>r2rr sz!http_broadcast..T)oksent_to)r^r_r;r<r9rr])rfrrrhttp_broadcasts"r)+fastapirrrtypingrrrr;r^r`rryr__annotations__r strappr PoolrvrArBrCrIon_eventr r$boolr0r2r6r?intrErJrh websocketrr|postrrrrrrsD   'E