o uvXj@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. __json_messagecsRt|tr#t|jtr dnd}|j||j|jg}|jr!||j|St |S)Nr) isinstancermessager message_keylevel extra_tagsappendsuperdefault)selfobj is_safedatar __class__c/var/www/html/myproject/venv/lib/python3.10/site-packages/django/contrib/messages/storage/cookie.pyrs   zMessageEncoder.default)__name__ __module__ __qualname____doc__rr __classcell__rrrrr sr cs(eZdZdZddZfddZZS)MessageDecoderzE Decode JSON that includes serialized ``Message`` instances. cst|tr9|r9|dtjkr0t|dkrt|ddS|dr(t|d|d<t|ddSfdd|DSt|trIfdd|DS|S) Nrr csg|]}|qSrprocess_messages).0itemrrr )sz3MessageDecoder.process_messages..csi|] \}}||qSrr$)r&keyvaluer(rr +sz3MessageDecoder.process_messages..) r listr rlenrr dictitems)rrrr(rr% s   zMessageDecoder.process_messagesc stj|fi|}||S)N)rdecoder%)rskwargsdecodedrrrr1/s zMessageDecoder.decode)rrrrr%r1r rrrrr!sr!c@sPeZdZdZdZdZdZddZddZdd d Z d d Z dddZ ddZ dS) CookieStoragez% Store messages in a cookie. messagesi__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. )requestCOOKIESget cookie_name_decode not_finishedpop)rargsr3datar6 all_retrievedrrr_get?s  zCookieStorage._getcCsF|r|j|j|tjtjp dtjpdtjddS|j|jtjddS)zz Either set the cookie with the encoded data if there is any data to store, or delete the cookie. N)domainsecurehttponlysamesite)rD) set_cookier<rSESSION_COOKIE_DOMAINSESSION_COOKIE_SECURESESSION_COOKIE_HTTPONLYSESSION_COOKIE_SAMESITE delete_cookie)r encoded_dataresponserrr_update_cookieNs zCookieStorage._update_cookieTc sg}||}|jrCtfdd}|rC|||jkrC|r'||dn|d||j||jg|d}|rC|||jks||||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)valcookierr stored_lengthnsz+CookieStorage._store..stored_lengthr) encode_empty)_encodemax_cookie_sizerrr?insertr>rP) rr6rO remove_oldestr@r3unstored_messagesrNrUrrSr_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+key_saltrrr_hash{szCookieStorage._hashFcCs2|s|rtdd}||}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 encoder_)rr6rVencoderr+rrrrWs   zCookieStorage._encodecCsh|sdS|dd}t|dkr/|\}}t|||r/ztj|tdWStjy.Ynwd|_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.rr_jsonloadsr!JSONDecodeErrorused)rrAbitshashr+rrrr=s  zCookieStorage._decodeN)T)F) rrrrr<rXr>rCrPr\r_rWr=rrrrr54s   r5)rh django.confr$django.contrib.messages.storage.baserr django.httprdjango.utils.cryptorrdjango.utils.safestringrr JSONEncoderr JSONDecoderr!r5rrrrs