o b@sBddZddZddZddZdd Zd d Z dd dZd S)cfdd}|S)zCreate an ``query_client`` function that can be used in authorization server and resource protector. :param session: SQLAlchemy session :param model_class: Client class cs}|j|dS)N) client_idquery filter_byfirst)rq model_classsessioni/var/www/secure340b-portal/env/lib/python3.10/site-packages/authlib/integrations/sqla_oauth1/functions.py query_clients z.create_query_client_func..query_clientr )r r r r rr create_query_client_funcsrcr)zCreate an ``query_token`` function that can be used in resource protector. :param session: SQLAlchemy session :param model_class: TokenCredential class cs}|j||dS)N)r oauth_tokenr)rrrrr r query_tokens z,create_query_token_func..query_tokenr )r r rr rr create_query_token_funcsrcsjfdd}fdd}fdd}fdd}|d ||d ||d ||d |d S)zRegister temporary credential related hooks to authorization server. :param authorization_server: AuthorizationServer instance :param session: SQLAlchemy session :param model_class: TemporaryCredential class cs.||d|d|d}||S)Nroauth_token_secret)rrrZoauth_callback)addcommit)tokenrZ redirect_uriitemrr r create_temporary_credential%s zHregister_temporary_credential_hooks..create_temporary_credentialcsj|d}|S)Nrrrrrr r get_temporary_credential0szEregister_temporary_credential_hooks..get_temporary_credentialcs*j|d}|jdddS)NrF)Zsynchronize_session)rrdeleterrrr r delete_temporary_credential4s  zHregister_temporary_credential_hooks..delete_temporary_credentialcs*||||_||SN) set_user_id get_user_idZoauth_verifierrr)Z credentialZ grant_userZverifier)r r r create_authorization_verifier9s  zJregister_temporary_credential_hooks..create_authorization_verifierrrrr N register_hook)authorization_serverr r rrrr r rr #register_temporary_credential_hookss    r$cr)zCreate an ``exists_nonce`` function that can be used in hooks and resource protector. :param session: SQLAlchemy session :param model_class: TimestampNonce class cs^jj|||d}|r|j|d}|}|rdS||||d}|dS)N)nonce timestamprrT)r%r&rrF)rr%rrrr)r%r&rrrrvrrr r exists_nonceQs&   z.create_exists_nonce_func..exists_noncer )r r r(r rr create_exists_nonce_funcJsr)cCst||}|d|dS)zRegister nonce related hooks to authorization server. :param authorization_server: AuthorizationServer instance :param session: SQLAlchemy session :param model_class: TimestampNonce class r(N)r)r")r#r r r(r r r register_nonce_hooksis r*csfdd}|d|dS)zRegister token credential related hooks to authorization server. :param authorization_server: AuthorizationServer instance :param session: SQLAlchemy session :param model_class: TokenCredential class cs>|d|d|d}||||S)Nrr)rrr)Z get_client_idrrrr)rZtemporary_credentialrrr r create_token_credential|s z@register_token_credential_hooks..create_token_credentialr+Nr!)r#r r r+r rr register_token_credential_hooksts r,NcCs<t||||durt||||durt|||dSdSr)r,r$r*)r#r Ztoken_credential_modelZtemporary_credential_modelZtimestamp_nonce_modelr r r register_authorization_hookssr-)NN)rrr$r)r*r,r-r r r r s .