U\hfddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z Gdde Z y)N)AnyOptionalUnion)utils)OTPceZdZdZ ddedededeedeeded dffd Zdd e ee jfd ed efd Z d efdZ dded ee jded e fdZ ddeedeedeed efdZd e jd efdZxZS)TOTPz. Handler for time-based OTP counters. Nsdigitsdigestnameissuerintervalreturnc`|tj}||_t||||||y)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__s F/var/www/html/trade_iq/venv/lib/python3.12/site-packages/pyotp/totp.pyrz TOTP.__init__s2" >\\F   1VFfUfor_timecounter_offsetct|tjs(tjjt|}|j |j ||zS)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)rrrs ratzTOTP.at'sM(H$5$56((66s8}EH  x!8>!IJJrcz|j|jtjjS)zL Generate the current time OTP :returns: OTP value )r"r#rnow)rs rr&zTOTP.now:s-   x/@/@/D/D/F!GHHrotp valid_windowc T|tjj}|rPt| |dzD]<}tjt |t |j ||s<yytjt |t |j |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 rTF)rr&ranger strings_equalstrr$)rr'rr(is rverifyz TOTP.verifyBs  ((,,.H L=,*:; &&s3xTWWXq5I1JK ""3s8S1B-CDDr issuer_nameimagec tj|j|r|n |j|r|n |j|j j|j |j|S)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 periodr0)r build_urisecretrrr r r)rrr/r0s rprovisioning_urizTOTP.provisioning_uriVsR KKDdii"-;4;;kkm((;;==  rc|jr9ttj|j |j z Stt j|j|j z S)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)rrs rr#z TOTP.timecodemsW ??xx'<'<'>?$--OP Pt{{8#5#5#784==HI Ir)NNN)r)Nr)NNN)__name__ __module__ __qualname____doc__r,r!rrrrrr$r&boolr.r6r# __classcell__)rs@rr r s8" $V VV V sm V  VV V.K5h&7&7!78K#KVYK&ISIE#E(2C2C)DE[^EgkE*ei SM 7?} T\]`Ta  . J!2!2 Js Jrr ) r9rrr<typingrrrrr'rr rrrJs- ''lJ3lJr