v^@sddlZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z GdddejZGd d d ejZGd d d eZdS) N)settings) BaseStorageMessage) SimpleCookie)constant_time_compare salted_hmac)SafeData mark_safecs.eZdZdZdZfddZS)MessageEncoderzI Compactly serialize instances of the ``Message`` class as JSON. Z__json_messagecsut|tret|jtr'dnd}|j||j|jg}|jra|j|j|Stj |S)Nr) isinstancermessager message_keylevelZ extra_tagsappendsuperdefault)selfobjZ is_safedatar ) __class__H/tmp/pip-build-8lau8j11/django/django/contrib/messages/storage/cookie.pyrs zMessageEncoder.default)__name__ __module__ __qualname____doc__rrrr)rrr s r cs4eZdZdZddZfddZS)MessageDecoderzE Decode JSON that includes serialized ``Message`` instances. cst|tr|r|dtjkrt|dkrNt|ddS|drlt|d|d)s z3MessageDecoder.process_messages..cs(i|]\}}j||qSr)r)r keyvalue)rrr +s z3MessageDecoder.process_messages..) r listr rlenrr dictitems)rrr)rrr s zMessageDecoder.process_messagesc s"tj||}|j|S)N)rdecoder)rskwargsdecoded)rrrr*/szMessageDecoder.decode)rrrrrr*rr)rrrs  rc@sveZdZdZdZdZdZddZddZd d d Z d d Z dddZ ddZ dS) CookieStoragez% Store messages in a cookie. messagesiZ__messagesnotfinished__cOsb|jjj|j}|j|}|o=|d|jk }|rX| rX|j||fS)a  Retrieve a list of messages from the messages cookie. If the not_finished sentinel value is found at the end of the message list, remove it and return a result indicating that not all messages were retrieved by this storage. r )requestZCOOKIESget cookie_name_decode not_finishedpop)rargsr,datar/Z all_retrievedrrr_get?s   zCookieStorage._getc Csi|rL|j|j|dtjdtjp-ddtjp<ddtjn|j|jdtjdS)zz Either set the cookie with the encoded data if there is any data to store, or delete the cookie. domainsecureNhttponlyZsamesite) set_cookier3rZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREZSESSION_COOKIE_HTTPONLYZSESSION_COOKIE_SAMESITEZ delete_cookie)r encoded_dataresponserrr_update_cookieNs   zCookieStorage._update_cookieTc sg}|j|}|jrtfdd}xs|r|||jkr|rv|j|jdn|jd|j|j||jgd|}q<W|j|||S)aT Store the messages to a cookie and return a list of any messages which could not be stored. If the encoded data is larger than ``max_cookie_size``, remove messages until the data fits (these are the messages which are returned), and add the not_finished sentinel value to indicate as much. cstj|dS)Nr )r' value_encode)val)cookierr stored_lengthnsz+CookieStorage._store..stored_lengthr encode_empty)_encodemax_cookie_sizerrr6insertr5r@) rr/r?Z remove_oldestr7r,Zunstored_messagesr>rDr)rCr_store^s    zCookieStorage._storecCsd}t||jS)z Create an HMAC/SHA1 hash based on the value and the project setting's SECRET_KEY, modified to make it unique for the present purpose. zdjango.contrib.messages)r hexdigest)rr$Zkey_saltrrr_hash{szCookieStorage._hashFcCsE|s |rAtdd}|j|}d|j||fSdS)a Return an encoded version of the messages list which can be stored as plain text. Since the data will be retrieved from the client-side, the encoded data also contains a hash to ensure that the data was not tampered with. separators,:z%s$%sN)rMrN)r encoderK)rr/rEencoderr$rrrrFs zCookieStorage._encodec Cs|s dS|jdd}t|dkr|\}}t||j|rytj|dtSWntjk rYnXd|_dS)z Safely decode an encoded text stream back into a list of messages. If the encoded text stream contained an invalid hash or was in an invalid format, return None. N$r rclsT) splitr'rrKjsonloadsrJSONDecodeErrorused)rr8bitshashr$rrrr4s  zCookieStorage._decodeN) rrrrr3rGr5r9r@rIrKrFr4rrrrr.4s     r.)rTZ django.confrZ$django.contrib.messages.storage.baserrZ django.httprZdjango.utils.cryptorrZdjango.utils.safestringrr JSONEncoderr JSONDecoderrr.rrrrs