g ddlmZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl Z ddl mZdd lmZej"d z Z ed z Z Gd d e ZGdde ZGdde ZddZddZddZy)) annotationsN)datetimetimezone)Path) TypedDict)urljoin)LogfireConfigError)UnexpectedResponsez.logfirez default.tomlc&eZdZUdZded<ded<y) UserTokenDatazUser token data.strtoken expirationN__name__ __module__ __qualname____doc____annotations__K/var/www/openai/venv/lib/python3.12/site-packages/logfire/_internal/auth.pyr r s JOrr ceZdZUdZded<y) DefaultFilez!Content of the default.toml file.zdict[str, UserTokenData]tokensNrrrrrrs + $$rrc&eZdZUdZded<ded<y) NewDeviceFlowz.Matches model of the same name in the backend.r device_codefrontend_auth_urlNrrrrrr#s8rrc tjd}t|d} |j|d|i}t j ||j}|d|dfS#t j$r}td|d}~wwxYw) aVRequest a device code from the Logfire API. Args: session: The `requests` session to use. base_api_url: The base URL of the Logfire instance. Returns: return data['device_code'], data['frontend_auth_url'] The device code and the frontend URL to authenticate the device at, as a `NewDeviceFlow` dict. r z/v1/device-auth/new/ machine_name)paramsz Failed to request a device code.Nrr ) platformunamerpostr raise_for_statusrequestsRequestExceptionr json)session base_api_urlr"device_auth_endpointresedatas rrequest_device_coder1*s>>#A&L"<1GHLll/8VlW++C0((*D  %8 9 99  $ $L !CD!KLs*A))B < BB c8t|d|}d} |j|d}tj||j }|r|S>#t j $r5}|dz }|dk\r td|tjdYd }~Dd }~wwxYw) 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)timeoutr zFailed to poll for token.z%Failed to poll for token. Retrying...N) rgetr r'r*r(r)r warningswarn)r+rr, auth_endpointerrorsr.opt_user_tokenr/s rpoll_for_tokenr<@sL,A+*OPM F  &++mR+8C  / / 44788:N%% (( C aKF{()DE1L MMA B B  Cs(AB$+BBc"|djD]y\}}tj|djdj t j }||k(sQtjt j |ksyyy)zjCheck if the user is logged in. Returns: True if the user is logged in, False otherwise. rrZ)tzinfo)tzTF)itemsr fromisoformatrstripreplacerutcnow)r0 logfire_urlurlinfo expiry_dates r is_logged_inrK`sx (^))+ T,,T,-?-F-Fs-KLTT\d\h\hTi + (,,(,,"?+"M , r)r+requests.Sessionr,rreturnztuple[str, str])r+rLrrr,rrMr )r0rrGrrMbool) __future__rr$r7rrpathlibrtypingr urllib.parserr(logfire.exceptionsr utilsr home HOME_LOGFIRE DEFAULT_FILEr rrr1r<rKrrrrXsy"' 1%tyy{Z' An, %I%)% I:,&@ r