File: //usr/local/CyberPanel/lib64/python3.10/site-packages/pyotp/__pycache__/totp.cpython-310.pyc
o
��h � @ s` d dl Z d dlZd dlZd dlZd dlmZmZmZ ddlm Z ddl
m Z G dd� de �Z dS )� N)�Any�Optional�Union� )�utils)�OTPc s� e Zd ZdZ d dedededee d ee d
ed df� fd d
�
Zd de ee
j
f ded efdd�Z d efdd�Z d!dedee
j
ded e
fdd�Z d"dee dee dee d efdd �Zde
j
d efd d �Z� ZS )#�TOTPz.
Handler for time-based OTP counters.
� N� �s�digits�digest�name�issuer�interval�returnc s. |du rt j}|| _t� j|||||d� dS )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
r r )�hashlib�sha1r �super�__init__)�selfr r r
r r r �� __class__� �@/usr/local/CyberPanel/lib/python3.10/site-packages/pyotp/totp.pyr s z
TOTP.__init__r �for_time�counter_offsetc C s0 t |tj�stj�t|��}| �| �|�| �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
)�
isinstance�datetime�
fromtimestamp�int� generate_otp�timecode)r r r r r r �at' s zTOTP.atc C s | � | �tj�� ��S )zL
Generate the current time OTP
:returns: OTP value
)r! r" r �now)r r r r r$ : s zTOTP.now�otp� valid_windowc C sl |du r t j �� }|r)t| |d �D ]}t�t|�t| �||���r& dS qdS t�t|�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
Nr TF)r r$ �ranger �
strings_equal�strr# )r r% r r& �ir r r �verifyB s
�z TOTP.verify� issuer_name�imagec C s8 t j| j|r|n| j|r|n| j| �� j| j| j|d�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 �periodr- )r � build_uri�secretr r r
r r )r r r, r- r r r �provisioning_uriV s
�zTOTP.provisioning_uric C s6 |j rtt�|�� �| j �S tt�|�� �| j �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 �calendar�timegm� utctimetupler �time�mktime� timetuple)r r r r r r" m s z
TOTP.timecode)r NNNr
)r )Nr )NNN)�__name__�
__module__� __qualname__�__doc__r) r r r r r r r# r$ �boolr+ r2 r" �
__classcell__r r r r r sH ��������""����
� r )
r4 r r r7 �typingr r r � r r% r r r r r r �<module> s