o uh@s<ddlZddlZddlZddlmZmZGdddeZdS)N)AnyOptionalc @seZdZdZdejddfdededede ede ed df d d Z d ed efd dZ d e fddZ eddeded e fddZdS)OTPz& Base class for OTP handlers. NsdigitsdigestnameissuerreturncCs6||_|dkr td||_||_|pd|_||_dS)N z!digits must be no greater than 10Secret)r ValueErrorrsecretr r )selfrrrr r r||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) rhmacnew byte_secretint_to_bytestringr bytearraystrr)rrhasher hmac_hashoffsetcodestr_coderrr generate_otps  zOTP.generate_otpcCs8|j}t|d}|dkr|dd|7}tj|ddS)Nrr=T)casefold)rlenbase64 b32decode)rrmissing_paddingrrrr!/s  zOTP.byte_secretripaddingcCsDt}|dkr||d@|dL}|dksttt||dS)z Turns an integer to the OATH specified bytestring, which is fed to the HMAC along with the secret rrr)r#appendbytesreversedrjust)r1r2resultrrrr"6s zOTP.int_to_bytestring)r)__name__ __module__ __qualname____doc__hashlibsha1r$intrrrr*r5r! staticmethodr"rrrrrs. r)r.r=rtypingrrobjectrrrrrs