B 0H[?@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$eZdZdZdZfddZZS)MessageEncoderzI Compactly serialize instances of the ``Message`` class as JSON. Z__json_messagecsRt|trFt|jtrdnd}|j||j|jg}|jrB||j|St |S)Nr) isinstancermessager message_keylevelZ extra_tagsappendsuperdefault)selfobjZ is_safedatar ) __class__jC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6bm3nxem\django\django\contrib\messages\storage\cookie.pyrs  zMessageEncoder.default)__name__ __module__ __qualname____doc__rr __classcell__rr)rrr sr cs(eZdZdZddZfddZZS)MessageDecoderzE Decode JSON that includes serialized ``Message`` instances. cst|trr|rr|dtjkr`t|dkr8t|ddS|drPt|d|d<t|ddSfdd|DSt|trfdd|DS|S) Nrr csg|]}|qSr)process_messages).0item)rrr )sz3MessageDecoder.process_messages..csi|]\}}||qSr)r )r!keyvalue)rrr +sz3MessageDecoder.process_messages..) r listr rlenrr dictitems)rrr)rrr s    zMessageDecoder.process_messagesc stj|f|}||S)N)rdecoder )rskwargsdecoded)rrrr+/szMessageDecoder.decode)rrrrr r+rrr)rrrsrc@sPeZdZdZdZdZdZddZddZdd d Z d d Z dddZ ddZ dS) CookieStoragez% Store messages in a cookie. messagesiZ__messagesnotfinished__cOsF|jj|j}||}|o*|d|jk }|r>|s>|||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. )requestZCOOKIESget cookie_name_decode not_finishedpop)rargsr-datar0Z all_retrievedrrr_get?s  zCookieStorage._getcCs@|r*|j|j|tjtjpdtjp"ddn|j|jtjddS)zz Either set the cookie with the encoded data if there is any data to store, or delete the cookie. N)domainsecurehttponly)r;) set_cookier4rZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREZSESSION_COOKIE_HTTPONLYZ delete_cookie)r encoded_dataresponserrr_update_cookieNszCookieStorage._update_cookieTc sg}||}|jrztfdd}xR|rx|||jkrx|rP||dn|d||j||jg|d}q(W||||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. cst|dS)Nr )r( value_encode)val)cookierr stored_lengthmsz+CookieStorage._store..stored_lengthr) encode_empty)_encodemax_cookie_sizerrr7insertr6rA) rr0r@Z remove_oldestr8r-Zunstored_messagesr?rEr)rDr_store]s     zCookieStorage._storecCsd}t||S)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_hashzszCookieStorage._hashFcCs2|s|r.tdd}||}d|||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. ),:) separatorsz%s$%sN)r encoderL)rr0rFencoderr%rrrrGs  zCookieStorage._encodecCsf|sdS|dd}t|dkr\|\}}t|||r\ytj|tdStk rZYnXd|_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 r)clsT) splitr(rrLjsonloadsr ValueErrorused)rr9bitshashr%rrrr5s  zCookieStorage._decodeN)T)F) rrrrr4rHr6r:rArJrLrGr5rrrrr/4s  r/)rUZ django.confrZ$django.contrib.messages.storage.baserrZ django.httprZdjango.utils.cryptorrZdjango.utils.safestringrr JSONEncoderr JSONDecoderrr/rrrrs