a j@s`ddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z Gddde Z dS)N)AnyOptionalUnion)utils)OTPc seZdZdZdeeeeeeeeddfdd Zde ee j feed d d Z ed d dZ deee j ee dddZdeeeeeeedddZe j edddZZS)TOTPz. Handler for time-based OTP counters. N)sdigitsdigestnameissuerintervalreturncs.|durtj}||_tj|||||ddS)a :param s: secret in base32 format :param interval: the time interval in seconds for OTP. This defaults to 30. :param digits: number of integers in the OTP. Some apps expect this to be 6 digits, others support more. :param digest: digest function to use in the HMAC (expected to be SHA1) :param name: account name :param issuer: issuer N)r r r rr)hashlibsha1rsuper__init__)selfr r r rrr __class__0E:\soccer\ml_env\Lib\site-packages\pyotp/totp.pyrsz TOTP.__init__r)for_timecounter_offsetrcCs0t|tjstjt|}||||S)a Accepts either a Unix timestamp integer or a datetime object. To get the time until the next timecode change (seconds until the current OTP expires), use this instead: .. code:: python totp = pyotp.TOTP(...) time_remaining = totp.interval - datetime.datetime.now().timestamp() % totp.interval :param for_time: the time to generate an OTP for :param counter_offset: the amount of ticks to add to the time counter :returns: OTP value ) isinstancedatetime fromtimestampint generate_otptimecode)rrrrrrat's zTOTP.at)rcCs||tjS)zL Generate the current time OTP :returns: OTP value )r!r"rnow)rrrrr$:szTOTP.now)otpr valid_windowrc Csl|durtj}|rRt| |dD]&}tt|t|||r&dSq&dStt|t||S)ah Verifies the OTP passed in against the current time OTP. :param otp: the OTP to check against :param for_time: Time to check OTP at (defaults to now) :param valid_window: extends the validity to this many counter ticks before and after the current one :returns: True if verification succeeded, False otherwise NrTF)rr$rangerZ strings_equalstrr#)rr%rr&irrrverifyBs  z TOTP.verify)r issuer_nameimagerc Cs8tj|j|r|n|j|r|n|j|j|j|j|dS)a Returns the provisioning URI for the OTP. This can then be encoded in a QR Code and used to provision an OTP app like Google Authenticator. See also: https://github.com/google/google-authenticator/wiki/Key-Uri-Format )r algorithmr Zperiodr,)rZ build_urisecretrrr r r)rrr+r,rrrprovisioning_uriVs   zTOTP.provisioning_uri)rrcCs:|jrtt||jStt||jSdS)z Accepts either a timezone naive (`for_time.tzinfo is None`) or a timezone aware datetime as argument and returns the corresponding counter value (timecode). N) tzinfor calendartimegm utctimetuplertimemktime timetuple)rrrrrr"msz TOTP.timecode)r NNNr )r)Nr)NNN)__name__ __module__ __qualname____doc__r(r rrrrrr#r$boolr*r/r" __classcell__rrrrr s0 r) r1rrr4typingrrrrr%rrrrrrs