o b@sddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZmZGdd d ejZGd d d ejZGd d d ZGdddeZdS)N)settings) BaseStorageMessage)signing) SimpleCookie)constant_time_compare salted_hmac)SafeData mark_safecs$eZdZdZdZfddZZS)MessageEncoderzI Compactly serialize instances of the ``Message`` class as JSON. Z__json_messagecsRt|tr#t|jtr dnd}|j||j|jg}|jr!||j|St |S)Nr) isinstancermessager message_keylevelZ extra_tagsappendsuperdefault)selfobjZ is_safedatar __class__e/var/www/secure340b-portal/env/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 rr rdictitemsrrrr&rr#"s  zMessageDecoder.process_messagesc stj|fi|}||S)N)rdecoder#)rskwargsdecodedrrrr/.s zMessageDecoder.decode)rrrrr#r/rrrrrrs rc@seZdZddZddZdS)MessageSerializercCstj|dtddS)N),:) separatorsclslatin-1)jsondumpsr encoder.rrrr:4szMessageSerializer.dumpscCstj|dtdS)Nr8r7)r9loadsr/r)rdatarrrr=;szMessageSerializer.loadsN)rrrr:r=rrrrr33s r3csleZdZdZdZdZdZdZfddZdd Z d d Z dd dZ ddZ dddZ ddZddZZS) CookieStoragez% Store messages in a cookie. messagesiZ__messagesnotfinished__django.contrib.messagescs&tj|i|tj|jd|_dS)N)salt)r__init__rZget_cookie_signerkey_saltsigner)rargsr1rrrrCKszCookieStorage.__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. )requestZCOOKIESget cookie_name_decode not_finishedpop)rrFr1r>r@Z all_retrievedrrr_getOs  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)rOrR) set_cookierJrZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREZSESSION_COOKIE_HTTPONLYZSESSION_COOKIE_SAMESITEZ delete_cookie)r encoded_dataresponserrr_update_cookie^s  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 )len value_encode)valcookierr stored_lengthsz+CookieStorage._store..stored_lengthr) encode_empty)_encodemax_cookie_sizerrrMinsertrLrV) rr@rUZ remove_oldestrFr1Zunstored_messagesrTr\rrZr_storers   zCookieStorage._storecCsd}t||S)z # RemovedInDjango40Warning: pre-Django 3.1 hashes will be invalid. 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. rA)r hexdigest)rr)rDrrr _legacy_hashs zCookieStorage._legacy_hashFcCs|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)rEZ sign_objectr3)rr@r]rrrr^szCookieStorage._encodec Cs|sdSz |jj|tdWStjy||}Yntjtj fy.|j |}Ynw|rDztj |t dWStj yCYnwd|_ 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)rdr<T)rEZ unsign_objectr3rZ BadSignature_legacy_decodebinasciiErrorr9JSONDecodeErrorZunsignr=rused)rr>r2rrrrKs"zCookieStorage._decodecCs8|dd}t|dkr|\}}t|||r|SdS)N$r r!)splitrWrrc)rr>bitshash_r)rrrrfs  zCookieStorage._legacy_decode)T)F)rrrrrJr_rLrDrCrNrVrarcr^rKrfrrrrrr??s    "r?)rgr9Z django.confrZ$django.contrib.messages.storage.baserrZ django.corerZ django.httprZdjango.utils.cryptorrZdjango.utils.safestringr r JSONEncoderr JSONDecoderrr3r?rrrrs