|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/django/contrib/postgres/fields/__pycache__/ |
Upload File : |
o
�a/) � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z d dl
m Z m Z m
Z
mZmZ d dlmZ d dlmZmZ d d lmZ d
d lmZ d d
lmZ dgZG dd� dee �Z G dd� d�Z e j G dd� de ej ��Z e j G dd� de ej!��Z"e j G dd� de e��Z#e j G dd� de ej$��Z%e j G dd� de��Z&e j G d d � d e��Z'G d d � d e�Z(G d d!� d!�Z)G d"d#� d#e�Z*G d$d%� d%�Z+dS )&� N)�lookups)�SimpleArrayField)�ArrayMaxLengthValidator)�checks�
exceptions)�Field�Func� IntegerField� Transform�Value)�CheckFieldDefaultMixin)�Exact�In)� gettext_lazy� )�prefix_validation_error� )�AttributeSetter�
ArrayFieldc s� e Zd ZdZed�ed�d�ZdZd,� fdd� Zed d
� �Z e j
d d
� �Z e � fd d
��Z � fdd�Z
� fdd�Zedd� �Zdd� Zdd� Zdd� Zd-dd�Z� fd d �Zd d � Zd d!� Zd"d#� Z� fd$d%�Z� fd&d'�Z� fd(d)�Z� fd*d+�Z� Z S ).r Fz+Item %(nth)s in the array did not validate:z(Nested arrays must have the same length.)� item_invalid�nested_array_mismatch)�listz[]Nc sR || _ || _| jrg | j�t| j��| _t| j d�r | j| _t� jdi |�� d S )N�
from_db_value� ) �
base_field�size�default_validatorsr �hasattr�_from_db_valuer �super�__init__)�selfr r �kwargs�� __class__r �F/usr/lib/python3/dist-packages/django/contrib/postgres/fields/array.pyr s zArrayField.__init__c C s, z| j d W S ty td| jj ��w )N�modelz$'%s' object has no attribute 'model')�__dict__�KeyError�AttributeErrorr$ �__name__�r! r r r% r& % s
�zArrayField.modelc C s || j d< || j_d S )Nr& )r' r r& )r! r& r r r% r& , s
c s t |ttf�p t� �|�S �N)�
isinstancer �tupler �_choices_is_value)�cls�valuer# r r% r/ 1 s z ArrayField._choices_is_valuec st t � jd i |��}| jjr|�tjd| dd�� |S | j�� }|r8d�dd� |D ��}|�tjd| | dd�� |S )
Nz/Base field for array cannot be a related field.z
postgres.E002)�obj�idz
c s s � | ] }d |j |jf V qdS )z%s (%s)N)�msgr3 )�.0�errorr r r% � <genexpr>C s � z#ArrayField.check.<locals>.<genexpr>z'Base field for array has errors:
%sz
postgres.E001r )r �checkr � remote_field�appendr �Error�join)r! r" �errors� base_errors�messagesr# r r% r8 5 s* ��
���zArrayField.checkc s t � �|� | j�|� d S r, )r �set_attributes_from_namer )r! �namer# r r% r@ M s z#ArrayField.set_attributes_from_namec C s d| j j S )Nz Array of %s)r � descriptionr+ r r r% rB Q s zArrayField.descriptionc C � | j pd}d| j�|�|f S �N� z%s[%s])r r �db_type�r! �
connectionr r r r% rF U �
zArrayField.db_typec C rC rD )r r � cast_db_typerG r r r% rJ Y rI zArrayField.cast_db_typec C s d� | �|��S )Nz%s::{})�formatrF )r! r1 �compilerrH r r r% �get_placeholder] s zArrayField.get_placeholderc s& t |ttf�r� �fdd�|D �S |S )Nc s g | ] }�j j|� d d��qS )F)�prepared)r �get_db_prep_value�r5 �i�rH r! r r% �
<listcomp>b s z0ArrayField.get_db_prep_value.<locals>.<listcomp>)r- r r. )r! r1 rH rN r rR r% rO ` s z ArrayField.get_db_prep_valuec sB t � �� \}}}}|dkrd}|�| j�� | jd�� ||||fS )Nz/django.contrib.postgres.fields.array.ArrayFieldz)django.contrib.postgres.fields.ArrayField)r r )r � deconstruct�updater �cloner )r! rA �path�argsr" r# r r% rT e s � zArrayField.deconstructc s* t |t�rt�|�}� fdd�|D �}|S )Nc s g | ]}� j �|��qS r )r � to_python)r5 �valr+ r r% rS s s z(ArrayField.to_python.<locals>.<listcomp>)r- �str�json�loads)r! r1 �valsr r+ r% rY o s
zArrayField.to_pythonc s" |d u r|S � ��fdd�|D �S )Nc s g | ]
}�j �|�� ��qS r )r r )r5 �item�rH �
expressionr! r r% rS y s ��z-ArrayField._from_db_value.<locals>.<listcomp>r )r! r1 ra rH r r` r% r v s
�zArrayField._from_db_valuec C sX g }| � |�}| j}|D ]}|d u r|�d � q t|j|�}|�|�|�� q t�|�S r, )�value_from_objectr r: r �attname�value_to_stringr\ �dumps)r! r2 �valuesr^ r rZ r r r% rd ~ s
zArrayField.value_to_stringc s� t � �|�}|r
|S d|vr(zt|�}W n ty Y n w |d7 }t|| j�S z|�d�\}}t|�d }t|�}W n
tyE Y d S w t||�S )N�_r )r �
get_transform�int�
ValueError�IndexTransformFactoryr �split�SliceTransformFactory)r! rA � transform�index�start�endr# r r% rh � s& � �
zArrayField.get_transformc s� t � �||� t|�D ])\}}z | j�||� W q tjy4 } zt|| jd dd|d id��d }~ww t| jt �rPt
dd� |D ��dkrRtj| jd dd��d S d S ) Nr �nthr ��prefix�code�paramsc S s h | ]}t |��qS r )�lenrP r r r% � <setcomp>� s z&ArrayField.validate.<locals>.<setcomp>r )ru ) r �validate� enumerater r �ValidationErrorr �error_messagesr- r rw )r! r1 �model_instancero �partr6 r# r r% ry � s*
��� ��zArrayField.validatec sj t � �|� t|�D ](\}}z| j�|� W q
tjy2 } zt|| jd dd|d id��d }~ww d S )Nr rr r rs )r �run_validatorsrz r r r{ r r| )r! r1 ro r~ r6 r# r r% r � s
����zArrayField.run_validatorsc s& t � jdi t| j�� | jd�|���S )N)�
form_classr �
max_lengthr )r � formfieldr r r )r! r" r# r r% r� � s
��zArrayField.formfieldr, )F) r* �
__module__� __qualname__�empty_strings_allowedrg �default_error_messages�
_default_hintr �propertyr& �setter� classmethodr/ r8 r@ rB rF rJ rM rO rT rY r rd rh ry r r� �
__classcell__r r r# r% r s: �
c s( e Zd Z� fdd�Z� fdd�Z� ZS )�
ArrayRHSMixinc sh t |ttf�r+g }|D ]}t|d�s |j}t|j�|��}|�|� q t |ddd��}t
� � ||� d S )N�resolve_expression�ARRAYz %(function)s[%(expressions)s])�function�template) r- r. r r � output_fieldr r �get_prep_valuer: r r r )r! �lhs�rhs� expressionsr1 �fieldr# r r% r � s
�zArrayRHSMixin.__init__c s0 t � �||�\}}| jj�|�}d||f |fS )Nz%s::%s)r � process_rhsr� r� rJ )r! rL rH r� �
rhs_params� cast_typer# r r% r� � s zArrayRHSMixin.process_rhs)r* r� r� r r� r� r r r# r% r� � s r� c @ � e Zd ZdS )�
ArrayContainsN�r* r� r� r r r r% r� � � r� c @ r� )�ArrayContainedByNr� r r r r% r� � r� r� c @ r� )�
ArrayExactNr� r r r r% r� � r� r� c @ r� )� ArrayOverlapNr� r r r r% r� � r� r� c @ s e Zd ZdZe� Zdd� ZdS )�ArrayLenTransformrw c C s |� | j�\}}dd|i |fS )NzRCASE WHEN %(lhs)s IS NULL THEN NULL ELSE coalesce(array_length(%(lhs)s, 1), 0) ENDr� )�compiler� �r! rL rH r� rv r r r% �as_sql� s ��zArrayLenTransform.as_sqlN)r* r� r� � lookup_namer r� r� r r r r% r� � s r� c s e Zd Z� fdd�Z� ZS )�
ArrayInLookupc sN t � �� }t|d�r |S g }|D ]}t|d�r |�|� q|�t|�� q|S )Nr� )r �get_prep_lookupr r: r. )r! rf �prepared_valuesr1 r# r r% r� s
z ArrayInLookup.get_prep_lookup)r* r� r� r� r� r r r# r% r� s r� c s0 e Zd Z� fdd�Zdd� Zedd� �Z� ZS )�IndexTransformc �" t � j|i |�� || _|| _d S r, )r r ro r )r! ro r rX r" r# r r% r �
zIndexTransform.__init__c C s$ |� | j�\}}d| || jg fS )Nz%s[%%s])r� r� ro r� r r r% r� s zIndexTransform.as_sqlc C s | j S r, )r r+ r r r% r� s zIndexTransform.output_field)r* r� r� r r� r� r� r� r r r# r% r� s
r� c @ � e Zd Zdd� Zdd� ZdS )rk c C � || _ || _d S r, )ro r )r! ro r r r r% r $ �
z IndexTransformFactory.__init__c O � t | j| jg|�R i |��S r, )r� ro r �r! rX r" r r r% �__call__( � z IndexTransformFactory.__call__N�r* r� r� r r� r r r r% rk " � rk c s$ e Zd Z� fdd�Zdd� Z� ZS )�SliceTransformc r� r, )r r rp rq )r! rp rq rX r" r# r r% r . r� zSliceTransform.__init__c C s( |� | j�\}}d| || j| jg fS )Nz %s[%%s:%%s])r� r� rp rq r� r r r% r� 3 s zSliceTransform.as_sql)r* r� r� r r� r� r r r# r% r� , s r� c @ r� )rm c C r� r, )rp rq )r! rp rq r r r% r : r� z SliceTransformFactory.__init__c O r� r, )r� rp rq r� r r r% r� > r� z SliceTransformFactory.__call__Nr� r r r r% rm 8 r� rm ),r\ �django.contrib.postgresr � django.contrib.postgres.formsr �"django.contrib.postgres.validatorsr � django.corer r �django.db.modelsr r r r
r � django.db.models.fields.mixinsr �django.db.models.lookupsr
r �django.utils.translationr rg �utilsr r �__all__r r� �register_lookup� DataContainsr� � ContainedByr� r� �Overlapr� r� r� r� rk r� rm r r r r% �<module> s> 8