o iN#@s|UddlmZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZddlZddlmZdd lmZdd lmZdd lmZd d lmZmZe dZ edZ e dZ!Gddde Z"dddddddZ#de$d< Gddde Z%Gddde dd Z&eGd!d"d"Z'eGd#d$d$Z(Gd%d&d&e Z)d2d-d.Z*d3d0d1Z+dS)4) 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@seZdZUded<ded<dS) _RegionDatastrbase_url gcp_regionN)__name__ __module__ __qualname____annotations__rr\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/auth.pyr!s  rzhttps://logfire-us.pydantic.devzus-east4)rrzhttps://logfire-eu.pydantic.devz europe-west4)useuzdict[str, _RegionData]REGIONSc@"eZdZUdZded<ded<dS) UserTokenDatazUser token data.rtoken expirationNrrr__doc__rrrrrr3  rc@seZdZUdZded<dS)UserTokensFileDataz/Content of the file containing the user tokens.zdict[str, UserTokenData]tokensNr"rrrrr%:s  r%F)totalc@sPeZdZUdZded<ded<ded<edd d Zedd d ZdddZ dS) UserTokenz A user token.rr rr!rreturnr cCs||d||ddS)Nr r!)r rr!r)clsrr rrrfrom_user_token_dataHs zUserToken.from_user_token_databoolcCs*tjtjdt|jdjtjdkS)zWhether the token is expired.)tzZ)tzinfo)rnowrutc fromisoformatr!rstripreplace)selfrrr is_expiredPszUserToken.is_expiredcCsd}t|j}r|d}|tvrd}|d|jd}|r3||d|ddd7}n ||jdd7}|d7}|S) Nrregion (z) - safe_partr z****)PYDANTIC_LOGFIRE_TOKEN_PATTERNmatchr grouprupperr)r5r7r< token_reprrrr__str__Ws "zUserToken.__str__N)rrr rr)r )r)r,)r)r) rrrr#r classmethodr+propertyr6r@rrrrr(@s   r(c@s^eZdZUdZded< ded< ddd d ZddddZddddZdddZd ddZ dS)!UserTokenCollectionzA 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_tokensrpathN Path | Noner)NonecCs^|dur|nt|_z ttt|j}Wn tyi}Ynwdd|diD|_dS)NcSs$i|]\}}|tdd|i|qS)rr)r().0urlr rrr ~s$z0UserTokenCollection.__init__..r&) DEFAULT_FILErErr%rFileNotFoundErrorgetitemsrD)r5rEdatarrr__init__vs  zUserTokenCollection.__init__r str | Noner(cCst|j}|dur|j|}|durtd|dnFt|dkr)|d}n;t|dkr`ddd t|dd D}tj d |dd d t dt|dDt t j dd}||d}ntd|jrotd|d|S)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). Nz%No user token was found matching the z8 Logfire URL. Please run `logfire auth` to authenticate.rr css4|]\}}|d|d|jrdnddVqdS)z. r8expiredvalid)Nr6)rHir rrr s  z0UserTokenCollection.get_token..)startz/Multiple user tokens found. Please select one: cSsg|]}t|qSr)r)rHrXrrr sz1UserTokenCollection.get_token..)file)choicesconsolezKYou are not logged into Logfire. Please run `logfire auth` to authenticate.z User token z7 is expired. Please run `logfire auth` to authenticate.)listrDvaluesrMr lenjoin enumerater askranger sysstderrr6)r5r tokens_listr choices_str int_choicerrr get_tokens0          zUserTokenCollection.get_tokenr,cs>durfdd|jD}n|j}tdd|DS)aCheck 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. Nc3s|] }|jkr|VqdSNrrHtrmrrrYsz3UserTokenCollection.is_logged_in..css|]}|j VqdSrlrWrnrrrrYs)rDr`any)r5rr&rrmr is_logged_ins z UserTokenCollection.is_logged_inrr rcCs"t|||j|<}||S)z#Add a user token to the collection.)r(r+rD_dump)r5rr user_tokenrrr add_tokenszUserTokenCollection.add_tokencCs|jd0}|jD]!\}}|d|d|d|jd|d|jdq WddS1s9wYdS)z}z|d7}|dkr/td|td WYd }~n d }~ww| }|rG|Sq ) 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/rT)timeoutrzFailed to poll for token.z%Failed to poll for token. Retrying...N) r rMrrrrr warningswarnr)r|rzr~ auth_endpointerrorsrropt_user_tokenrrrpoll_for_tokens" r)r|r}r~rr)r)r|r}rzrr~rr)r), __future__rrrerfr dataclassesrrrpathlibrtypingrr urllib.parser r rich.consoler rich.promptr typing_extensionsr logfire.exceptionsr utilsrrhome HOME_LOGFIRErKcompiler;rrrrr%r(rCryrrrrrrsR         &_