o }e@sddlZddlZddlZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd l m Z dd l m Z dd l mZdd lmZddlmZejeefZejeZejeZejrlddlZGdddeZGdddeZdS)N)datetime)timezone) base64_decode) base64_encode) bytes_to_int) int_to_bytes) want_bytes) BadSignature)BadTimeSignature)SignatureExpired) Serializer)Signerc seZdZdZdefddZdedefddZdede fd d Z e j dd ede ddde fddZe j dd ede dddeje effddZ dd ede dedejeje efe fffdd Zdd ede defddZZS)TimestampSignerzWorks 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. returncCs ttS)zTReturns the current timestamp. The function must return an integer. )inttime)selfr[/Users/abhinavbhinder/Documents/map/venv/lib/python3.10/site-packages/itsdangerous/timed.py get_timestamp!s zTimestampSigner.get_timestamptscCstj|tjdS)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 fromtimestamprutc)rrrrrtimestamp_to_datetime'sz%TimestampSigner.timestamp_to_datetimevaluecCs@t|}tt|}t|j}|||}||||S)z:Signs the given string and also attaches time information.)r rrrsepZ get_signature)rr timestamprrrrsign1s   zTimestampSigner.signNF signed_valuemax_agereturn_timestampz_te.Literal[False]cCdSNrrr r!r"rrrunsign<zTimestampSigner.unsignTz_te.Literal[True]cCr#r$rr%rrrr&Er'c sz t|}d}Wnty#}z |}|jpd}WYd}~nd}~wwt|j}||vr7|r1|td|d||d\}} d} d} ztt | } Wn t yTYnw|dur| dur{z| | } Wnt t tfyz} ztd|d| d} ~ wwtt||| d| durtd|d|dur|| } | |krtd| d |d || | d| d krtd| d || | d|r|| | fS|S) aWorks 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. Nztimestamp missing)payloadrzMalformed timestamp)r)Z date_signedzSignature age z > z secondsrz < 0 seconds)superr&r r)r rr rsplitrr Exceptionr ValueErrorOSError OverflowErrorstrrr )rr r!r"resultZ sig_errorerrZts_bytesZts_intZts_dtexcZage __class__rrr&Nsj       cCs*z |j||dWdStyYdSw)zeOnly validates the given signed value. Returns ``True`` if the signature exists and is valid.)r!TF)r&r )rr r!rrrvalidates  zTimestampSigner.validate)NF)NTr$)__name__ __module__ __qualname____doc__rrrr _t_str_bytesbytesrtypingoverload _t_opt_intr&_tTupleboolUnionr6 __classcell__rrr4rrsR       Xrc seZdZUdZeZejeed< dde dej effdd Z   dd e d e d ede dejf d d Z  dd e d e de dejeejffddZZS)TimedSerializerzOUses :class:`TimestampSigner` instead of the default :class:`.Signer`. default_signerNsaltrcstdt|S)Nz_t.Iterator[TimestampSigner])r@castr*iter_unsigners)rrGr4rrrIszTimedSerializer.iter_unsignersFsr!r"c Cst|}d}||D]:}z|j||dd\}}||} |r&| |fWS| WSty2tyE} z| }WYd} ~ q d} ~ wwtt|)aReverse 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 rIr&Z load_payloadr r r@rH) rrJr!r"rGZlast_exceptionsignerZbase64drr)errrrrloadss$    zTimedSerializer.loadscCs|j||d|idS)Nr!)Z load_kwargs)Z_loads_unsafe_impl)rrJr!rGrrr loads_unsafeszTimedSerializer.loads_unsafer$)NFN)NN)r7r8r9r:rrFr@Type__annotations___t_opt_str_bytesIteratorrIr;r?rBAnyrMrArNrDrrr4rrEsD   (rE) rr=r@rrencodingrrrrr r3r r r serializerr rKrrCr0r<r;OptionalrQrr? TYPE_CHECKINGZtyping_extensionsZ_terrErrrrs.