o b@sddlZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZddl mZmZmZdd lmZmZmZd Zd ZGd d d eZddZddZdS)Ngenerate_token)extract_params) to_native) OAuth1Request)SIGNATURE_HMAC_SHA1SIGNATURE_PLAINTEXTSIGNATURE_RSA_SHA1SIGNATURE_TYPE_HEADERSIGNATURE_TYPE_BODYSIGNATURE_TYPE_QUERY)sign_hmac_sha1 sign_rsa_sha1sign_plaintext)prepare_form_encoded_bodyprepare_headersprepare_request_uri_queryz!application/x-www-form-urlencodedzmultipart/form-datac @speZdZeeeeeeiZ e ddZ ddddddee ddf ddZ ddZd d Zd d Zdd dZddZdS) ClientAuthcCs||j|<dS)aExtend client signature methods. :param name: A string to represent signature method. :param sign: A function to generate signature. The ``sign`` method accept 2 parameters:: def custom_sign_method(client, request): # client is the instance of Client. return 'your-signed-string' Client.register_signature_method('custom-name', custom_sign_method) N)SIGNATURE_METHODS)clsnamesignra/var/www/secure340b-portal/env/lib/python3.10/site-packages/authlib/oauth1/rfc5849/client_auth.pyregister_signature_method%sz$ClientAuth.register_signature_methodNFc CsF||_||_||_||_||_||_| |_||_||_| |_ | |_ dSN) client_id client_secrettoken token_secret redirect_urisignature_methodsignature_typersa_keyverifierrealmforce_include_body) selfrrrr r!r$r%r"r#r&r'rrr__init__6s zClientAuth.__init__cCs4|j|j}|s tdt||||d}|||S)awGet an OAuth signature to be used in signing a request To satisfy `section 3.4.1.2`_ item 2, if the request argument's headers dict attribute contains a Host item, its value will replace any netloc part of the request argument's uri attribute value. .. _`section 3.4.1.2`: https://tools.ietf.org/html/rfc5849#section-3.4.1.2 zInvalid signature method.)bodyheaders)rgetr" ValueErrorr)r(methodurir+r*rrequestrrrget_oauth_signatureHs   zClientAuth.get_oauth_signaturecCshd|fd|fdd|jfd|jfg}|jr|d|jf|jr'|d|jf|jr2|d|jf|S) NZ oauth_nonceZoauth_timestamp)Z oauth_versionz1.0Zoauth_signature_methodZoauth_consumer_keyZ oauth_tokenZoauth_callbackZoauth_verifier)r"rrappendr!r%)r(nonce timestamp oauth_paramsrrrget_oauth_paramsYszClientAuth.get_oauth_paramscCs~|jtkrt|||jd}n,|jtkr+t|ddvr*t|p g}t||}t|d<n|jt kr6t ||}nt d|||fS)N)r& Content-Typez!Unknown signature type specified.) r#r rr&r CONTENT_TYPE_FORM_URLENCODEDr,rrr rr-)r(r/r+r*r5Z decoded_bodyrrr_renderis       zClientAuth._renderc Cs|durt}|durt}|durd}t|}|duri}|||}|||||\}}}|||||}|d|f|||||\}}}|||fS)a?Sign the HTTP request, add OAuth parameters and signature. :param method: HTTP method of the request. :param uri: URI of the HTTP request. :param body: Body payload of the HTTP request. :param headers: Headers of the HTTP request. :param nonce: A string to represent nonce value. If not configured, this method will generate one for you. :param timestamp: Current timestamp. If not configured, this method will generate one for you. :return: uri, headers, body Nr8Zoauth_signature)generate_noncegenerate_timestampstrr6r:r1r2) r(r.r/r+r*r3r4r5sigrrrrws   zClientAuth.signcCst|dd}|jtkrt}n|st|rt}t|vr,t|d<|||||\}}}n|jr;|||||\}}}n ||||d\}}}d}|||fS)zAdd OAuth parameters to the request. Parameters may be included from the body if the content-type is urlencoded, if no content type is set, a guess is made. r7r8)rr,r#r r9rrr')r(r.r/r+r* content_type_rrrprepares   zClientAuth.prepare)NN)__name__ __module__ __qualname__rrr rr rr classmethodrr r)r1r6r:rrArrrrrs&    #rcCstSrrrrrrr;sr;cCstttSr)r=inttimerrrrr<sr<)rGZauthlib.common.securityrZauthlib.common.urlsrZauthlib.common.encodingrwrapperr signaturerr r r r r rrr parametersrrrr9CONTENT_TYPE_MULTI_PARTobjectrr;r<rrrrs