U ۼ|e@sddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z m Z GdddejZGd d d ejZGd d d ZGd ddeZdS)N)settings) BaseStorageMessage)signing) SimpleCookie)SafeData mark_safecs$eZdZdZdZfddZZS)MessageEncoderzI Compactly serialize instances of the ``Message`` class as JSON. Z__json_messagecsVt|trJt|jtrdnd}|j||j|jg}|jdk rF||j|St |S)Nr) isinstancermessager message_keylevel extra_tagsappendsuperdefault)selfobjZ is_safedatar  __class__c/var/www/website-v5/atlas_env/lib/python3.8/site-packages/django/contrib/messages/storage/cookie.pyrs   zMessageEncoder.default)__name__ __module__ __qualname____doc__r r __classcell__rrrrr sr cs(eZdZdZddZfddZZS)MessageDecoderzE Decode JSON that includes serialized ``Message`` instances. cszt|trV|rV|dtjkrD|dr4t|d|d<t|ddSfdd|DSt|trvfdd|DS|S) Nrr csg|]}|qSrprocess_messages).0itemrrr (sz3MessageDecoder.process_messages..csi|]\}}||qSrr!)r#keyvaluer%rr *sz3MessageDecoder.process_messages..)r listr r rrdictitemsrrrr%rr""s zMessageDecoder.process_messagesc stj|f|}||S)N)rdecoder")rskwargsdecodedrrrr.-szMessageDecoder.decode)rrrrr"r.rrrrrrs rc@seZdZddZddZdS)MessageSerializercCstj|dtddS)N),:) separatorsclslatin-1)jsondumpsr encoder-rrrr93szMessageSerializer.dumpscCstj|dtdS)Nr7)r6)r8loadsr.rrdatarrrr;:szMessageSerializer.loadsN)rrrr9r;rrrrr22sr2cs\eZdZdZdZdZdZdZfddZdd Z d d Z dd dZ dddZ ddZ ZS) CookieStoragez% Store messages in a cookie. messagesiZ__messagesnotfinished__zdjango.contrib.messagescs"tj||tj|jd|_dS)N)salt)r__init__rget_cookie_signerkey_saltsigner)rargsr0rrrrAKszCookieStorage.__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)rrEr0r=r? all_retrievedrrr_getOs  zCookieStorage._getcCsH|r.|j|j|tjtjpdtjp"dtjdn|j|jtjtjddS)zz Either set the cookie with the encoded data if there is any data to store, or delete the cookie. N)domainsecurehttponlysamesite)rPrS) set_cookierJrSESSION_COOKIE_DOMAINSESSION_COOKIE_SECURESESSION_COOKIE_HTTPONLYSESSION_COOKIE_SAMESITE delete_cookie)r encoded_dataresponserrr_update_cookie^s zCookieStorage._update_cookieTc sg}||}|jrvtfdd}|rv|||jkrv|rN||dn|d||j||jg|d}q&||||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 )len value_encode)valcookierr stored_lengthsz+CookieStorage._store..stored_lengthr) encode_empty)_encodemax_cookie_sizerrrMinsertrLr\) rr?r[ remove_oldestrEr0unstored_messagesrZrbrr`r_storess     zCookieStorage._storeFcCs|s|r|jj|tddSdS)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. T) serializercompressN)rD sign_objectr2)rr?rcrrrrds zCookieStorage._encodec CsF|sdSz|jj|tdWStjtjtjfk r:YnXd|_ 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)rjT) rD unsign_objectr2r BadSignaturebinasciiErrorr8JSONDecodeErrorusedr<rrrrKszCookieStorage._decode)T)F)rrrrrJrerLrCrArOr\rirdrKrrrrrr>>s   r>)ror8 django.confr$django.contrib.messages.storage.baserrZ django.corer django.httprdjango.utils.safestringrr JSONEncoderr JSONDecoderrr2r>rrrrs