Ld&ddlZddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z ddl mZmZmZmZmZmZddl mZmZmZdd lmZmZmZd Zd ZGd d eZdZdZ dS)Ngenerate_token)extract_params) to_nativeto_bytes to_unicode) 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 HF:\djangOuth\env\Lib\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,    "4r c|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.methodurir2r1rrequests rget_oauth_signaturezClientAuth.get_oauth_signatureJs\%))$*?@@ :899 9$HHHtD'"""r c 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 Br cn|jtkrt|||j}n|jtkrIt |ddvr+t|pg}t||}t |d<n0|jtkrt||}ntd|||fS)N)r, Content-Typez!Unknown signature type specified.) r)rrr,rCONTENT_TYPE_FORM_URLENCODEDr3rrrrr4)r.r6r2r1rE 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!!r cpt}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 Nr rHoauth_body_hashzutf-8oauth_signature)generate_noncegenerate_timestampstrrFr3rJbase64 b64encodehashlibsha1digestrBdecoderLr8) r.r5r6r2r1rCrDrErNsigs 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!!r ct|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. rHrIr )rr3r)rrJrrr-)r.r5r6r2r1 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 classmethodrrr/r8rFrLrr]r rrr s^M^ ++[+ 15$"D4"5 5 5555$###" " " """""""H"""""r rctSr"rrbr rrPrPs   r c\tttjSr")rRinttimerbr rrQrQs s49;;  r )!rfrSrUauthlib.common.securityrauthlib.common.urlsrauthlib.common.encodingrrrwrapperr signaturer r r rrrrrr parametersrrrrJCONTENT_TYPE_MULTI_PARTobjectrrPrQrbr rros 222222......CCCCCCCCCC""""""  C/S"S"S"S"S"S"S"S"l!!!!!r