o {,h!@sddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z m Z GdddejZGd d d ejZGd d d ZGd ddZGdddZGdddeZddZddZdS)N)settings) BaseStorageMessage)signing) SimpleCookie)SafeData mark_safecs$eZdZdZdZfddZZS)MessageEncoderzI Compactly serialize instances of the ``Message`` class as JSON. __json_messagecsVt|tr%t|jtr dnd}|j||j|jg}|jdur#||j|St |S)Nr) isinstancermessager message_keylevel extra_tagsappendsuperdefault)selfobj is_safedatar  __class__\/var/www/html/rh/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. cszt|tr+|r+|dtjkr"|drt|d|d<t|ddSfdd|DSt|tr;fdd|DS|S) Nrr csg|]}|qSrprocess_messages).0itemrrr (sz3MessageDecoder.process_messages..csi|] \}}||qSrr#)r%keyvaluer'rr *sz3MessageDecoder.process_messages..)r listr rrrdictitemsrrrr'rr$"s zMessageDecoder.process_messagesc stj|fi|}||SN)rdecoder$)rskwargsdecodedrrrr1-s zMessageDecoder.decode)rrrrr$r1rrrrrr s r c@eZdZddZdS)MessagePartSerializercCsdd|DS)NcSsg|] }tj|dtdqS)),:) separatorscls)jsondumpsr )r%orrrr(4sz/MessagePartSerializer.dumps..rr/rrrr<3szMessagePartSerializer.dumpsNrrrr<rrrrr62 r6c@r5)MessagePartGatherSerializercCsdd|ddS)z The parameter is an already serialized list of Message objects. No need to serialize it again, only join the list together and encode it. [r7]latin-1)joinencoder/rrrr<?sz!MessagePartGatherSerializer.dumpsNr>rrrrr@>r?r@c@r5)MessageSerializercCstj|dtdS)NrC)r:)r;loadsr1r rdatarrrrGHszMessageSerializer.loadsN)rrrrGrrrrrFGr?rFcspeZdZdZdZdZdZedZ dZ fddZ dd Z d d Z dd dZdddZdddZddZZS) CookieStoragez% Store messages in a cookie. messagesi__messagesnotfinished__zdjango.contrib.messagescs&tj|i|tj|jd|_dS)N)salt)r__init__rget_cookie_signerkey_saltsigner)rargsr3rrrrNZszCookieStorage.__init__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)rrRr3rIrK all_retrievedrrr_get^s  zCookieStorage._getcCsJ|r|j|j|tjtjp dtjpdtjddS|j|jtjtjddS)zz Either set the cookie with the encoded data if there is any data to store, or delete the cookie. N)domainsecurehttponlysamesite)r]r`) set_cookierWrSESSION_COOKIE_DOMAINSESSION_COOKIE_SECURESESSION_COOKIE_HTTPONLYSESSION_COOKIE_SAMESITE delete_cookie)r encoded_dataresponserrr_update_cookiems   zCookieStorage._update_cookieTc sg}t|}|}jrYtfddfdd|rY|r@t|fddd} |d| }|| d}nt|fd dd} || d}|d| }|||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. cs|o t|djkS)Nr )len value_encodemax_cookie_size)rI)cookierrris_too_large_for_cookiesz5CookieStorage._store..is_too_large_for_cookiecsj|jgddS)NT encode_empty) _encode_partsnot_finished_json)some_serialized_msgr'rr compute_msgs z)CookieStorage._store..compute_msgc |Sr0rmrtrnrr z&CookieStorage._store..)fnNcrur0rrvrxrrryrz)r6r<rqrlrbisect_keep_rightbisect_keep_leftri) rrKrh remove_oldestrRr3unstored_messagesserialized_messagesrgidxr)rtrmrnrr_stores.        zCookieStorage._storeFcCs|s|r |jj|tddSdS)a Return an encoded version of the serialized 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. T) serializercompressN)rQ sign_objectr@)rrKrprrrrqs zCookieStorage._encode_partscCst|}|j||dS)z Return an encoded version of the messages list which can be stored as plain text. Proxies MessagePartSerializer.dumps and _encoded_parts. ro)r6r<rq)rrKrprrrr_encodes zCookieStorage._encodec CsD|sdSz |jj|tdWStjtjtjfyYnwd|_ 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)rT) rQ unsign_objectrFr BadSignaturebinasciiErrorr;JSONDecodeErrorusedrHrrrrXszCookieStorage._decode)T)F)rrrrrWrlrYr;r<rrrPrNr\rirrqrrXrrrrrrJLs    -  rJcCsNd}t|}||kr%||d}||d|dr|}n|d}||ks |S)z Find the index of the first element from the start of the array that verifies the given condition. The function is applied from the start of the array to the pivot. rr"Nr rjar{lohimidrrrr}s r}cCsJd}t|}||kr#||d}|||dr|d}n|}||ks |S)z Find the index of the first element from the end of the array that verifies the given condition. The function is applied from the pivot to the end of array. rr"Nr rrrrrr|s  r|)rr; django.confr$django.contrib.messages.storage.baserr django.corer django.httprdjango.utils.safestringrr JSONEncoderr JSONDecoderr r6r@rFrJr}r|rrrrs