HEX

File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/sql/__pycache__/annotation.cpython-310.pyc
o

���gEG�@sUdZddlmZddlZddlmZddlmZddlmZddlmZddlmZdd	lm	Z	dd
lm
Z
ddlmZddlmZdd
lm
Z
ddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZer�ddlmZddlmZe	e efZ!ej"Z#de$d<Gdd �d e�Z%Gd!d"�d"e%�Z&Gd#d$�d$e&�Z'Gd%d&�d&e%�Z(iZ)d'e$d(<ed)d d*�Z*dQd/d0�Z+	dRd1d1dd2�dSd;d<�Z,e	dRdTd@dA��Z-e	dRdUdBdA��Z-	dRdVdDdA�Z-dWdEdF�Z.dXdKdL�Z/dYdOdP�Z0dS)Za*The :class:`.Annotated` class and related routines; creates hash-equivalent
copies of SQL constructs which contain context-specific markers and
associations.

Note that the :class:`.Annotated` concept as implemented in this module is not
related in any way to the pep-593 concept of "Annotated".


�)�annotationsN)�Any)�Callable)�cast)�Dict)�	FrozenSet)�Mapping)�Optional)�overload)�Sequence)�Tuple)�Type)�
TYPE_CHECKING)�TypeVar�)�	operators)�HasCacheKey��anon_map)�ExternallyTraversible)�InternalTraversal�)�util)�Literal)�Self)�_EntityNamespace)�_TraverseInternalsType�util.immutabledict[str, Any]�EMPTY_ANNOTATIONSc@s�eZdZUdZeZded<ded<ded<ddd
�Ze		d d!dd��Z	e		d d"dd��Z			d#d$dd�Z	e
jd%dd��Zd&dd�Z
dS)'�SupportsAnnotations�r�_annotationsz*util.generic_fn_descriptor[FrozenSet[Any]]�	proxy_set�bool�
_is_immutable�values�_AnnotationDict�returnrcC�t���N��NotImplementedError��selfr%r r �D/usr/local/lib/python3.10/dist-packages/sqlalchemy/sql/annotation.py�	_annotate>�zSupportsAnnotations._annotate.�
Literal[None]�clonecC�dSr)r �r-r%r2r r r.�_deannotateA�zSupportsAnnotations._deannotate�
Sequence[str]cCr3r)r r4r r r.r5Hr6NF�Optional[Sequence[str]]cCr(r)r*r4r r r.r5Os�Tuple[Any, ...]cCst�}|�|�Sr))r�_gen_annotations_cache_key)r-�	anon_map_r r r.�_annotations_cache_keyVs
z*SupportsAnnotations._annotations_cache_keyrcs.dt�fdd��fdd�t�j�D�D��fS)Nr!c3s2�|]\}}|t|t�r|��g�n|fVqdSr))�
isinstancer�_gen_cache_key��.0�key�valuerr r.�	<genexpr>as�	���
�zASupportsAnnotations._gen_annotations_cache_key.<locals>.<genexpr>csg|]	}|�j|f�qSr )r!)r@rA�r-r r.�
<listcomp>js��zBSupportsAnnotations._gen_annotations_cache_key.<locals>.<listcomp>)�tuple�sortedr!)r-rr )rr-r.r:\s
	���z.SupportsAnnotations._gen_annotations_cache_key�r%r&r'r�..�r%r1r2r#r'r�r%r7r2r#r'r�NF�r%r8r2r#r'r)r'r9)rrr'r9)�__name__�
__module__�__qualname__�	__slots__rr!�__annotations__r/r
r5r�memoized_propertyr<r:r r r r.r5s(

���rc@szeZdZUdZded<erejddd��Zddd�Z	dd
d�Z
e		dddd��Ze		dd dd��Z		d!d"dd�ZdS)#�SupportsWrappingAnnotationsr z*Callable[..., SupportsWrappingAnnotations]�_constructorr'rcCr3r)r rDr r r.�entity_namespaceysz,SupportsWrappingAnnotations.entity_namespacer%r&rcC�t�||�S)�greturn a copy of this ClauseElement with annotations
        updated by the given dictionary.

        ��	Annotated�_as_annotated_instancer,r r r.r/|�z%SupportsWrappingAnnotations._annotatecCrW)�hreturn a copy of this ClauseElement with annotations
        replaced by the given dictionary.

        rYr,r r r.�_with_annotations�r\z-SupportsWrappingAnnotations._with_annotations.r1r2r#cCr3r)r r4r r r.r5�r6z'SupportsWrappingAnnotations._deannotater7rcCr3r)r r4r r r.r5�r6NFr8cCs|r|��}|S|S)��return a copy of this :class:`_expression.ClauseElement`
        with annotations
        removed.

        :param values: optional tuple of individual values
         to remove.

        )�_clone)r-r%r2�sr r r.r5�s
