]j-&%SSKJr SSKrSSKrSSKrSSKrSSKJr SSKJrJ r SSK J r SSK J r Jr SSKJr SSKrSSKJr SS KJr SS KJr SS KJr S S KJrJr \ R:"5S- r\S- r\R@"S5r!"SS\ 5r"SSS.SSS.S.r#S\$S'"SS\ 5r%"SS\ SS 9r&\"S!S"55r'\"S#S$55r("S%S&\ 5r)S)S'jr*S*S(jr+g)+) annotationsN) dataclass)datetimetimezone)Path) TypedDictcast)urljoin)Console) IntPrompt)Self)LogfireConfigError)UnexpectedResponseread_toml_filez.logfirez default.tomlzU^(?Ppylf_v(?P[0-9]+)_(?P[a-z]+)_)(?P[a-zA-Z0-9]+)$c*\rSrSr%S\S'S\S'Srg) _RegionData!strbase_url gcp_regionN)__name__ __module__ __qualname____firstlineno____annotations____static_attributes__rS/home/mande/repo/quber/.venv/lib/python3.13/site-packages/logfire/_internal/auth.pyrr!s MOrrzhttps://logfire-us.pydantic.devzus-east4)rrzhttps://logfire-eu.pydantic.devz europe-west4)useuzdict[str, _RegionData]REGIONSc.\rSrSr%SrS\S'S\S'Srg) UserTokenData3zUser token data.rtoken expirationrNrrrr__doc__rrrrr r%r%3s JOrr%c$\rSrSr%SrS\S'Srg)UserTokensFileData:z/Content of the file containing the user tokens.zdict[str, UserTokenData]tokensrNr)rrr r,r,:s 9 $$rr,F)totalcj\rSrSr%SrS\S'S\S'S\S'\S Sj5r\S Sj5r SS jr S r g ) UserToken@z A user token.rr'rr(cU"USUUSS9$)Nr'r()r'rr(r)clsrr's r from_user_token_dataUserToken.from_user_token_dataHs!.\*  rc[R"[RS9[R"UR R S55R[RS9:$)zWhether the token is expired.)tzZ)tzinfo)rnowrutc fromisoformatr(rstripreplace)selfs r is_expiredUserToken.is_expiredPsS||x||,0F0FtG]G]^aGb0c0k0k<<1l1   rc\Sn[RUR5=n(aURS5nU[;aSnUR 5SUR S3nU(a)X2RS5URS5SS-- nOX0RSS- nUS- nU$) Nr!region (z) - safe_partr'z****)PYDANTIC_LOGFIRE_TOKEN_PATTERNmatchr'groupr#upperr)r@rDrI token_reprs r __str__UserToken.__str__Ws288D D5 D[[*FW$ 'r$--=  ++k2U[[5I"15MM MJ **Ra. (Jf rrN)rrr'r%returnr )rObool)rOr) rrrrr*r classmethodr5propertyrArMrrrr r1r1@s? JMO    rr1c~\rSrSr%SrS\S'S\S'SSSjjrSSS jjrSSS jjrSS jr SSS jjr SS jr Sr g)UserTokenCollectiongzA collection of user tokens, read from a user tokens file. Args: path: The path where the user tokens will be stored. If the path doesn't exist, an empty collection is created. Defaults to `~/.logfire/default.toml`. zdict[str, UserToken] user_tokensrpathNc (UbUO[Ul[[[ UR55nUR S05R5VVs0sHup4U[SSU0UD6_M snnUl g![ a 0nNUf=fs snnf)Nr.rr) DEFAULT_FILErWr r,rFileNotFoundErrorgetitemsr1rV)r@rWdataurlr's r __init__UserTokenCollection.__init__vs!,D,  **N499,EFDUYT\T\]egiTjTpTpTrsTrjcC!AC!A5!AATrs! *')D *ss#A<B< B  B c d[URR55nUb.URRU5nUc[ SUS35eO[ U5S:XaUSnO[ U5S:aSR S[USS955n[R"S US3[S[ U5S-5Vs/sHn[U5PM sn[[RS 9S 9nX&S- nO [ S 5eUR(a[ S US35eU$s snf)aGet a user token from the collection. Args: base_url: Only look for user tokens valid for this base URL. If not provided, all the tokens of the collection will be considered: if only one token is available, it will be used, otherwise the user will be prompted to choose a token. Raises: LogfireConfigError: If no user token is found (no token matched the base URL, the collection is empty, or the selected token is expired). %No user token was found matching the 8 Logfire URL. Please run `logfire auth` to authenticate.rr c3b# UH%upUSUSUR(aSOSS3v M' g7f)z. rEexpiredvalid)NrA).0ir's r 0UserTokenCollection.get_token..s7$ ?HA#Rwbe.>.>G LAN ?s-/)startz/Multiple user tokens found. Please select one: )file)choicesconsoleKYou are not logged into Logfire. Please run `logfire auth` to authenticate.z User token z7 is expired. Please run `logfire auth` to authenticate.)listrVvaluesr[rlenjoin enumerater askrangerr sysstderrrA)r@r tokens_listr' choices_strrl int_choices r get_tokenUserTokenCollection.get_tokens?4++2245  $$((2E}(;H:FAA   "NE   "))$ )+Q ?$K#B;-rR).q#k2BQ2F)GH)GAQ)GHSZZ0J  Q/E$%rs s   ${5'9p%qr r IsD- c^Tb%U4SjURR55nOURR5n[SU55$)zCheck whether the user token collection contains at least one valid user token. Args: base_url: Only check for user tokens valid for this base URL. If not provided, all the tokens of the collection will be considered. c3J># UHoRT:XdMUv M g7fN)r)rktrs r rm3UserTokenCollection.is_logged_in..sU!:AjjH>Taa!:s# #c3B# UHoR(+v M g7frrj)rkrs r rmrs4V||##Vs)rVruany)r@rr.s ` r is_logged_in UserTokenCollection.is_logged_insI  U!1!1!8!8!:UF%%,,.F4V444rcn[RX5=URU'nUR5 U$)z#Add a user token to the collection.)r1r5rV_dump)r@rr' user_tokens r add_tokenUserTokenCollection.add_tokens22;2P2PQY2aa"Z rcUR(d [S5eUbXR;a[SUS35eUbU/O"[URR55nUHnURU M UR 5 U$)z)Remove user token(s) from the collection.rsrbrc)rVrrtkeysr)r@rremovedr^s r logoutUserTokenCollection.logouts$%rs s  H4D4D$D$7zB==  !) 48*$t?O?O?T?T?V:WC  % rcZURRS5nURR5HXup#UR SUS35 UR SUR S35 UR SUR S35 MZ SSS5 g!,(df  g=f)z>#A&L"<1GHLll/8VlW++C0((*D  %8 9 99  $ $L !CD!KLs(A((B < BB cH[USU35nSnURUSS9n[R"U5 UR 5nU(aU$MC![ R a7nUS- nUS:a [S5Ue[R"S5 S nANHS nAff=f) aPoll the Logfire API for the user token. This function will keep polling the API until it receives a user token, not that each request should take ~10 seconds as the API endpoint will block waiting for the user to complete authentication. Args: session: The `requests` session to use. device_code: The device code to poll for. base_api_url: The base URL of the Logfire instance. Returns: The user token. z/v1/device-auth/wait/r)timeoutrzFailed to poll for token.z%Failed to poll for token. Retrying...N) r r[rrrrrrwarningswarn)rrr auth_endpointerrorsropt_user_tokenrs r poll_for_tokenrsL,A+*OPM F  &++mR+8C  / / 44788:N%% (( C aKF{()DE1L MMA B B  Cs&AB!*-BB!)rrequests.SessionrrrOztuple[str, str])rrrrrrrOr%), __future__rrrer{r dataclassesrrrpathlibrtypingrr urllib.parser r rich.consoler rich.promptr typing_extensionsr logfire.exceptionsrutilsrrhome HOME_LOGFIRErYcompilerHrr#rr%r,r1rTrrrrrr rs " !'"  !"15yy{Z' An, %"$\" )6   6$  #  $I%%%  ## #L nEnE nEbI:,&r