i/lUddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl Z ddl mZddlmZdd lmZdd lmZmZdd lmZejd Zejd ZdZiaded<ddiZGddeZdddfd9d"Z d:d&Z!d;d)Z"dd2Z%d>d3Z&d?d7Z'd@d8Z(dS)A) annotationsN)Callable)AnyLiteral)parse_qs)Session) OAuth1Session)http) OAuth1Token OAuth2Token)GarthExceptionzname="_csrf"\s+value="(.+?)"z(.+?)z5https://thegarth.s3.amazonaws.com/oauth_consumer.jsonzdict[str, str]OAUTH_CONSUMERz User-Agentz%com.garmin.android.apps.connectmobilec$eZdZ ddfd ZxZS)GarminOAuth1SessionNparentSession | Nonec Hts+tjtat jtdtdfi||;|d|jd|j |_ |j |_ dSdS)N consumer_keyconsumer_secretzhttps://) rrequestsgetOAUTH_CONSUMER_URLjsonsuper__init__mountadaptersproxiesverify)selfrkwargs __class__s D/root/projects/butler/venv/lib/python3.11/site-packages/garth/sso.pyrzGarminOAuth1Session.__init__s E%\*<==BBDDN > * , -       JJz6?:#> ? ? ?!>DL -DKKK  N)rr)__name__ __module__ __qualname__r __classcell__)r#s@r$rrsG"&(((((((((((r%rc tdS)Nz MFA code: )inputr%r$r.5s% *=*=r%Fclienthttp.Client | None prompt_mfaCallable | None return_on_mfaboolemailstrpasswordreturnMtuple[OAuth1Token, OAuth2Token] | tuple[Literal['needs_mfa'], dict[str, Any]]c n|p tj}d|jd}|d}tdd|}i|t|||||}|dd | |dd |d t |jj} |dd |d t||d| t|jj} d| vr5|s|d||dfSt|||t|jj} | dkrtd| t|S)a9Login to Garmin Connect. Args: email: Garmin account email password: Garmin account password client: Optional HTTP client to use prompt_mfa: Callable that prompts for MFA code. Returns on MFA if None. return_on_mfa: If True, returns dict with MFA info instead of prompting Returns: If return_on_mfa=False (default): Tuple[OAuth1Token, OAuth2Token]: OAuth tokens after login If return_on_mfa=True and MFA required: dict: Contains needs_mfa and client_state for resume_login() https://sso.z/ssoz/embedz gauth-widgettrue)id embedWidget gauthHost)r?servicesourceredirectAfterAccountLoginUrlredirectAfterAccountCreationUrlsso /sso/embed)paramsz /sso/signinT)rFreferrer)usernamer7embed_csrfrFrGdataMFAN needs_mfa) signin_paramsr/SuccesszUnexpected title: ) r r/domaindictrget_csrf_token last_resptextpost get_title handle_mfar_complete_login) r5r7r/r1r3SSO SSO_EMBEDSSO_EMBED_PARAMS SIGNIN_PARAMS csrf_tokentitles r$loginr`0s4 "t{F - , , ,CI     )2,5     M JJul+;J<<< JJ     0 566J KK          f&+ , ,E ~~  J.!. !!  6=*555&*/00 9%99::: 6 " ""r%ticket http.Clientr cdt|j}d|jd}d|jd}|d|d|d}||t|j }|t|j}d | D}td d |ji|S) N)rhttps://connectapi./oauth-service/oauth/r;rEzpreauthorized?ticket=z &login-url=z&accepts-mfa-tokens=true)headerstimeoutc&i|]\}}||dS)rr-).0kvs r$ z$get_oauth1_token..s" 0 0 0AQ! 0 0 0r%rQr-) rsessrQr USER_AGENTrgraise_for_statusrrUitemsr ) rar/rmbase_url login_urlurlrespparsedtokens r$get_oauth1_tokenrws fk 2 2 2DIV]IIIH8v}888I  # #& # #Y # # # 88    D   di F 0 0 0 0 0E  5 5fm 5u 5 55r%oauth1r cxt|j|j|j}|jrt |jni}d|jd}|d}itddi}|||||j }| | }td it|S) N)resource_owner_keyresource_owner_secretr) mfa_tokenrdrezexchange/user/2.0z Content-Typez!application/x-www-form-urlencoded)rfrLrgr-)r oauth_tokenoauth_token_secretrmr|rRrQrnrVrgrorr set_expirations) rxr/rmrLrqrsrfrtrvs r$exchangers !-$7{   D 06/? G4&* + + + +RDIV]IIIH  ( ( (C  > ?G 99     D   IIKKE  0 0// 0 00r%rOrRrNonec t|jj}tj|rt j|}n |}|dd|d|d|dddS)NrDz /sso/verifyMFA/loginEnterMfaCodeTr<setupEnterMfaCode)zmfa-coderIrJfromPagerK)rSrTrUinspectiscoroutinefunctionasynciorunrV)r/rOr1r^mfa_codes r$rXrXs  0 566J":.. ;zz||,,:<< KK * +        r%rvcttj|dz|d<ttj|dz|d<|S)N expires_in expires_atrefresh_token_expires_inrefresh_token_expires_at)inttime)rvs r$rrsTdikkE,,??@@E,(+ e677))E $% Lr%htmlct|}|std|dS)NzCouldn't find CSRF tokenr )CSRF_REsearchrgrouprms r$rSrSs:tA 97888 771::r%ct|}|std|dS)NzCouldn't find titler )TITLE_RErrrrs r$rWrWs:A 42333 771::r% client_statertuple[OAuth1Token, OAuth2Token]cj|d}|d}t||fdt|S)a Complete login after MFA code is provided. Args: client_state: The client state from login() when MFA was needed mfa_code: The MFA code provided by the user Returns: Tuple[OAuth1Token, OAuth2Token]: The OAuth tokens after login r/rOcSr&r-)rsr$r.zresume_login..shr%)rXrY)rrr/rOs ` r$ resume_loginrsC( #F 1Mv}&6&6&6&6777 6 " ""r%ctjd|jj}|st d|d}t ||}t||}||fS)zComplete the login process after successful authentication. Args: client: The HTTP client Returns: Tuple[OAuth1Token, OAuth2Token]: The OAuth tokens zembed\?ticket=([^"]+)"z Couldn't find ticket in responser )rerrTrUrrrwr)r/rrarxoauth2s r$rYrYsr +V-=-BCCA   .   WWQZZF ff - -F ff % %F 6>r%) r/r0r1r2r3r4r5r6r7r6r8r9)rar6r/rbr8r )rxr r/rbr8r )r/rbrOrRr1rr8r)rvrRr8rR)rr6r8r6)rrRrr6r8r)r/rbr8r)) __future__rrrrrcollections.abcrtypingrr urllib.parserrrrequests_oauthlibr r auth_tokensr r excrcompilerrrr__annotations__rnrr`rwrrXrrSrWrrYr-r%r$rs,""""""" $$$$$$!!!!!!++++++11111111 "*4 5 5 2:- . .L!#####C D (((((-(((2"&"="= W#W#W#W#W#t6666&11110,####$r%