o uh@s`ddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z Gddde Z dS)N)AnyOptionalUnion)utils)OTPcseZdZdZ     ddedededeed eed ed dffd d Zd de ee j fded efddZ d efddZ d!dedee j ded e fddZ d"deedeedeed efddZde j d efddZZS)#TOTPz. Handler for time-based OTP counters. Nsdigitsdigestnameissuerintervalreturncs.|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__=/var/www/html/venv/lib/python3.10/site-packages/pyotp/totp.pyrsz TOTP.__init__rfor_timecounter_offsetcCs0t|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.atcCs||tjS)zL Generate the current time OTP :returns: OTP value )r!r"rnow)rrrrr$:szTOTP.nowotp valid_windowc Csl|dur tj}|r)t| |dD]}tt|t|||r&dSqdStt|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$ranger strings_equalstrr#)rr%rr&irrrverifyBs z TOTP.verify issuer_nameimagec 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 periodr-)r build_urisecretrrr r r)rrr,r-rrrprovisioning_uriVs   zTOTP.provisioning_uricCs6|jrtt||jStt||jS)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). ) tzinfor calendartimegm utctimetuplertimemktime timetuple)rrrrrr"msz TOTP.timecode)r NNNr )r)Nr)NNN)__name__ __module__ __qualname____doc__r)r rrrrrr#r$boolr+r2r" __classcell__rrrrr sH"" r) r4rrr7typingrrrrr%rrrrrrs