ĀiddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlZGdd ZeZd)d Zd*d ZGd d eZe dGddZej0j3ddZej0j3ddeZdZdedZdZdZdZ e jBdz Z"d+dZ#edZ$d,dZ%d-d Z&d.d!Z'd"d#d/d$Z(d0d%Z)d*d&Z*d1d'Z+d2d(Z,y)3) annotationsN)contextmanager) dataclass)Path)Lock)OptionalceZdZdZy)GoogleColabStatec>d|_t|_d|_y)NF) is_checkedrlocksecretselfs S/var/www/html/content-pipeline/venv/lib/python3.12/site-packages/fal_client/auth.py__init__zGoogleColabState.__init__sF %) N)__name__ __module__ __qualname__rrrr r s*rr cf ddlm}dt|vS#t$rYyt$rYywxYw)Nr get_ipythonz google.colabF)IPythonrstrModuleNotFoundError NameErrorrs ris_google_colabrs9'[]!333  s 000ctsytj5tjrtjcdddS ddlm} |jd}|jt_dt_tjcdddS#t$r YdddywxYw#t$rdt_YQwxYw#1swYyxYw)Nr)userdataFAL_KEYT) r _colab_stater r r google.colabr! ImportErrorgetstrip Exception)r!tokens rget_colab_tokenr*%s     #  " "&&##  - 'LL+E"'++-L #' ""!##  ##    '"&L  '##sM C B*B38C  B0%C /B00C 3C C  C  C  Cc eZdZy)MissingCredentialsErrorN)rrrrrrr,r,;srr,T)frozenc@eZdZUdZded<ded<eddZd dZy) AuthCredentialsz)Represents an authorization header value.rschemer)c8|jd|jS)N )r0r)rs r header_valuezAuthCredentials.header_valueFs++a |,,rcd|jiS)N Authorization)r3rs r as_headerszAuthCredentials.as_headersJs!2!233rNreturnr)r8zdict[str, str])rrr__doc____annotations__propertyr3r6rrrr/r/?s&3 K J --4rr/ FAL_RUN_HOSTzfal.runFAL_QUEUE_RUN_HOSTzqueue.z auth.fal.aizhttps://z /oauth/token TwXR51Vz8JbY8GUUMy6EyuVR0fTO7N4N auth0_tokenz .portalockz.falcvttjdttj S)N FAL_HOME_DIR)rosgetenvrDEFAULT_FAL_HOME expanduserrrr_get_fal_home_dirrFYs'  .#.>*?@ A L L NNrc#K ddl}ttz }|jj dd|j jt|dd5ddddy#t$rdYywxYw#1swYyxYww)z%Best effort lock shared with fal-cli.rNTparentsexist_okF)fail_when_lockedtimeout) portalockerr(rFAUTH_LOCK_FILENAMEparentmkdirutilsTemporaryFileLockr)rN lock_files r _token_lockrU]s "#&88I 4$7    , , I -     s@B A+AB A>" B + A;8B :A;;B >BB c6ttz }|jsy|jj Dcgc]#}|j s|j %}}|sy|d}d}t |dkDr|d}|xsd|xsdfScc}w)N)NNr)rFAUTH_TOKEN_FILENAMEexists read_text splitlinesr'len)pathlinelines refresh_token access_tokens r_read_auth_tokensrbos  !4 4D ;;=&*nn&6&A&A&C Tdtzz|TZZ\ TE T !HM"&L 5zA~Qx  D,"6$ 66 Us BBcttz }|jjdd|g}|r|j ||j dj |y)NTrH )rFrXrPrQappend write_textjoin)r`rar]contentss r_write_auth_tokensrisR  !4 4DKKdT2H %OODIIh'(rc |jdd}dt| dzz}tj||z}t j |j d}t|jdS#t$rYywxYw)z?Returns exp claim in seconds since epoch or None if unreadable..rW=zutf-8expN) splitr\base64urlsafe_b64decodejsonloadsdecodeintr&r()r)payload_segmentpaddingpayloadclaimss r_decode_jwt_exprzs++c*1-#o..23**?W+DEGNN7346::e$%% sA;A>> B  B i,)leeway_secondscRt|}|ytj|z|k\S)NT)rztime)r)r{rns r_is_access_token_expiredr~s* % C { 99; '3 ..rctjtdt|dd} |j }|j dk7sd|vr td|S#t $ri}Y-wxYw)Nr`) grant_type client_idr`)datarMrazDFailed to refresh fal auth token. Please run `fal auth login` again.)httpxpostAUTH0_TOKEN_URLAUTH0_CLIENT_IDrrr( status_coder,)r`response token_datas r_refresh_access_tokenrs{zz)(* H]]_ s"nJ&F% R     sA A$#A$ct5t\}}dddsyr t|s|St|}|j d|}|d}t5t ||ddd|S#1swY^xYw#1swY|SxYw)z Try to reuse tokens created by `fal auth login`. We share the same file layout as fal-cli: - refresh token on first line - optional cached access token on second line Nr`ra)rUrbr~rr&ri)r`rar new_refresh new_accesss r_load_bearer_token_from_loginrs :&7&9# |: 4\B&}5J..-@KN+J 4; 34 !::4 sA6 B6A?B ctjjddk(}|sztjdx}r t d|Stjdx}r(tjdx}rt d|d|St x}r t d|St x}r t d|Std ) z Return credentials using this priority: 1) FAL_KEY / FAL_KEY_ID+FAL_KEY_SECRET / Colab secret (unless FAL_FORCE_AUTH_BY_USER=1) 2) Tokens saved by `fal auth login` FAL_FORCE_AUTH_BY_USER1r"Key FAL_KEY_IDFAL_KEY_SECRET:Bearerz_No credentials found. Set FAL_KEY (or FAL_KEY_ID/FAL_KEY_SECRET) or login via `fal auth login`.)rBenvironr&rCr/r*rr,)force_user_authkeykey_idfal_key_secret colab_tokenbearers rfetch_auth_credentialsrsjjnn%=>#EO ))I& &3 &"5#. . ,//f/ ii(89 9N 9"5VHAn5E*FG G+- -[ -"5+6 6.00v0x00 !i rc~t}|jjdk7r td|jS)z Legacy helper kept for backwards compatibility. It only returns Key-based credentials; user-based auth (Bearer) will raise MissingCredentialsError so callers don't accidentally send it as a key. rzFKey credentials not found. Set FAL_KEY (or FAL_KEY_ID/FAL_KEY_SECRET).)rr0lowerr,r))auths rfetch_credentialsrs= " #D {{e#% T   ::r)r8bool)r8 Optional[str])r8r)r8z#tuple[Optional[str], Optional[str]])r`rrarr8None)r)rr8z Optional[int])r)rr{rur8r)r`rr8dict)r8r/r7)- __future__rrprrrBr} contextlibr dataclassesrpathlibr threadingrtypingrrr r#rr*r(r,r/rr&r<r= AUTH0_DOMAINrrrXrOhomerDrFrUrbrirzr~rrrrrrrrs!" %! ** ! #, i  $ 4 4 4zz~~ni8 ZZ^^$8F<.:QR \N,74#!499;'O"7") CF/086r