B h@s`ddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z Gddde Z dS)N)AnyUnionOptional)utils)OTPc seZdZdZdejdddfeeee ee eeddfdd Z de ee j feed d d Z ed d dZdee e j eedddZde ee ee eedddZe j edddZZS)TOTPz. Handler for time-based OTP counters. N)sdigitsdigestnameissuerintervalreturncs ||_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 )r r r rrN)rsuper__init__)selfr r r rrr) __class__GC:\Users\sanjo\AppData\Local\Qlobot\Launcher\ext_packages\pyotp\totp.pyrs z 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:: 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)rrrrrrats zTOTP.at)rcCs||tjS)zL Generate the current time OTP :returns: OTP value )rrrnow)rrrrr!-szTOTP.now)otpr valid_windowrc Csn|dkrtj}|rTx8t| |dD]$}tt|t|||r(dSq(WdStt|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#irrrverify5s  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 periodr*)r build_urisecretrrr r r)rrr)r*rrrprovisioning_uriIs  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) tzinforcalendartimegm utctimetuplertimemktime timetuple)rrrrrrZsz TOTP.timecode)r)Nr)NNN)__name__ __module__ __qualname____doc__hashlibsha1r&rrrrrrr r!boolr(r/r __classcell__rr)rrr s, "r) r1rr;r4typingrrrrr"rrrrrrs