~hddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddl mZddl mZddlmZdZdZGddZdZdZy)N) to_nativegenerate_token)extract_params)prepare_form_encoded_body)prepare_headers)prepare_request_uri_query)SIGNATURE_HMAC_SHA1)SIGNATURE_PLAINTEXT)SIGNATURE_RSA_SHA1)SIGNATURE_TYPE_BODY)SIGNATURE_TYPE_HEADER)SIGNATURE_TYPE_QUERY)sign_hmac_sha1)sign_plaintext) sign_rsa_sha1) OAuth1Requestz!application/x-www-form-urlencodedzmultipart/form-datac feZdZeeeeeeiZ e dZ ddddddee ddf dZ dZdZdZdZd Zy) ClientAuthc"||j|<y)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)clsnamesigns ^/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/authlib/oauth1/rfc5849/client_auth.pyregister_signature_methodz$ClientAuth.register_signature_method"s '+d#NFc ||_||_||_||_||_||_| |_||_||_| |_ | |_ yN) client_id client_secrettoken token_secret redirect_urisignature_methodsignature_typersa_keyverifierrealmforce_include_body) selfr!r"r#r$r%r(r)r&r'r*r+s r__init__zClientAuth.__init__4sW#* (( 0,    "4rc|jj|j}|s tdt ||||}|||S)axGet 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,methodurir0r/rrequests rget_oauth_signaturezClientAuth.get_oauth_signatureNsL%%))$*?*?@89 9$HD'""rcBd|fd|fdd|jfd|jfg}|jr|jd|jf|jr|jd|jf|j r|jd|j f|S) N oauth_nonceoauth_timestamp) oauth_versionz1.0oauth_signature_methodoauth_consumer_key oauth_tokenoauth_callbackoauth_verifier)r&r!r#appendr%r))r,nonce timestamp oauth_paramss rget_oauth_paramszClientAuth.get_oauth_params_s E "  * $ %t'<'< = !4>> 2   ::    ; <      !143D3D E F ==   !14== A BrcZ|jtk(rt|||j}n{|jtk(r=t |j ddvrPt|xsg}t||}t |d<n+|jtk(r t||}n td|||fS)N)r* Content-Typez!Unknown signature type specified.) r'rr r*rCONTENT_TYPE_FORM_URLENCODEDr1rrrr r2)r,r4r0r/rC decoded_bodys r_renderzClientAuth._renderos   "7 7%lG4::NG  $7 7+w{{>2/NN-d39r 0|L*F'  $8 8+L#>C@A AGT!!rct}t}|d}t|}|i}|j||}|rp|j dt k7rXt jtj|j}|jd|jdf|j||||\}}}|j||||} |jd| f|j||||\}}}|||fS)a5Sign 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. :return: uri, headers, body rrFoauth_body_hashzutf-8oauth_signature)generate_noncegenerate_timestampstrrDr1rHbase64 b64encodehashlibsha1digestr@decoderJr6) r,r3r4r0r/rArBrCrLsigs rrzClientAuth.sign}s  &( <D N ?G,,UI>  GKK/3OO$..w||D/A/H/H/JKO   !2O4J4J74S T U!\\#wlKWd&&vsGTB.45!\\#wlKWdGT!!rcnt|jdd}|jtk(rt}n|st |rt}t|vr"t|d<|j ||||\}}}n?|jr|j ||||\}}}n|j |||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. rFrGr)rr1r'rrHrrr+)r,r3r4r0r/ content_type_s rpreparezClientAuth.prepares !^R!@A   "5 57L."67L '< 7&BGN #!%63!F C$  $ $!%63!F C$#iiWcBOC!DGT!!r)__name__ __module__ __qualname__r rr rr rr classmethodrrr-r6rDrJrr[rrrrsl^M^ ++(,, 54#" """H"rrctSr rr`rrrNrNs  rcNtttjSr )rPinttimer`rrrOrOs s499;  r)rQrSrdauthlib.common.encodingrauthlib.common.securityrauthlib.common.urlsr parametersrr r signaturer r r rrrrrrwrapperrrHCONTENT_TYPE_MULTI_PARTrrNrOr`rrrlsa  -2.1'1**)*,+%%$"B/\"\"~!r