U #iW@s ddlmZddlmZddlmZmZddlmZddl m Z ddl m Z m Z m Z mZmZmZmZddlmZmZddlmZd d lmZmZmZmZd d lmZd d lmZm Z d d l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.GdddZ/dS)) annotations)partial)dumpsloads)Timer)time)AnyCallableDictOptionalTupleUnioncast)parse_qsurlparse)uuid4)COOKIE_OPTIONSDEFAULT_HEADERS GOTRUE_URL STORAGE_KEY)APIError) model_dumpmodel_validate)AuthChangeEvent CookieOptionsProviderSession SubscriptionUserUserAttributesUserAttributesDict) SyncGoTrueAPI)SyncMemoryStorageSyncSupportedStoragec @s`eZdZeiddeeeddddd dddddd d ddd d d ddZddddZ d dddZ d dddZ d dddZ ddddddd d d d dddddZ dddddddddd d d d d d d dd!d" d#d$Zdd%ddd dd&d'd(Zd)dd*d+Zd,dd-d.Zd/dd0d1Zd2d3d4d5d6Zdd/d7d8d9Zdd/d:d;d<Zdd=ddd/d>d?d@Zd ddAdBZdd dCdDdEZdFdGdHdIdJZddd d/dKdLdMZddd/dNdOdPZdQd d ddRdSdTZdUddVdWZd ddXdYZd ddZd[Zdd\d d/d7d]d^Z d_d d`dadbZ!d/d dcdddeZ"d/d dcdfdgZ#d ddhdiZ$djd dkdldmZ%dS)nSyncGoTrueClientTNF) urlheadersauto_refresh_tokenpersist_session local_storagecookie_optionsapireplace_default_headersverifyproxystrzDict[str, str]boolr%rzOptional[SyncGoTrueAPI]z Optional[str]None) r'r(r)r*r+r,r-r.r/r0returnc Csp|drtdi|_d|_d|_d|_||_||_||_|rDint } || ||| | d} |pht f| |_ dS)aDCreate a new client url : str The URL of the GoTrue server. headers : Dict[str, str] Any additional headers to send to the GoTrue server. auto_refresh_token : bool Set to "true" if you want to automatically refresh the token before expiring. persist_session : bool Set to "true" if you want to automatically save the user session into local storage. local_storage : SupportedStorage The storage engine to use for persisting the session. cookie_options : CookieOptions The options for the cookie. verify: bool Verify SSL, True by default, False disables verification. proxy: str HTTP Proxy string or None, None by default, None disables proxy. zhttp://z`Warning: DO NOT USE HTTP IN PRODUCTION FOR GOTRUE EVER! GoTrue REQUIRES HTTPS to work securely.N)r'r(r,r/r0) startswithprintstate_change_emittersrefresh_token_timer current_usercurrent_sessionr)r*r+rr#r-) selfr'r(r)r*r+r,r-r.r/r0Zempty_or_default_headersargsr=7/tmp/pip-unpacked-wheel-b8wmx3ip/gotrue/_sync/client.py__init__s&#  zSyncGoTrueClient.__init__)r4cCs|SNr=r;r=r=r> __enter__VszSyncGoTrueClient.__enter__cCs |dSr@)close)r;Zexc_tZexc_vexc_tbr=r=r>__exit__YszSyncGoTrueClient.__exit__cCs|jdSr@)r-rCrAr=r=r>rC\szSyncGoTrueClient.closecCs||dS)z/Recover the current session from local storage.N)_recover_session_recover_and_refreshrAr=r=r> init_recover_szSyncGoTrueClient.init_recover)emailphonepassword redirect_todatazOptional[Dict[str, Any]]zUnion[Session, User])rIrJrKrLrMr4cCs||r&|r&|jj||||d}n2|rB|rB|jj|||d}n|sPtdntdt|tr||j|d|jt j d|S)a7Creates a new user. If email and phone are provided, email will be used and phone will be ignored. Parameters --------- email : Optional[str] The user's email address. phone : Optional[str] The user's phone number. password : Optional[str] The user's password. redirect_to : Optional[str] A URL or mobile address to send the user to after they are confirmed. data : Optional[Dict[str, Any]] Optional user metadata. Returns ------- response : Union[Session, User] A logged-in session if the server has "autoconfirm" ON A user if the server has "autoconfirm" OFF Raises ------ APIError If an error occurs. )rIrKrLrM)rJrKrMz(Password must be defined, can't be None.z3Email or phone must be defined, both can't be None.sessionevent) _remove_sessionr-Zsign_up_with_emailZsign_up_with_phone ValueError isinstancer _save_session_notify_all_subscribersr SIGNED_IN)r;rIrJrKrLrMresponser=r=r>sign_upds*$   zSyncGoTrueClient.sign_up)rIrJrK refresh_tokenproviderrLscopes create_userzOptional[Provider]zOptional[Union[Session, str]]) rIrJrKrZr[rLr\r]r4c Cs||r4|r"|j|||d} q|jj||d} n`|r^|rL|j||d} q|jj||d} n6|rv|j|d|j} n|r|j|||d} nt d| S)a.Log in an existing user, or login via a third-party provider. If email and phone are provided, email will be used and phone will be ignored. Parameters --------- email : Optional[str] The user's email address. phone : Optional[str] The user's phone number. password : Optional[str] The user's password. refresh_token : Optional[str] A valid refresh token that was returned on login. provider : Optional[Provider] One of the providers supported by GoTrue. redirect_to : Optional[str] A URL or mobile address to send the user to after they are confirmed. scopes : Optional[str] A space-separated list of scopes granted to the OAuth application. Returns ------- response : Optional[Union[Session, str]] If only email are provided between the email and password, None is returned and send magic link to email If email and password are provided, a logged-in session is returned. If only phone are provided between the phone and password, None is returned and send message to phone If phone and password are provided, a logged-in session is returned. If refresh_token is provided, a logged-in session is returned. If provider is provided, an redirect URL is returned. Otherwise, error is raised. Raises ------ APIError If an error occurs. rIrKrL)rIr]rJrK)rJr]rZr[rLr\zLEmail, phone, refresh_token, or provider must be defined, all can't be None.) rR_handle_email_sign_inr-Zsend_magic_link_email_handle_phone_sign_inZsend_mobile_otp_call_refresh_tokenr:_handle_provider_sign_inrS) r;rIrJrKrZr[rLr\r]rXr=r=r>sign_ins@8 zSyncGoTrueClient.sign_in)rL)rJtokenrLr4cCsB||jj|||d}t|tr>|j|d|jtjd|S)a^Log in a user given a User supplied OTP received via mobile. Parameters ---------- phone : str The user's phone number. token : str The user's OTP. redirect_to : Optional[str] A URL or mobile address to send the user to after they are confirmed. Returns ------- response : Union[Session, User] A logged-in session if the server has "autoconfirm" ON A user if the server has "autoconfirm" OFF Raises ------ APIError If an error occurs. )rJrgrLrNrP) rRr-Zverify_mobile_otprTrrUrVrrW)r;rJrgrLrXr=r=r> verify_otps  zSyncGoTrueClient.verify_otpzOptional[User]cCs|jS)z4Returns the user data, if there is a logged in user.)r9rAr=r=r>user&szSyncGoTrueClient.userzOptional[Session]cCs|jS)z8Returns the session data, if there is an active session.)r:rAr=r=r>rO*szSyncGoTrueClient.sessionrcCs|jstd|S)zForce refreshes the session. Force refreshes the session including the user data incase it was updated in a different session. Not logged in.)r:rSrdrAr=r=r>refresh_session.sz SyncGoTrueClient.refresh_sessionz)Union[UserAttributesDict, UserAttributes]r) attributesr4cCsd|jstdt|tr$tf|}n|}|jj|jj|d}||j_|j |jd|j t j d|S)aUpdates user data, if there is a logged in user. Parameters ---------- attributes : UserAttributesDict | UserAttributes Attributes to update, could be: email, password, email_change_token, data Returns ------- response : User The updated user data. Raises ------ APIError If an error occurs. rj)jwtrlrNrP) r:rSrTdictr r-Z update_user access_tokenrirUrVrZ USER_UPDATED)r;rlZattributes_to_updaterXr=r=r>update8s  zSyncGoTrueClient.update)rZr4cCs,|jj|d}|j|d|jtjd|S)aTSets the session data from refresh_token and returns current Session Parameters ---------- refresh_token : str A JWT token Returns ------- response : Session A logged-in session Raises ------ APIError If an error occurs. r`rNrP)r-refresh_access_tokenrUrVrrWr;rZrXr=r=r> set_session[s zSyncGoTrueClient.set_session)ror4c CsTt|ddddddd}|jrD|jj|_|jj|_|jj|_|jj|_|j|d|S)a}Overrides the JWT on the current client. The JWT will then be sent in all subsequent network requests. Parameters ---------- access_token : str A JWT token Returns ------- response : Session A logged-in session Raises ------ APIError If an error occurs. ZbearerNro token_typeri expires_in expires_atrZprovider_tokenrN)rr:rvrwrZrxrU)r;rorOr=r=r>set_authrs       zSyncGoTrueClient.set_auth) store_session)r'rzr4c Cst|}t|j}|d}|d}|d}|d}|d} |rVt|dd|rb|dsltdd|rx|dstd d| r| dstd d|r|dstd dzttt|d} Wntk rtd dYnX|j j |dd } |d} t |d| d| t|d| |d| r4| dndd} |r|j | d|d}|j tjd|r|ddkr|j tjd| S)aGets the session data from a URL string. Parameters ---------- url : str The URL string. store_session : bool Optionally store the session in the browser Returns ------- response : Session A logged-in session Raises ------ APIError If an error occurs. error_descriptionrorvrZrurizNo access_token detected.zNo refresh_token detected.zNo token_type detected.zNo expires_in detected.zInvalid expires_in.rmrxNrtrNtyperPZrecovery)rrquerygetrroundrintrSr-Zget_userrrUrVrrWZPASSWORD_RECOVERY)r;r'rzrMr~r{rorvrZrurwrXrxrOZ recovery_moder=r=r>get_session_from_urlsN                   z%SyncGoTrueClient.get_session_from_urlcCs>d}|jr|jj}||jtjd|r:|jj|ddS)zLog the user out.NrPr|)r:rorRrVrZ SIGNED_OUTr-sign_out)r;ror=r=r>rszSyncGoTrueClient.sign_out)idr4cCs|j|dS)z Unsubscribe from a subscription.N)r7pop)r;rr=r=r> _unsubscribeszSyncGoTrueClient._unsubscribez4Callable[[AuthChangeEvent, Optional[Session]], None]r)callbackr4cCs0t}t||t|j|jdd}||j|j<|S)aReceive a notification every time an auth event happens. Parameters ---------- callback : Callable[[AuthChangeEvent, Optional[Session]], None] The callback to call when an auth event happens. Returns ------- subscription : Subscription A subscription object which can be used to unsubscribe itself. Raises ------ APIError If an error occurs. )r)rrZ unsubscribe)rrrrhexr7)r;rZ unique_idZ subscriptionr=r=r>on_auth_state_changes z%SyncGoTrueClient.on_auth_state_change)rIrKrLr4cCs0|jj|||d}|j|d|jtjd|S)z Sign in with email and password.r^rNrP)r-Zsign_in_with_emailrUrVrrW)r;rIrKrLrXr=r=r>rbs z&SyncGoTrueClient._handle_email_sign_in)rJrKr4cCs.|jj||d}|j|d|jtjd|S)z Sign in with phone and password.r_rNrP)r-Zsign_in_with_phonerUrVrrW)r;rJrKrXr=r=r>rcs z&SyncGoTrueClient._handle_phone_sign_inr)r[rLr\r4cCs|jj|||dS)zSign in with provider.ra)r-Zget_url_for_provider)r;r[rLr\r=r=r>res z)SyncGoTrueClient._handle_provider_sign_inz"Optional[Tuple[Session, int, int]]cCsv|jt}|sdSt|}|d}|d}|rrt|trr|rrt|trrtt |}t|}t t }|||fSdS)zRecover common logicNrOrw) r+Zget_itemrrrrTrrnrrrr)r;jsonrMZ session_rawZexpires_at_rawrOrwtime_nowr=r=r>_recover_common*s$     z SyncGoTrueClient._recover_commoncCs@|}|sdS|\}}}||kr<|j|d|jtjddS)z-Attempts to get the session from LocalStorageNrNrP)rrUrVrrWr;resultrOrwrr=r=r>rF=s  z!SyncGoTrueClient._recover_sessioncCs|}|sdS|\}}}||kr`|jr`|jr`z|j|jdWqtk r\|YqXn6||ksr|rr|js||n|j|d|jt j ddS)z4Recovers the session from LocalStorage and refreshesNr`rNrP) rr)rZrdrrRrirUrVrrWrr=r=r>rGGs   z%SyncGoTrueClient._recover_and_refreshr`cCs`|dkr |jr|jj}ntd|jjtt|d}|j|d|jt j d|jt j d|S)Nz2No current session and refresh_token not supplied.r`rNrP) r:rZrSr-rqrr1rUrVrZTOKEN_REFRESHEDrWrrr=r=r>rdXs  z$SyncGoTrueClient._call_refresh_tokenr)rQr4cCs"|jD]}|||jq dS)z0Notify all subscribers that auth event happened.N)r7valuesrr:)r;rQvaluer=r=r>rVdsz(SyncGoTrueClient._notify_all_subscribers)rOr4cCsd||_|j|_|jrHtt}|j|}|dkr4dnd}|j||d|jr`|jr`|j|ddS)zSave session to client.<g?)rrNN) r:rir9rwrr_start_auto_refresh_tokenr*_persist_session)r;rOrZ expire_inZrefresh_duration_before_expiresr=r=r>rUis   zSyncGoTrueClient._save_sessioncCs*t||jd}|jtt|tddS)N)rOrw)default)rrwr+Zset_itemrrr1)r;rOrMr=r=r>rwsz!SyncGoTrueClient._persist_sessioncCs,d|_d|_|jr|j|jtdS)zRemove the session.N)r:r9r8cancelr+Z remove_itemrrAr=r=r>rR{s  z SyncGoTrueClient._remove_sessionfloat)rr4cCs>|jr|j|dks|js"dSt||j|_|jdS)Nr)r8rr)rrdstart)r;rr=r=r>rs  z*SyncGoTrueClient._start_auto_refresh_token)&__name__ __module__ __qualname__rr$rZ parse_objrr?rBrErCrHrYrfrhrirOrkrprsryrrrrrbrcrerrFrGrdrVrUrrRrr=r=r=r>r&sn&9@"b( #(A   r&N)0 __future__r functoolsrrrr threadingrrtypingrr r r r r r urllib.parserruuidr constantsrrrr exceptionsrZhelpersrrtypesrrrrrrr r!r-r#Zstorager$r%r&r=r=r=r>s    $  (