_@sddlZddlmZddlmZddlmZddlmZddlmZddlmZdd lm Z dd l m Z dd l m Z dd l m Z dd lmZddlmZGdddeZGdddeZdS)N)datetime) text_type) base64_decode) base64_encode) bytes_to_int) int_to_bytes) want_bytes) BadSignature)BadTimeSignature)SignatureExpired) Serializer)Signerc@s[eZdZdZddZddZddZdd d d Zdd d ZdS)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. cCsttjS)zTReturns the current timestamp. The function must return an integer. )inttime)selfr:/tmp/pip-build-5gj8f0j9/itsdangerous/itsdangerous/timed.py get_timestampszTimestampSigner.get_timestampcCs tj|S)zaUsed to convert the timestamp from :meth:`get_timestamp` into a datetime object. )rutcfromtimestamp)rtsrrrtimestamp_to_datetimesz%TimestampSigner.timestamp_to_datetimecCsVt|}tt|j}t|j}|||}|||j|S)z:Signs the given string and also attaches time information.)r rrrsepZ get_signature)rvalue timestamprrrrsign$s  zTimestampSigner.signNFc Csytj||}d}Wn:tk rX}z|}|jpCd}WYdd}~XnXt|j}||kr|r|tdd||j|d\}}ytt |}Wnt k rd}YnX|dk rtt |d|d||dkr#tdd||dk rv|j |} | |krvt d| |fd|d|j||r||j|fS|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 a naive :class:`datetime.datetime` object in UTC. Nztimestamp missingpayloadrZ date_signedzMalformed timestampzSignature age %s > %s seconds)runsignr rr rr rsplitrr Exceptionrrr r) rrmax_agereturn_timestampresultZ sig_errorerrZagerrrr,s> "       zTimestampSigner.unsignc Cs8y|j|d|dSWntk r3dSYnXdS)zeOnly validates the given signed value. Returns ``True`` if the signature exists and is valid.r"TFN)rr )rZ signed_valuer"rrrvalidatees  zTimestampSigner.validate) __name__ __module__ __qualname____doc__rrrrr&rrrrrs    9rc@sCeZdZdZeZdddddZddddZdS)TimedSerializerzOUses :class:`TimestampSigner` instead of the default :class:`.Signer`. NFc Cst|}d}x|j|D]}yE|j||dd\}}|j|} |rh| |fS| SWq"tk rYq"tk r} z | }WYdd} ~ Xq"Xq"W|dS)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. Nr#T)r Ziter_unsignersrZ load_payloadr r ) rsr"r#saltZlast_exceptionsignerZbase64drrerrrrrloadsvs   zTimedSerializer.loadscCs(d|i}i}|j||||S)Nr")Z_loads_unsafe_impl)rr,r"r-Z load_kwargsZload_payload_kwargsrrr loads_unsafes zTimedSerializer.loads_unsafe)r'r(r)r*rZdefault_signerr0r1rrrrr+os r+)rr_compatrencodingrrrrr excr r r serializerr r.rrr+rrrrs ^