a j@s<ddlZddlZddlZddlmZmZGdddeZdS)N)AnyOptionalc@sreZdZdZdejddfeeee ee eddddZ eeddd Z e d d d Z edeee dddZdS)OTPz& Base class for OTP handlers. N)sdigitsdigestnameissuerreturncCs6||_|dkrtd||_||_|p(d|_||_dS)N z!digits must be no greater than 10ZSecret)r ValueErrorrsecretr r )selfrrrr r r/E:\soccer\ml_env\Lib\site-packages\pyotp/otp.py__init__ s z OTP.__init__)inputr cCs|dkrtdt||||j}t|}|dd@}||d@d>||dd@d >B||d d@d >B||d d@B}td |d|j}||j dS)z :param input: the HMAC counter value to use as the OTP input. Usually either the counter, or the computed integer based on the Unix timestamp rzinput must be positive integerld( r N) r hmacnew byte_secretint_to_bytestringr bytearraystrr)rrZhasherZ hmac_hashoffsetcodeZstr_coderrr generate_otps  zOTP.generate_otp)r cCs8|j}t|d}|dkr*|dd|7}tj|ddS)Nrr=T)casefold)rlenbase64 b32decode)rrmissing_paddingrrrr /s  zOTP.byte_secretr)ipaddingr cCs>t}|dkr&||d@|dL}qttt||dS)z Turns an integer to the OATH specified bytestring, which is fed to the HMAC along with the secret rrr)r"appendbytesreversedrjust)r-r.resultrrrr!6s  zOTP.int_to_bytestring)r)__name__ __module__ __qualname____doc__hashlibsha1r#intrrrr&r1r staticmethodr!rrrrrs" r)r*r9rtypingrrobjectrrrrrs