�r'rrHrIrJrKrLrM)
rNrOrPrQrRrr�ro_non_memoized_propertyrVr/r^r
r5r r r r.rTrs&


���rTc@sxeZdZUejs
dZdejfgZde	d<dd	d
�Z
ddd�Ze	
	
dddd��Z
e	
	
dddd��Z
		dddd�Z
dS) �SupportsCloneAnnotationsr r!r�%_clone_annotations_traverse_internalsr%r&r'rcCs6|��}|j�|�|_|j�dd�|j�dd�|S)rXr<N�_generate_cache_key)r`r!�union�__dict__�pop�r-r%�newr r r.r/�s
z"SupportsCloneAnnotations._annotatecCs4|��}t�|�|_|j�dd�|j�dd�|S)r]r<Nrf)r`r�
immutabledictr!rhrirjr r r.r^�s
z*SupportsCloneAnnotations._with_annotations.r1r2r#cCr3r)r r4r r r.r5�r6z$SupportsCloneAnnotations._deannotater7rcCr3r)r r4r r r.r5�r6NFr8cCs2|s|jr|��}t��|_|j�dd�|S|S)r_r<N)r!r`rrlrhri)r-r%r2rkr r r.r5�s


rHrIrJrKrLrM)rNrOrP�typingrrQr�dp_annotations_keyrerRr/r^r
r5r r r r.rd�s$
�

���rdc@s�eZdZUdZdZed<dd	��Zd
ed<ded<d
ed<d=dd�Zd>dd�Z	d?dd�Z
d?dd�Ze		d@dAd d!��Z
e		d@dBd#d!��Z
	$	%dCdDd(d!�Z
ejsZdEd+d,�Zed-d.��ZdFd/d0�ZdGd2d3�ZdHd4d5�ZdId7d8�ZejdJd:d;��Zd$S)KrZaJclones a SupportsAnnotations and applies an 'annotations' dictionary.

    Unlike regular clones, this clone also mimics __hash__() and
    __eq__() of the original element so that it takes its place
    in hashed collections.

    A reference to the original element is maintained, for the important
    reason of keeping its hash value current.  When GC'ed, the
    hash value may be reused, causing conflicts.

    .. note::  The rationale for Annotated producing a brand new class,
       rather than placing the functionality directly within ClauseElement,
       is **performance**.  The __hash__() method is absent on plain
       ClauseElement which leads to significantly reduced function call
       overhead, as the use of sets and dictionaries against ClauseElement
       objects is prevalent, but most are not "annotated".

    F�elementrTr%r&r'cCs8zt|j}Wntyt|j|�}Ynw|||�Sr))�annotated_classes�	__class__�KeyError�_new_annotation_type)�clsror%r r r.r[
s�
z Annotated._as_annotated_instancerr!�_Annotated__element�int�_hashrt�
Type[Self]�argsrrcGs
t�|�Sr))�object�__new__)rtryr r r.r{s
zAnnotated.__new__cCsH|j��|_|j�dd�|j�dd�||_t�|�|_t|�|_dS�Nr<rf)	rh�copyrirurrlr!�hashrw)r-ror%r r r.�__init__szAnnotated.__init__cCs|j�|�}|�|�}|Sr))r!rgr^)r-r%�_valuesrkr r r.r/(s
zAnnotated._annotatecCsF|j�|j�}|j��|_|j�dd�|j�dd�t�|�|_|Sr|)rqr{rhr}rirrlr!r4r r r.r^-szAnnotated._with_annotations.r1r2r#cCr3r)r r4r r r.r55r6zAnnotated._deannotater7cCr3r)r r4r r r.r5<r6NTr8rcs2�dur|jS|�t��fdd�|j��D���S)Ncsi|]\}}|�vr||�qSr r r?�r%r r.�
<dictcomp>Ms
�z)Annotated._deannotate.<locals>.<dictcomp>)rur^rrlr!�itemsr4r r�r.r5Cs
����visitor�kwcKs|jjj||fi|��Sr))rurq�_compiler_dispatch)r-r�r�r r r.r�Ws
��zAnnotated._compiler_dispatchcCs|jjSr))rurUrDr r r.rU\szAnnotated._constructorcKs<|jjdi|��}||jur|S|j�|j�|�||j�S)Nr )rur`rh�updaterqr!)r-r�r2r r r.r``s

zAnnotated._clone�'Tuple[Type[Annotated], Tuple[Any, ...]]cCs|j|j|jffSr))rqrur!rDr r r.�
__reduce__kszAnnotated.__reduce__cCs|jSr))rwrDr r r.�__hash__nr0zAnnotated.__hash__�othercCs&|jr|jj�||�St|�t|�kSr))�_is_column_operatorsrurq�__eq__r~)r-r�r r r.r�qszAnnotated.__eq__rcCs$d|jvrtt|jd�jS|jjS)NrV)r!rrTrVrurDr r r.rVws
��zAnnotated.entity_namespace)rorTr%r&r'rZ)rtrxryrr'r)rorTr%r&rHrIrJ)r%r7r2r#r'rZ)NTrM)r�rr�rr'r)r�rr'r)r'r�)r'rv)r�rr'r#rb)rNrOrP�__doc__r��classmethodr[rRr{rr/r^r
r5rmrr��propertyrUr`r�r�r�rrcrVr r r r.rZ�sB
	




