U ʼbw@s^ddlZddlmZddlmZddlmZGdddZGdddeZGd d d eZdS) N) CountryCode)convert_response)prefix_country_codec@sXeZdZdZdZdZdZdZddZdd Z d d Z d d Z ddZ ddZ ddZdS) BaseClientz.Base class for both OTPClient & AsyncOTPClientzhttps://api.msg91.comz/api/sendotp.phpz/api/retryotp.phpz/api/verifyRequestOTP.phpcCs ||_dSN)auth_key)selfrr 4/tmp/pip-unpacked-wheel-c1x4am86/msg91_otp/client.py__init__szBaseClient.__init__cCs |j|jSr)base_url otp_endpointrr r r get_otp_urlszBaseClient.get_otp_urlcCs |j|jSr)r otp_retry_endpointrr r r get_resend_otp_urlszBaseClient.get_resend_otp_urlcCs |j|jSr)r verify_otp_endpointrr r r get_verify_otp_urlszBaseClient.get_verify_otp_urlcKsttjt|}||jd}d|kr2|d|d<d|krH|d|d<d|kr|d|d<d|krd|dkr|ddd|d<n(d|krd|dkr|dd|d<d|kr|d|d<|S) NmobileauthkeymessageZsenderotpz##OTP##z ##OTP##Z otp_length)rrINDIAstrrgetreplacestriprreceiverkwargspayloadr r r get_otp_payloads"zBaseClient.get_otp_payloadcKs ttjt|}||jd}tS)Nr)rrrrrZpaylaodrr r r get_resend_otp_payload6s z!BaseClient.get_resend_otp_payloadcCs.ttjt|}|j||d}ddi}||fS)N)rrrz Content-typez!application/x-www-form-urlencoded)rrrrr)rr otp_valuer"headersr r r get_verify_payload_headers=sz%BaseClient.get_verify_payload_headersN)__name__ __module__ __qualname____doc__r r rrr rrrr#r$r'r r r r rsrc@s(eZdZdZddZddZddZdS) OTPClientzJSynchronous OTP client Use this class for blocking request to API cKs4|}|j|f|}tj||d}t||d}|S)aRequest the Service to send OTP message to given number Args: receiver(str): 10 digit mobile number with country code of reciever message(str, optional): text message to send along with OTP sender(str, optional): the name to appear in SMS as sender otp (int, optional): the opt value to send, if not service will generate otp_length (int, optional): the length of otp. default 4, max 9 Returns: a response object with status and status message params) identifier)rr#httpxpostr)rr r!otp_urlr"service_response _responser r r send_otpKs  zOTPClient.send_otpcKs0|}|j|f|}tj||d}t|}|S)zResend OTP request Args: receiver(str): 10 digit mobile number with country code of receiver Returns: a response object with status and status message r-)rr$r0r1r)rr r! retry_urlr"r3r4r r r resend_otp]s zOTPClient.resend_otpcCs4|}|||\}}tj|||d}t|}|S)aRequest to verify OTP with given mobile number Args: mobile(str): mobile number to verify otp otp_value(int): the otp value to verify against Returns: a response object with status and status message r.r&)rr'r0r1r)rrr% verify_urlr"r&r3r4r r r verify_otpjs zOTPClient.verify_otpNr(r)r*r+r5r7r:r r r r r,Gs r,c@s(eZdZdZddZddZddZdS) AsyncOTPClientzOAsynchronous OTP client Use this class for non-blocking request to API c sX|}|j|f|}t4IdH}|j||dIdH}W5QIdHRXt|}|SNr-)rr#r0 AsyncClientr1r)rr r!r2r"clientr3r4r r r r5}s $zAsyncOTPClient.send_otpc sX|}|j|f|}t4IdH}|j||dIdH}W5QIdHRXt|}|Sr=)rZget_resend_payloadr0r>r1r)rr r!r6r"r?r3r4r r r r7s $zAsyncOTPClient.resend_otpc s\|}|||\}}t4IdH}|j|||dIdH}W5QIdHRXt|}|S)Nr8)rr'r0r>r1r) rrr%r9r"r&r?r3r4r r r r:s &zAsyncOTPClient.verify_otpNr;r r r r r<ysr<) r0Zmsg91_otp.country_coderZmsg91_otp.utilsrrrr,r<r r r r s    ?2