o
    5Wc                     @   sF   d dl Z d dlmZ d dlmZ d dlmZmZmZ G dd dZ	dS )    N)constant_time)InvalidToken)HOTP_ALLOWED_HASH_TYPES_generate_uric                   @   s   e Zd Z		ddededededejdefd	d
Z	dej
eef defddZdededdfddZdedeje defddZdS )TOTPNTkeylength	algorithm	time_stepbackendenforce_key_lengthc                 C   s   || _ t||||d| _d S )N)r   )
_time_stepr   _hotp)selfr   r	   r
   r   r   r    r   OD:\Flask\env\Lib\site-packages\cryptography/hazmat/primitives/twofactor/totp.py__init__   s   	zTOTP.__init__timereturnc                 C   s   t || j }| j|S )N)intr   r   generate)r   r   counterr   r   r   r      s   zTOTP.generatetotpc                 C   s   t | ||stdd S )Nz#Supplied TOTP value does not match.)r   Zbytes_eqr   r   )r   r   r   r   r   r   verify#   s   zTOTP.verifyaccount_nameissuerc                 C   s   t | jd||dt| jfgS )Nr   Zperiod)r   r   r   r   )r   r   r   r   r   r   get_provisioning_uri'   s   zTOTP.get_provisioning_uri)NT)__name__
__module____qualname__bytesr   r   typingAnyboolr   Unionfloatr   r   strOptionalr   r   r   r   r   r      s2    
r   )
r"   Zcryptography.hazmat.primitivesr   Z(cryptography.hazmat.primitives.twofactorr   Z-cryptography.hazmat.primitives.twofactor.hotpr   r   r   r   r   r   r   r   <module>   s
   