X@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.eZdZdZdZfddZS)MessageEncoderzJ Compactly serializes instances of the ``Message`` class as JSON. Z__json_messagecs~t|trht|jtr'dnd}|j||j|jg}|jrd|j|jn|Stt |j |S)Nr) isinstancermessager message_keylevelZ extra_tagsappendsuperr default)selfobjZ is_safedatar) __class__T/home/ubuntu/projects/ifolica/build/django/django/contrib/messages/storage/cookie.pyrs zMessageEncoder.default)__name__ __module__ __qualname____doc__rrrr)rrr s r cs4eZdZdZddZfddZS)MessageDecoderzF Decodes JSON that includes serialized ``Message`` instances. cst|tr|r|dtjkrt|dkrNt|ddS|drot|d|d*s z3MessageDecoder.process_messages..cs(i|]\}}j||qSr)r )r!keyvalue)rrr ,s z3MessageDecoder.process_messages..) r listr rlenrr dictr iteritems)rrr)rrr !s zMessageDecoder.process_messagesc s(tt|j||}|j|S)N)rrdecoder )rskwargsdecoded)rrrr+0szMessageDecoder.decode)rrrrr r+rr)rrrs  rc@sveZdZdZdZdZdZddZddZd d d Z d d Z dddZ ddZ dS) CookieStoragez& Stores messages in a cookie. messagesiZ__messagesnotfinished__cOse|jjj|j}|j|}|o=|d|jk }|r[| r[|jn||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-datar0Z all_retrievedrrr_get@s   zCookieStorage._getc Cs`|rC|j|j|dtjdtjp-ddtjp<dn|j|jdtjdS)z| Either sets the cookie with the encoded data if there is any data to store, or deletes the cookie. domainsecureNhttponly) set_cookier4rZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREZSESSION_COOKIE_HTTPONLYZ delete_cookie)r encoded_dataresponserrr_update_cookieOs  zCookieStorage._update_cookieTc sg}|j|}|jrtfdd}xv|r|||jkr|rv|j|jdn|jd|j|j||jgd|}q<Wn|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_sizerrr7insertr6rA) rr0r@Z remove_oldestr8r-Zunstored_messagesr?rEr)rDr_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._hashFcCsE|s |rAtdd}|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. separators,:z%s$%sN)rNrO)r encoderL)rr0rFencoderr%rrrrGs zCookieStorage._encodec Cs|s dS|jdd}t|dkr|\}}t||j|rytj|dtSWqtk r}YqXqnd|_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 rclsT) splitr(rrLjsonloadsr ValueErrorused)rr9bitshashr%rrrr5s    zCookieStorage._decodeN) rrrrr4rHr6r:rArJrLrGr5rrrrr/5s     r/)rUZ django.confrZ$django.contrib.messages.storage.baserrZ django.httprZ django.utilsrZdjango.utils.cryptorrZdjango.utils.safestringr r JSONEncoderr JSONDecoderrr/rrrrs