File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/__pycache__/evaluator.cpython-310.pyc
o
���gA0 � @ s� d Z ddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddlmZ dd lm Z dd
l m
Z
dd l mZ dd lmZ dd
lmZ ddlmZ ddlmZ G dd� dej�ZG dd� dej�ZG dd� dej�Ze� Ze� ZG dd� d�Z d d d �Z d S ) z�Evaluation functions used **INTERNALLY** by ORM DML use cases.
This module is **private, for internal use by SQLAlchemy**.
.. versionchanged:: 2.0.4 renamed ``EvaluatorCompiler`` to
``_EvaluatorCompiler``.
� )� annotations)�Type� )�exc)�LoaderCallableStatus)� PassiveFlag� )�inspect)�and_)� operators)� Concatenable)�Integer)�Numeric)�warn_deprecatedc @ s e Zd ZdS )�UnevaluatableErrorN)�__name__�
__module__� __qualname__� r r �C/usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/evaluator.pyr % s r c @ � e Zd Zdd� Zdd� ZdS )� _NoObjectc O � d S �Nr ��self�arg�kwr r r �operate* � z_NoObject.operatec O r r r r r r r �reverse_operate- r z_NoObject.reverse_operateN�r r r r r r r r r r ) � r c @ r )�_ExpiredObjectc O � | S r r r r r r r 2 r z_ExpiredObject.operatec O r$ r r r r r r r 5 r z _ExpiredObject.reverse_operateNr! r r r r r# 1 r" r# c @ s
e Zd Zd6dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� ZeZeZeZeZeZeZeZ eZ eZ eZ eZ d(d)� Z!d*d+� Z"d,d-� Z#d.d/� Z$d0d1� Z%d2d3� Z&d4d5� Z'dS )7�_EvaluatorCompilerNc C s
|| _ d S r )�
target_cls)r r& r r r �__init__> �
z_EvaluatorCompiler.__init__c G sH |r
t |g|�R � }t| d|j� �d �}|s tdt|�j� ���||�S )N�visit_�Cannot evaluate )r
�getattr�__visit_name__r �typer )r �clause�clauses�methr r r �processA s �z_EvaluatorCompiler.processc C s | � |j�S r )r1 �element�r r. r r r �visit_groupingL s z!_EvaluatorCompiler.visit_groupingc C � dd� S )Nc S r r r ��objr r r �<lambda>P � z/_EvaluatorCompiler.visit_null.<locals>.<lambda>r r3 r r r �
visit_nullO � z _EvaluatorCompiler.visit_nullc C r5 )Nc S � dS )NFr r6 r r r r8 S r9 z0_EvaluatorCompiler.visit_false.<locals>.<lambda>r r3 r r r � visit_falseR r; z _EvaluatorCompiler.visit_falsec C r5 )Nc S r< )NTr r6 r r r r8 V r9 z/_EvaluatorCompiler.visit_true.<locals>.<lambda>r r3 r r r �
visit_trueU r; z _EvaluatorCompiler.visit_truec
s� z|j d }W n ty } ztd|� ��|�d }~ww | jr.t| j|j�s.td|j� ���|�� z|j| j}W n t j
yO } ztd|� ��|�d }~ww |j | j � � fdd�}|S )N� parentmapperzCannot evaluate column: z0Can't evaluate criteria against alternate class z Cannot evaluate expression: c s> | d u rt S t| �}|j}� j||tjd�}|tju r tS |S )N)�passive) �
_NO_OBJECTr �dict�getr �PASSIVE_NO_FETCHr �PASSIVE_NO_RESULT�_EXPIRED_OBJECT)r7 �state�dict_�value��implr r �get_corresponding_attr s �
z?_EvaluatorCompiler.visit_column.<locals>.get_corresponding_attr)
� _annotations�KeyErrorr r& �
issubclass�class_�_check_configure�_columntoproperty�key�orm_exc�UnmappedColumnError�
class_managerrK )r r. r? �kerS �errrL r rJ r � visit_columnX s@ ������� ����
z _EvaluatorCompiler.visit_columnc C �
| � |�S r ��visit_clauselistr3 r r r � visit_tuple� r( z _EvaluatorCompiler.visit_tuplec C rZ r r[ r3 r r r �visit_expression_clauselist� r( z._EvaluatorCompiler.visit_expression_clauselistc sX � fdd�|j D �}d|jj�d�� d�}t� |d �}|r$||j||�S td|j� ���)Nc s g | ]}� � |��qS r )r1 )�.0r. �r r r �
<listcomp>� s z7_EvaluatorCompiler.visit_clauselist.<locals>.<listcomp>r) �_�_clauselist_opz)Cannot evaluate clauselist with operator )r/ �operatorr �rstripr+ r )r r. �
evaluators�dispatchr0 r r` r r\ � s �
�z#_EvaluatorCompiler.visit_clauselistc C sj | � |j�}| � |j�}d|jj�d�� d�}t| |d �}|r'||j|||�S tdt|�j� d|j� ���)Nr) rb �
_binary_opr* � with operator ) r1 �left�rightrd r re r+ r r- )r r. � eval_left�
eval_rightrg r0 r r r � visit_binary� s ��z _EvaluatorCompiler.visit_binaryc � � fdd�}|S )Nc sH d}� D ]}|| �}|t u rt S |r dS |p |d u }q|r"d S dS �NFT�rF )r7 �has_null� sub_evaluaterI �rf r r �evaluate� s z;_EvaluatorCompiler.visit_or_clauselist_op.<locals>.evaluater �r rd rf r. ru r rt r �visit_or_clauselist_op� s
z)_EvaluatorCompiler.visit_or_clauselist_opc ro )Nc sF � D ] }|| �}|t u rt S |s |d u s|tu r d S dS qdS rp )rF rA )r7 rs rI rt r r ru � s �z<_EvaluatorCompiler.visit_and_clauselist_op.<locals>.evaluater rv r rt r �visit_and_clauselist_op� s z*_EvaluatorCompiler.visit_and_clauselist_opc ro )Nc sN g }� D ] }|| �}|t u rt S |d u s|tu r d S |�|� qt|�S r )rF rA �append�tuple)r7 �valuesrs rI rt r r ru � s zA_EvaluatorCompiler.visit_comma_op_clauselist_op.<locals>.evaluater rv r rt r � visit_comma_op_clauselist_op� s z/_EvaluatorCompiler.visit_comma_op_clauselist_opc C s( |j r | �||||�S td|j�d���)NzCustom operator zR can't be evaluated in Python unless it specifies a callable using `.python_impl`.)� python_impl�_straight_evaluater �opstring�r rd rl rm r. r r r �visit_custom_op_binary_op� s � �z,_EvaluatorCompiler.visit_custom_op_binary_opc � � �fdd�}|S )Nc s, � | �}�| �}|t u s|t u rt S ||kS r rq �r7 �left_val� right_val�rl rm r r ru � �
z7_EvaluatorCompiler.visit_is_binary_op.<locals>.evaluater �r rd rl rm r. ru r r� r �visit_is_binary_op� � z%_EvaluatorCompiler.visit_is_binary_opc r� )Nc s, � | �}�| �}|t u s|t u rt S ||kS r rq r� r� r r ru � r� z;_EvaluatorCompiler.visit_is_not_binary_op.<locals>.evaluater r� r r� r �visit_is_not_binary_op� r� z)_EvaluatorCompiler.visit_is_not_binary_opc s � ��fdd�}|S )Nc sJ � | �}�| �}|t u s|t u rt S |d u s|d u r d S �� | ��| ��S r rq r� �rl rm rd r r ru s z7_EvaluatorCompiler._straight_evaluate.<locals>.evaluater r� r r� r r~ � s
z%_EvaluatorCompiler._straight_evaluatec C sX |j jjttfvs|jjjttfvr$td|j� d|j j� d|jj� ���| �||||�S )Nz Cannot evaluate math operator "�" for datatypes �, ) rj r- �_type_affinityr r
rk r r r~ r� r r r � _straight_evaluate_numeric_only s �
���z2_EvaluatorCompiler._straight_evaluate_numeric_onlyc C � | � dd� |||�S )Nc S s | t ur| |v S d S r �rA ��a�br r r r8 ( � z:_EvaluatorCompiler.visit_in_op_binary_op.<locals>.<lambda>�r~ r� r r r �visit_in_op_binary_op& s �z(_EvaluatorCompiler.visit_in_op_binary_opc C r� )Nc S s | t ur| |vS d S r r� r� r r r r8 2 r� z>_EvaluatorCompiler.visit_not_in_op_binary_op.<locals>.<lambda>r� r� r r r �visit_not_in_op_binary_op. s �z,_EvaluatorCompiler.visit_not_in_op_binary_opc C sX t |jjjt�rt |jjjt�s"td|j� d|jj� d|jj� ���| �dd� |||�S )Nz&Cannot evaluate concatenate operator "r� r� c S s | | S r r r� r r r r8 F s z>_EvaluatorCompiler.visit_concat_op_binary_op.<locals>.<lambda>) rO rj r- r� r rk r r r~ r� r r r �visit_concat_op_binary_op8 s"
������ �z,_EvaluatorCompiler.visit_concat_op_binary_opc C r� )Nc S rZ r )�
startswithr� r r r r8 M �
zB_EvaluatorCompiler.visit_startswith_op_binary_op.<locals>.<lambda>r� r� r r r � visit_startswith_op_binary_opI � �z0_EvaluatorCompiler.visit_startswith_op_binary_opc C r� )Nc S rZ r )�endswithr� r r r r8 T r� z@_EvaluatorCompiler.visit_endswith_op_binary_op.<locals>.<lambda>r� r� r r r �visit_endswith_op_binary_opP r� z._EvaluatorCompiler.visit_endswith_op_binary_opc sD | � |j�� |jtju r� fdd�}|S tdt|�j� d|j� ���)Nc s& � | �}|t u r
t S |d u rd S | S r rq )r7 rI ��
eval_innerr r ru [ s z0_EvaluatorCompiler.visit_unary.<locals>.evaluater* ri )r1 r2 rd r �invr r- r )r r. ru r r� r � visit_unaryW s ��z _EvaluatorCompiler.visit_unaryc s" |j r|� � � n|j� � fdd�S )Nc s � S r r r6 ��valr r r8 n r9 z4_EvaluatorCompiler.visit_bindparam.<locals>.<lambda>)�callablerI r3 r r� r �visit_bindparami s
z"_EvaluatorCompiler.visit_bindparamr )(r r r r' r1 r4 r: r= r> rY r] r^ r\ rn rw rx r| r� r� r� r~ r� �visit_add_binary_op�visit_mul_binary_op�visit_sub_binary_op�visit_mod_binary_op�visit_truediv_binary_op�visit_lt_binary_op�visit_le_binary_op�visit_ne_binary_op�visit_gt_binary_op�visit_ge_binary_op�visit_eq_binary_opr� r� r� r� r� r� r� r r r r r% = sL
6
r% �name�str�return�Type[_EvaluatorCompiler]c C s* | dkr t dd� tS tdt�d| ����)N�EvaluatorCompilerz�Direct use of 'EvaluatorCompiler' is not supported, and this name will be removed in a future release. '_EvaluatorCompiler' is for internal use onlyz2.0zmodule z has no attribute )r r% �AttributeErrorr )r� r r r � __getattr__q s �r� N)r� r� r� r� ) �__doc__�
__future__r �typingr � r rT �baser r r �sqlr
r � sql.sqltypesr r
r �utilr �InvalidRequestErrorr �ColumnOperatorsr r# rA rF r% r� r r r r �<module> s. 6