���





rZz8Dict[Type[SupportsWrappingAnnotations], Type[Annotated]]rp�_SA)�bound�to_annotaterr&r'cCs(z|j}W||�Sty|YSwr))r/�AttributeError)r�rr/r r r.�_safe_annotate�s��r�F)�detect_subquery_cols�ind_cols_on_fromclause�annotate_callablero�exclude�'Optional[Sequence[SupportsAnnotations]]r�r#r�r��OOptional[Callable[[SupportsAnnotations, _AnnotationDict], SupportsAnnotations]]cs<i�d	�������fdd��|durtt�|��}d�|S)
z�Deep copy the given ClauseElement, annotating each element
    with the given annotations dictionary.

    Elements within the exclude collection will be cloned but not annotated.

    �elemrr�rr'cs��|d<t|�}|�vr�|S�r(t|d�r(|j���r(|jdd�i|��}n'�|jkrM�r=|jr=|jdd�i|��}n|}�rG�|��}nt|��}n|}|j��d�|�|<|S)Nr�r"r2)r2r�r )	�id�hasattrr"�intersectionr`r!r$r��_copy_internals)r�r��id_�newelemr��r�rr2�
cloned_idsr�r�r�r r.r2�s0
��
�

�z_deep_annotate.<locals>.cloneN�r�rr�rr'r�rr�)rorr�r�r�r�r r�r.�_deep_annotate�s)r�r1r%r8cCr3r)r �ror%r r r.�_deep_deannotate��r�cCr3r)r r�r r r.r��r��
Optional[_SA]cs4i�d	���fdd��|durtt�|��}d�|S)
z2Deep copy the given element, removing annotations.r�rr�rr'csH�rt|�}n|}|�vr |j�dd�}|j�d�|�|<|S�|S)NT)r%r2)r2)r�r5r�)r�r�rAr��r2�clonedr%r r.r2�s
z_deep_deannotate.<locals>.cloneNr�r�r�r r�r.r��scCs|�|�}|��|S)a
Annotate the given ClauseElement and copy its internals so that
    internal objects refer to the new annotated object.

    Basically used to apply a "don't traverse" annotation to a
    selectable, without digging throughout the whole
    structure wasting time.
    )r/r�)rorr r r.�_shallow_annotates
r�rt�!Type[SupportsWrappingAnnotations]�base_cls�Type[Annotated]cCs�t|t�r|S|tvrt|S|jD]}|tvrt|}nqttttd|j||fi��t|<}|t�d|j<d|j	vrMt
|j�dtj
fg|_n|j	�dd�r`t
|j�dtj
fg|_|j	�dd�rkd|_nd|j	vrv|j	d|_t|tj�|_|S)z[Generates a new class that subclasses Annotated and proxies a given
    element type.

    zAnnotated%s�_traverse_internalsr!�
inherit_cacheFT)�
issubclassrZrp�__mro__rr
�typerN�globalsrh�listr�rrn�getr�r�ColumnOperatorsr�)rtr��super_�anno_clsr r r.rss8

��

��
rs�target_hierarchy�NonecCst�|�D]}t||�qdSr))r�walk_subclassesrs)r�r�rtr r r.�_prepare_annotationsDs�r�)r�r�rr&r'r�r))ror�rr&r�r�r�r#r�r#r�r�r'r�)ror1r%r8r'r1)ror�r%r8r'r�)ror�r%r8r'r�)ror�rr&r'r�)rtr�r�r�r'r�)r�r�r�r�r'r�)1r��
__future__rrmrrrrrrr	r
rrr
rr�r�	cache_keyr�visitorsrrrr�util.typingrr�baserr�strr&�
EMPTY_DICTrrRrrTrdrZrpr�r�r�r�r�rsr�r r r r.�<module>sn
=:K
�
��F���


/