File: //usr/lib/python3/dist-packages/django/contrib/messages/storage/__pycache__/cookie.cpython-310.pyc
o
3�a� � @ s� d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl m
Z
d dl m Z m
Z
d dlmZmZ G dd � d ej�ZG d
d � d ej�ZG d d
� d
�ZG dd� de�ZdS )� N)�settings)� BaseStorage�Message)�signing)� SimpleCookie)�constant_time_compare� salted_hmac)�SafeData� mark_safec s$ e Zd ZdZdZ� fdd�Z� ZS )�MessageEncoderzI
Compactly serialize instances of the ``Message`` class as JSON.
�__json_messagec sR t |t�r#t |jt�r
dnd}| j||j|jg}|jr!|�|j� |S t� � |�S )N� r )
�
isinstancer �messager � message_key�level�
extra_tags�append�super�default)�self�obj� is_safedatar �� __class__� �H/usr/lib/python3/dist-packages/django/contrib/messages/storage/cookie.pyr s
zMessageEncoder.default)�__name__�
__module__� __qualname__�__doc__r r �
__classcell__r r r r r s r c s( e Zd ZdZdd� Z� fdd�Z� ZS )�MessageDecoderzE
Decode JSON that includes serialized ``Message`` instances.
c sz t |t�r+|r+|d tjkr"|d rt|d �|d< t|dd � � S � fdd�|D �S t |t�r;� fdd�|�� D �S |S ) Nr r
� � c s g | ]}� � |��qS r ��process_messages)�.0�item�r r r �
<listcomp>( s z3MessageDecoder.process_messages.<locals>.<listcomp>c s i | ]
\}}|� � |��qS r r% )r' �key�valuer) r r �
<dictcomp>* s �z3MessageDecoder.process_messages.<locals>.<dictcomp>)r �listr r r
r �dict�items�r r r r) r r& " s
�z MessageDecoder.process_messagesc s t � j|fi |��}| �|�S )N)r �decoder& )r �s�kwargs�decodedr r r r2 . s
zMessageDecoder.decode)r r r r r& r2 r! r r r r r" s r" c @ s e Zd Zdd� Zdd� ZdS )�MessageSerializerc C s t j|dtd��d�S )N)�,�:)�
separators�cls�latin-1)�json�dumpsr �encoder1 r r r r= 4 s ��zMessageSerializer.dumpsc C s t j|�d�td�S )Nr; �r: )r< �loadsr2 r" )r �datar r r r@ ; s zMessageSerializer.loadsN)r r r r= r@ r r r r r6 3 s r6 c sl e Zd ZdZdZdZdZdZ� fdd�Zdd � Z d
d � Z
dd
d�Z dd� Z ddd�Z
dd� Zdd� Z� ZS )�
CookieStoragez%
Store messages in a cookie.
�messagesi �__messagesnotfinished__�django.contrib.messagesc s& t � j|i |�� tj| jd�| _d S )N)�salt)r �__init__r �get_cookie_signer�key_salt�signer)r �argsr4 r r r rG K s zCookieStorage.__init__c O sF | j j�| 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.
���)�request�COOKIES�get� cookie_name�_decode� not_finished�pop)r rK r4 rA rC �
all_retrievedr r r �_getO s
zCookieStorage._getc C sJ |r|j | j|tjtjp
dtjpdtjd� dS |j| jtjtjd� dS )zz
Either set the cookie with the encoded data if there is any data to
store, or delete the cookie.
N)�domain�secure�httponly�samesite)rV rY )�
set_cookierP r �SESSION_COOKIE_DOMAIN�SESSION_COOKIE_SECURE�SESSION_COOKIE_HTTPONLY�SESSION_COOKIE_SAMESITE�
delete_cookie)r � encoded_data�responser r r �_update_cookie^ s
�
�z CookieStorage._update_cookieTc s� g }| � |�}| jrCt� � � fdd�}|rC||�| jkrC|r'|�|�d�� n|�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.
c s t � �| �d �S )Nr
)�len� value_encode)�val��cookier r �
stored_length� s z+CookieStorage._store.<locals>.stored_lengthr )� encode_empty)�_encode�max_cookie_sizer r rS �insertrR rb ) r rC ra �
remove_oldestrK r4 �unstored_messagesr` rh r rf r �_storer s
�� zCookieStorage._storec C s d}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.
rE )r � hexdigest)r r, rI r r r � _legacy_hash� s zCookieStorage._legacy_hashFc C s |s|r
| j j|tdd�S dS )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)�
serializer�compressN)rJ � sign_objectr6 )r rC ri r r r rj � s �zCookieStorage._encodec C s� |sdS z | j j|td�W S tjy | �|�}Y n tjtj fy. | j �
|�}Y nw |rDztj |t d�W S tj yC Y nw d| _
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)rr r? T)rJ �
unsign_objectr6 r � BadSignature�_legacy_decode�binascii�Errorr< �JSONDecodeError�unsignr@ r" �used)r rA r5 r r r rQ � s" ��zCookieStorage._decodec C s8 |� dd�}t|�dkr|\}}t|| �|��r|S d S )N�$r
r$ )�splitrc r rq )r rA �bits�hash_r, r r r rw � s z CookieStorage._legacy_decode)T)F)r r r r rP rk rR rI rG rU rb ro rq rj rQ rw r! r r r r rB ? s
"rB )rx r< � django.confr �$django.contrib.messages.storage.baser r � django.corer � django.httpr �django.utils.cryptor r �django.utils.safestringr r
� JSONEncoderr � JSONDecoderr" r6 rB r r r r �<module> s