o
    uhp
                     @   sD   d dl Z d dlmZmZ ddlmZ ddlmZ G dd deZdS )    N)AnyOptional   )utils)OTPc                       s   e Zd ZdZ					ddedededee d	ee d
eddf fddZdedefddZ	dedede
fddZ				ddee d
ee dee dee def
ddZ  ZS )HOTPz.
    Handler for HMAC-based OTP counters.
       Nr   sdigitsdigestnameissuerinitial_countreturnc                    s.   |du rt j}|| _t j|||||d dS )a}  
        :param s: secret in base32 format
        :param initial_count: starting HMAC counter value, defaults to 0
        :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__ =/var/www/html/venv/lib/python3.10/site-packages/pyotp/hotp.pyr      s   zHOTP.__init__countc                 C   s   |  | j| S )zz
        Generates the OTP for the given count.

        :param count: the OTP HMAC counter
        :returns: OTP
        )generate_otpr   )r   r   r   r   r   at$   s   zHOTP.atotpcounterc                 C   s   t t|t| |S )z
        Verifies the OTP passed in against the current counter OTP.

        :param otp: the OTP to check against
        :param counter: the OTP HMAC counter
        )r   strings_equalstrr   )r   r   r   r   r   r   verify-   s   zHOTP.verifyissuer_nameimagec              	   C   s@   t j| j|r|n| j|r|n| j|r|n| j|  j| j|dS )a5  
        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

        :param name: name of the user account
        :param initial_count: starting HMAC counter value, defaults to 0
        :param issuer_name: the name of the OTP issuer; this will be the
            organization title of the OTP entry in Authenticator
        :returns: provisioning URI
        )r   r   r   	algorithmr
   r"   )r   	build_urisecretr   r   r   r   r
   )r   r   r   r!   r"   r   r   r   provisioning_uri6   s   zHOTP.provisioning_uri)r   NNNr   )NNNN)__name__
__module____qualname____doc__r   intr   r   r   r   boolr    r&   __classcell__r   r   r   r   r      sN    	r   )	r   typingr   r    r   r   r   r   r   r   r   r   <module>   s
    