([fddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZmZmZddl mZmZmZdd lmZmZmZd Zd ZGd d ZdZdZdS)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 neZdZeeeeeeiZ e dZ ddddddee ddf dZ dZdZdZdZd ZdS) ClientAuthc||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)clsnamesigns V/var/www/piapp/venv/lib/python3.11/site-packages/authlib/oauth1/rfc5849/client_auth.pyregister_signature_methodz$ClientAuth.register_signature_method's'+d###NFc ||_||_||_||_||_||_| |_||_||_| |_ | |_ dSN) 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__8s] #* (( 0,    "4rc|j|j}|stdt ||||}|||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,methodurir0r/rrequests rget_oauth_signaturezClientAuth.get_oauth_signatureJs\%))$*?@@ :899 9$HHHtD'"""rc d|fd|fdd|jfd|jfg}|jr|d|jf|jr|d|jf|jr|d|jf|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   : =    ; < < <   G   !143D E F F F = C   !14= A B B Brcn|jtkrt|||j}n|jtkrIt |ddvr+t|pg}t||}t |d<n0|jtkrt||}ntd|||fS)N)r* Content-Typez!Unknown signature type specified.) r'r rr*r CONTENT_TYPE_FORM_URLENCODEDr1rrrrr2)r,r4r0r/rC decoded_bodys r_renderzClientAuth._renderks  "7 7 7%lG4:NNNGG  $7 7 7+w{{>2/N/NNN-d339r 0|LL*F'  $8 8 8+L#>>CC@AA AGT!!rcpt}t}|d}t|}|i}|||}|r|dt krbt jtj | }| d| df| ||||\}}}|||||} | d| f| ||||\}}}|||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 NrrFoauth_body_hashzutf-8oauth_signature)generate_noncegenerate_timestampstrrDr1rHbase64 b64encodehashlibsha1digestr@decoderJr6) r,r3r4r0r/rArBrCrLsigs rrzClientAuth.signys<  &(( <D NN ?G,,UI>>   VGKK//3OOO$.w|D/A/A/H/H/J/JKKO   !2O4J4J74S4S T U U U!\\#wlKKWd&&vsGTBB.4555!\\#wlKKWdGT!!rct|dd}|jtkrt}n|st |rt}t|vr't|d<|||||\}}}nB|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. rFrGr)rr1r'r rHrrr+)r,r3r4r0r/ content_type_s rpreparezClientAuth.prepares !^R!@!@AA  "5 5 57LL 8."6"6 87L '< 7 7&BGN #!%63!F!F C$$  $ !%63!F!F C$$#iiWcBBOC!DGT!!r)__name__ __module__ __qualname__r rr rr rr classmethodrr r-r6rDrJrr[rrrr s^M^ ++[+ 15$"D4"5 5 5555$###" " " """""""H"""""rrctSr rr`rrrNrNs   rc\tttjSr )rPinttimer`rrrOrOs s49;;  r)rdrQrSauthlib.common.securityrauthlib.common.urlsrauthlib.common.encodingrwrapperr signaturer r r r r rrrr parametersrrrrHCONTENT_TYPE_MULTI_PARTrrNrOr`rrrls 222222......------""""""  C/S"S"S"S"S"S"S"S"l!!!!!r