3 (h@sddlZddlmZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZmZGdd d ejZGd d d ejZGd d d eZdS)N)settings) BaseStorageMessage) SimpleCookie)six)constant_time_compare salted_hmac)SafeData mark_safecs$eZdZdZdZfddZZS)MessageEncoderzJ Compactly serializes instances of the ``Message`` class as JSON. Z__json_messagecsVt|trFt|jtrdnd}|j||j|jg}|jrB|j|j|Stt |j |S)Nr) isinstancermessager message_keylevelZ extra_tagsappendsuperr default)selfobjZ is_safedatar) __class__J/tmp/pip-install-q3hcpn_q/Django/django/contrib/messages/storage/cookie.pyrs  zMessageEncoder.default)__name__ __module__ __qualname____doc__rr __classcell__rr)rrr sr cs(eZdZdZddZfddZZS)MessageDecoderzF Decodes JSON that includes serialized ``Message`` instances. cst|trr|rr|dtjkr`t|dkr8t|ddS|drPt|d|d<t|ddSfdd|DSt|trfddtj |DS|S) Nrr csg|]}j|qSr)process_messages).0item)rrr *sz3MessageDecoder.process_messages..csi|]\}}j||qSr)r!)r"keyvalue)rrr ,sz3MessageDecoder.process_messages..) r listr rlenrr dictr iteritems)rrr)rrr!!s   zMessageDecoder.process_messagesc stt|j|f|}|j|S)N)rrdecoder!)rskwargsdecoded)rrrr,0szMessageDecoder.decode)rrrrr!r,rrr)rrrsrc@sPeZdZdZdZdZdZddZddZdd d Z d d Z dddZ ddZ dS) CookieStoragez& Stores messages in a cookie. messagesiZ__messagesnotfinished__cOsH|jjj|j}|j|}|o*|d|jk }|r@| r@|j||fS)a  Retrieves 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.datar1Z all_retrievedrrr_get@s   zCookieStorage._getcCs@|r*|j|j|tjtjpdtjp"ddn|j|jtjddS)z| Either sets the cookie with the encoded data if there is any data to store, or deletes the cookie. N)domainsecurehttponly)r<) set_cookier5rZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREZSESSION_COOKIE_HTTPONLYZ delete_cookie)r encoded_dataresponserrr_update_cookieOszCookieStorage._update_cookieTc sg}|j|}|jrztfdd}xR|rx|||jkrx|rP|j|jdn|jd|j|j||jg|d}q(W|j|||S)aV Stores the messages to a cookie, returning a list of any messages which could not be stored. If the encoded data is larger than ``max_cookie_size``, removes 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_sizerrr8insertr7rB) rr1rAZ remove_oldestr9r.Zunstored_messagesr@rFr)rEr_store^s     zCookieStorage._storecCsd}t||jS)z Creates 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._hashFcCs2|s|r.tdd}|j|}d|j||fSdS)a  Returns 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)rNrO)r encoderM)rr1rGencoderr&rrrrHs  zCookieStorage._encodec Csf|sdS|jdd}t|dkr\|\}}t||j|r\ytj|tdStk rZYnXd|_dS)z Safely decodes 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, ``None`` is returned. N$r r )clsT) splitr)rrMjsonloadsr ValueErrorused)rr:bitshashr&rrrr6s  zCookieStorage._decodeN)T)F) rrrrr5rIr7r;rBrKrMrHr6rrrrr05s  r0)rVZ django.confrZ$django.contrib.messages.storage.baserrZ django.httprZ django.utilsrZdjango.utils.cryptorrZdjango.utils.safestringr r JSONEncoderr JSONDecoderrr0rrrrs