o j@sddlmZddlmZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZddlmZGdddeZGdddeeZdS)) annotationsN)datetime)timezone) base64_decode) base64_encode) bytes_to_int) int_to_bytes) want_bytes) BadSignature)BadTimeSignature)SignatureExpired) _TSerialized) Serializer)Signercs~eZdZdZd!ddZd"dd Zd#d dZej  d$d%ddZ ej  d&d'ddZ  d$d(fdd Z d)d*dd Z Z S)+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. returnintcCs ttS)zTReturns the current timestamp. The function must return an integer. )rtime)selfr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/itsdangerous/timed.py get_timestamps zTimestampSigner.get_timestamptsrcCstj|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_datetimevalue str | bytesbytescCs@t|}tt|}t|j}|||}||||S)z:Signs the given string and also attaches time information.)r rr rsep get_signature)rr timestampr!rrrsign-s   zTimestampSigner.signNF signed_valuemax_age int | Nonereturn_timestampt.Literal[False]cCdSNrrr%r&r(rrrunsign8zTimestampSigner.unsignTt.Literal[True]tuple[bytes, datetime]cCr*r+rr,rrrr-@r.booltuple[bytes, datetime] | bytesc 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)r4 date_signedzSignature age z > z secondsrz < 0 seconds)superr-r r4r r!r rsplitrr Exceptionr ValueErrorOSError OverflowErrorstrrr )rr%r&r(result sig_errorer!rts_bytests_intts_dtexcage __class__rrr-Hsj       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)rr)rrrr)rrrr )NF)r%rr&r'r(r)rr )NT)r%rr&r'r(r/rr0)r%rr&r'r(r1rr2r+)r%rr&r'rr1) __name__ __module__ __qualname____doc__rrr$toverloadr-rG __classcell__rrrErrs"     XrcsVeZdZUdZeZded< ddfd d Z  ddddZ  ddddZ Z S)TimedSerializerzOUses :class:`TimestampSigner` instead of the default :class:`.Signer`. ztype[TimestampSigner]default_signerNsaltstr | bytes | Nonercabc.Iterator[TimestampSigner]cstdt|S)NrS)rLcastr6iter_unsigners)rrQrErrrUszTimedSerializer.iter_unsignersFsrr&r'r(r1t.Anyc 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 rUr- load_payloadr r rLrT) rrVr&r(rQlast_exceptionsignerbase64dr#r4errrrrloadss$    zTimedSerializer.loadstuple[bool, t.Any]cCs|j||d|idS)Nr&) load_kwargs)_loads_unsafe_impl)rrVr&rQrrr loads_unsafeszTimedSerializer.loads_unsafer+)rQrRrrS)NFN) rVrr&r'r(r1rQrRrrW)NN)rVrr&r'rQrRrr^) rHrIrJrKrrP__annotations__rUr]rarNrrrErrOs    (rO) __future__rcollections.abcabccabcrtypingrLrrencodingrrrr r rCr r r serializerrrrZrrrOrrrrs(