a
    �ÏOcî  ã                   @   sü   d dl Z d dlZd dlZd dlmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 dd	lm
Z
 dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ejeef Zeje Zeje ZejrØd dlZG dd„ deƒZG dd„ deƒZdS )é    N)Údatetime)Útimezoneé   )Úbase64_decode)Úbase64_encode)Úbytes_to_int)Úint_to_bytes)Ú
want_bytes)ÚBadSignature)ÚBadTimeSignature)ÚSignatureExpired)Ú
Serializer)ÚSignerc                       sÆ   e Zd ZdZedœdd„Zeedœdd„Zee	dœd	d
„Z
ejdeede	dœdd„ƒZejdeedeje	ef dœdd„ƒZdeeeejeje	ef e	f dœ‡ fdd„Zdeeedœdd„Z‡  ZS )ÚTimestampSignerzþWorks like the regular :class:`.Signer` but also records the time
    of the signing and can be used to expire signatures. The
    :meth:`unsign` method can raise :exc:`.SignatureExpired` if the
    unsigning failed because the signature is expired.
    )Úreturnc                 C   s   t t ¡ ƒS )zTReturns the current timestamp. The function must return an
        integer.
        )ÚintÚtime)Úself© r   úI/var/www/html/gsl.tools/lib/python3.9/site-packages/itsdangerous/timed.pyÚget_timestamp!   s    zTimestampSigner.get_timestamp)Útsr   c                 C   s   t j|tjd�S )a   Convert the timestamp from :meth:`get_timestamp` into an
        aware :class`datetime.datetime` in UTC.

        .. versionchanged:: 2.0
            The timestamp is returned as a timezone-aware ``datetime``
            in UTC rather than a naive ``datetime`` assumed to be UTC.
        )Útz)r   Úfromtimestampr   Úutc)r   r   r   r   r   Útimestamp_to_datetime'   s    z%TimestampSigner.timestamp_to_datetime)Úvaluer   c                 C   s@   t |ƒ}tt|  ¡ ƒƒ}t | jƒ}|| | }|| |  |¡ S )z:Signs the given string and also attaches time information.)r	   r   r   r   ÚsepZget_signature)r   r   Ú	timestampr   r   r   r   Úsign1   s
    
zTimestampSigner.signNFz_te.Literal[False])Úsigned_valueÚmax_ageÚreturn_timestampr   c                 C   s   d S ©Nr   ©r   r    r!   r"   r   r   r   Úunsign<   s    zTimestampSigner.unsignTz_te.Literal[True]c                 C   s   d S r#   r   r$   r   r   r   r%   E   s    c              
      s°  zt ƒ  |¡}d}W n2 tyF } z|}|jp0d}W Y d}~n
d}~0 0 t| jƒ}||vrn|rb|‚td|d�‚| |d¡\}}	d}
d}ztt	|	ƒƒ}
W n t
y¨   Y n0 |du�r|
du�rz|  |
¡}W n: tttf�y } ztd|d�|‚W Y d}~n
d}~0 0 tt|ƒ||d�‚|
du �r0td|d�‚|du�r˜|  ¡ |
 }||k�rrtd|› d	|› d
�||  |
¡d�‚|dk �r˜td|› d�||  |
¡d�‚|�r¬||  |
¡fS |S )aí  Works like the regular :meth:`.Signer.unsign` but can also
        validate the time. See the base docstring of the class for
        the general behavior. If ``return_timestamp`` is ``True`` the
        timestamp of the signature will be returned as an aware
        :class:`datetime.datetime` object in UTC.

        .. versionchanged:: 2.0
            The timestamp is returned as a timezone-aware ``datetime``
            in UTC rather than a naive ``datetime`` assumed to be UTC.
        Nó    ztimestamp missing)Úpayloadr   zMalformed timestamp)r'   Zdate_signedzSignature age z > z secondsr   z < 0 seconds)Úsuperr%   r
   r'   r	   r   r   Úrsplitr   r   Ú	Exceptionr   Ú
ValueErrorÚOSErrorÚOverflowErrorÚstrr   r   )r   r    r!   r"   ÚresultZ	sig_errorÚer   r   Zts_bytesZts_intZts_dtÚexcZage©Ú	__class__r   r   r%   N   s`     


ÿþ


ý

ý)r    r!   r   c                 C   s.   z| j ||d� W dS  ty(   Y dS 0 dS )zeOnly validates the given signed value. Returns ``True`` if
        the signature exists and is valid.)r!   TFN)r%   r
   )r   r    r!   r   r   r   Úvalidate¦   s
    zTimestampSigner.validate)NF)NT)NF)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   Ú_t_str_bytesÚbytesr   ÚtypingÚoverloadÚ
_t_opt_intr%   Ú_tÚTupleÚboolÚUnionr4   Ú__classcell__r   r   r2   r   r      s>   
  üû  üû  üûXr   c                       s€   e Zd ZU dZeZeje ed< de	ej
e dœ‡ fdd„Zdeeee	ejdœd	d
„Zdeee	ejeejf dœdd„Z‡  ZS )ÚTimedSerializerzOUses :class:`TimestampSigner` instead of the default
    :class:`.Signer`.
    Údefault_signerN)Úsaltr   c                    s   t  dtƒ  |¡¡S )Nz_t.Iterator[TimestampSigner])r>   Úcastr(   Úiter_unsigners)r   rE   r2   r   r   rG   ·   s    zTimedSerializer.iter_unsignersF)Úsr!   r"   rE   r   c                 C   s¢   t |ƒ}d}|  |¡D ]z}z:|j||dd�\}}|  |¡}	|rL|	|fW   S |	W   S  tyh   ‚ Y q tyŽ }
 z|
}W Y d}
~
qd}
~
0 0 qt t|¡‚dS )a†  Reverse of :meth:`dumps`, raises :exc:`.BadSignature` if the
        signature validation fails. If a ``max_age`` is provided it will
        ensure the signature is not older than that time in seconds. In
        case the signature is outdated, :exc:`.SignatureExpired` is
        raised. All arguments are forwarded to the signer's
        :meth:`~TimestampSigner.unsign` method.
        NT)r!   r"   )r	   rG   r%   Zload_payloadr   r
   r>   rF   )r   rH   r!   r"   rE   Zlast_exceptionÚsignerZbase64dr   r'   Úerrr   r   r   Úloads¿   s     ÿ


zTimedSerializer.loads)rH   r!   rE   r   c                 C   s   | j ||d|id�S )Nr!   )Zload_kwargs)Z_loads_unsafe_impl)r   rH   r!   rE   r   r   r   Úloads_unsafeä   s    zTimedSerializer.loads_unsafe)N)NFN)NN)r5   r6   r7   r8   r   rD   r>   ÚTypeÚ__annotations__Ú_t_opt_str_bytesÚIteratorrG   r9   r=   r@   ÚAnyrK   r?   rL   rB   r   r   r2   r   rC   °   s2   
 ÿþ   ûú(  üûrC   ) r   r;   r>   r   r   Úencodingr   r   r   r   r	   r1   r
   r   r   Ú
serializerr   rI   r   rA   r.   r:   r9   ÚOptionalrO   r   r=   ÚTYPE_CHECKINGZtyping_extensionsZ_ter   rC   r   r   r   r   Ú<module>   s,   

 