관리-도구
편집 파일: clsregistry.cpython-38.pyc
U -?�f&F � @ s� U d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z dd lm Z dd lm!Z! dd lm Z dd l"m#Z# dd l$m%Z% e�r�dd!l&m'Z' dd"l"m(Z( dd#l"m)Z) ed$ed%�Z*e e+ee,d&f f Z-e.� Z/d'e0d(< d)d*d+d,d-�d.d/�Z1d)d0d+d,d-�d1d2�Z2d)d+d3d4d5�d6d7�Z3G d8d&� d&�Z4G d9d:� d:e4�Z5G d;d<� d<e4�Z6G d=d>� d>�Z7G d?d@� d@�Z8e!�9e8�dAdB� � G dCdD� dD�Z:d)dEd@dF�dGdH�Z;G dIdJ� dJ�Z<da=dKe0dL< d0dMdNdO�dPdQ�Z>dS )Rz�Routines to handle the string class registry used by declarative. This system allows specification of classes and expressions used in :func:`_orm.relationship` using strings. � )� annotationsN)�Any)�Callable)�cast)�Dict)� Generator)�Iterable)�List)�Mapping)�MutableMapping)�NoReturn)�Optional)�Set)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union� )� attributes)� interfaces)�SynonymProperty)�ColumnProperty)� class_mapper� )�exc)� inspection)�util)�_get_table_key)�CallableReference)�RelationshipProperty)�MetaData)�Table�_T)�bound�ClsRegistryTokenzSet[ClsRegistryToken]� _registries�strzType[_T]�_ClsRegistryType�None)� classname�cls�decl_class_registry�returnc C s | |kr6|| }t |t�s>t|td|�g� }|| < n||| < ztt|d �}W n& tk rv tdd� |d< }Y nX |j�d�}|�r |�d�}|�|�}|D ]}|�|�}q�z|� | |� W q� t k �r } z&t |t�s�t � d| � d��|�n� W 5 d}~X Y q�X q�dS )z^Add a class to the _decl_class_registry associated with the given declarative class. � Type[Any]�_sa_module_registryN�.r �name "�-" matches both a class name and a module name) � isinstance�_MultipleClassMarkerr � _ModuleMarker�KeyError� __module__�split�pop� get_module� add_class�AttributeErrorr �InvalidRequestError) r* r+ r, �existing� root_module�tokens�token�module�ae� rD �J/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/clsregistry.pyr; @ s> � �� ��r; r. c C s� | |kr,|| }t |t�r&|�|� n|| = ztt|d �}W n tk rT Y d S X |j�d�}|r�|�d�}|� |�}|D ]}|� |�}q~z|� | |� W qb t k r� t |t�s�n� Y qbX qbd S )Nr/ r0 r ) r3 r4 � remove_itemr r5 r6 r7 r8 r9 r: � remove_classr< )r* r+ r, r> r? r@ rA rB rD rD rE rG v s0 � rG zCallable[[Any], bool]�bool)�keyr, �testr- c C sP | |kr dS || }t |t�rB|jD ]}||�r$ dS q$td��n ||� S dS )a� test if a key is empty of a certain object. used for unit tests against the registry to see if garbage collection is working. "test" is a callable that will be passed an object should return True if the given object is the one we were looking for. We can't pass the actual object itself b.c. this is for testing garbage collection; the caller will have to have removed references to the object itself. TFzunknown codepathN)r3 r4 �contents�NotImplementedError)rI r, rJ �thingZ sub_thingrD rD rE � _key_is_empty� s rN c @ s e Zd ZdZdZdS )r% zAan object that can be in the registry._class_registry as a value.rD N)�__name__r7 � __qualname__�__doc__� __slots__rD rD rD rE r% � s c @ s� e Zd ZU dZdZded< ded< d"dd d �d d �Zd dd�dd�Zdd�dd�Zddd d�dd�Z ddd �d d �Z d dd �d d!�Z dS )#r4 zRrefers to multiple classes of the same name within _decl_class_registry. )� on_removerK � __weakref__zSet[weakref.ref[Type[Any]]]rK z/CallableReference[Optional[Callable[[], None]]]rS NzIterable[Type[Any]]z Optional[Callable[[], None]])�classesrS c s( |� _ � fdd�|D �� _t�� � d S )Nc s h | ]}t �|� j��qS rD )�weakref�ref� _remove_item)�.0�item��selfrD rE � <setcomp>� s z0_MultipleClassMarker.__init__.<locals>.<setcomp>)rS rK r& �add)r\ rU rS rD r[ rE �__init__� s �z _MultipleClassMarker.__init__r. r) )r+ r- c C s | � t�|�� d S �N)rX rV rW �r\ r+ rD rD rE rF � s z _MultipleClassMarker.remove_itemz*Generator[Optional[Type[Any]], None, None]�r- c C s dd� | j D �S )Nc s s | ] }|� V qd S r` rD �rY rW rD rD rE � <genexpr>� s z0_MultipleClassMarker.__iter__.<locals>.<genexpr>�rK r[ rD rD rE �__iter__� s z _MultipleClassMarker.__iter__� List[str]r' )�pathrI r- c C sV t | j�dkr*t�dd�||g � ��n(t| j�d }|� }|d krNt|��|S d S )Nr zxMultiple classes found for path "%s" in the registry of this declarative base. Please use a fully module-qualified path.r0 r )�lenrK r r= �join�list� NameError)r\ rh rI rW r+ rD rD rE � attempt_get� s ��z _MultipleClassMarker.attempt_getzweakref.ref[Type[Any]])rW r- c C s. | j �|� | j s*t�| � | jr*| �� d S r` )rK �discardr& rS )r\ rW rD rD rE rX � s z!_MultipleClassMarker._remove_item)rZ r- c C sX dd� dd� t | j�D �D �}|j|kr>t�d|j|jf � | j�t�|| j �� d S )Nc S s h | ]}|d k r|j �qS r` )r7 )rY r+ rD rD rE r] � s �z0_MultipleClassMarker.add_item.<locals>.<setcomp>c S s g | ] }|� �qS rD rD rc rD rD rE � <listcomp>� s z1_MultipleClassMarker.add_item.<locals>.<listcomp>z�This declarative base already contains a class with the same class name and module name as %s.%s, and will be replaced in the string-lookup table.) rk rK r7 r �warnrO r^ rV rW rX )r\ rZ �modulesrD rD rE �add_item� s � ��z _MultipleClassMarker.add_item)N) rO r7 rP rQ rR �__annotations__r_ rF rf rm rX rr rD rD rD rE r4 � s � r4 c @ s� e Zd ZU dZdZded< ded< ded< d ed < d dd �d d�Zd dd�dd�Zd dd�dd�Zd dd�dd�Z d dd�dd �Z d d d�d d �Z d d dd �d!d"�Z d d dd �d#d$�Z d%S )&r5 z>Refers to a module name within _decl_class_registry. )�parent�namerK �mod_nsrh rT zOptional[_ModuleMarker]rt z5Dict[str, Union[_ModuleMarker, _MultipleClassMarker]]rK �_ModNSrv rg rh r' )ru rt c C sJ || _ || _i | _t| �| _| j r6| j j| jg | _ng | _t�| � d S r` )rt ru rK rw rv rh r& r^ )r\ ru rt rD rD rE r_ s z_ModuleMarker.__init__rH )ru r- c C s || j kS r` re �r\ ru rD rD rE � __contains__ s z_ModuleMarker.__contains__r% c C s | j | S r` re rx rD rD rE � __getitem__ s z_ModuleMarker.__getitem__r) c C s: | j �|d � | j s6| jd k r6| j�| j� t�| � d S r` )rK r9 rt rX ru r& rn rx rD rD rE rX s z_ModuleMarker._remove_item�Union[_ModNS, Type[Any]]�rI r- c C s | j �|�S r` )rv � __getattr__�r\ rI rD rD rE � resolve_attr& s z_ModuleMarker.resolve_attrc C s4 || j kr t|| �}|| j |< ntt| j | �}|S r` )rK r5 r )r\ ru �markerrD rD rE r: ) s z_ModuleMarker.get_moduler. )ru r+ r- c s� � �j krptt�j � �}z|�|� W q� tk rl } z&t|t�sZt�d� � d��|�n� W 5 d }~X Y q�X n"t|g� �fdd�d� }�j � < d S )Nr1 r2 c s �� � �S r` )rX rD �ru r\ rD rE �<lambda>@ � z)_ModuleMarker.add_class.<locals>.<lambda>)rS )rK r r4 rr r< r3 r r= )r\ ru r+ r> rC rD r� rE r; 1 s �� �z_ModuleMarker.add_classc C s( || j kr$tt| j | �}|�|� d S r` )rK r r4 rF )r\ ru r+ r> rD rD rE rG C s z_ModuleMarker.remove_classN)rO r7 rP rQ rR rs r_ ry rz rX r r: r; rG rD rD rD rE r5 s r5 c @ s8 e Zd ZU dZded< dd�dd�Zddd �d d �Zd S ) rw )Z__parentr5 �_ModNS__parent)rt c C s || _ d S r` )r� )r\ rt rD rD rE r_ N s z_ModNS.__init__r' r{ r| c C st z| j j| }W n tk r$ Y n8X |d k r\t|t�r>|jS t|t�sLt�|�| j j |�S t d| j j |f ��d S )Nz<Module %r has no mapped classes registered under the name %r) r� rK r6 r3 r5 rv r4 �AssertionErrorrm rh rl ru )r\ rI �valuerD rD rE r} Q s ��z_ModNS.__getattr__N�rO r7 rP rR rs r_ r} rD rD rD rE rw I s rw c @ s8 e Zd ZU dZded< dd�dd�Zddd�d d �Zd S ) � _GetColumns�r+ r. r+ c C s || _ d S r` r� ra rD rD rE r_ h s z_GetColumns.__init__r' r r| c C s� t | jdd�}|r�||jkr.td| j|f ��|j| }|jtjjkr�t|t j �sVt �|j }t|t �rn|j}nt|t�s�t�d| ��t| j|�S )NF)� configurez/Class %r does not have a mapped column named %rzaProperty %r is not an instance of ColumnProperty (i.e. does not correspond directly to a Column).)r r+ Zall_orm_descriptorsr< Zextension_typer Z NotExtensionZ NOT_EXTENSIONr3 r ZQueryableAttributer� �propertyr ru r r r= �getattr)r\ rI �mp�desc�proprD rD rE r} k s* �� ��z_GetColumns.__getattr__Nr� rD rD rD rE r� c s r� c C s t �| j�S r` )r �inspectr+ )�targetrD rD rE r� � r� r� c @ sB e Zd ZU dZded< ded< ddd�dd�Zddd �d d �Zd S ) � _GetTable�rI �metadatar' rI r! r� c C s || _ || _d S r` r� )r\ rI r� rD rD rE r_ � s z_GetTable.__init__r" r| c C s | j jt|| j� S r` )r� �tablesr rI r~ rD rD rE r} � s z_GetTable.__getattr__Nr� rD rD rD rE r� � s r� r )rI r� r- c C s t |t�r|�g | �}t|�S r` )r3 r4 rm r� )rI r� rD rD rE �_determine_container� s r� c @ s� e Zd ZU dZded< ded< ded< ded < d ed < d ed < d"ddddd d�dd�Zddd�dd�Zdddd�dd�Zdd �d d �Zdd �d d �Z d!S )#�_class_resolver)r+ r� �arg�fallback�_dict� _resolvers� favor_tablesr. r+ �RelationshipProperty[Any]r� �Mapping[str, Any]r� r' r� rH r� z Tuple[Callable[[str], Any], ...]r� F)r+ r� r� r� r� c C s6 || _ || _|| _|| _t�| j�| _d| _|| _ d S )NrD ) r+ r� r� r� r Z PopulateDict� _access_clsr� r� r� )r\ r+ r� r� r� r� rD rD rE r_ � s z_class_resolver.__init__r r| c C s | j }t�|�}|j}|d k s"t�|j}|j}| jrd||jkrH|j| S ||j krdt |t |d|��S ||krzt ||| �S | js�||jkr�|j| S ||j kr�t |t |d|��S d|kr�|t t|d �kr�t t|d �}|�|�S | j�r| jD ] }||�} | d k r�| S q�| j| S )Nr� r/ )r+ r Zmanager_of_class�registryr� Z_class_registryr� r� r� Z_schemasr� r� r� r r5 r r� r� ) r\ rI r+ �managerZ decl_baser, r� r� Zresolvr� rD rD rE r� � sB � � z_class_resolver._access_cls� Exceptionr )ru �errr- c C s� t �d|�}|r^|�d��d�}t�d| jj� d| j�d| jj � d|�d�� d |� d � �|�n t�d | jj| j|| j f �|�d S ) Nz (.+)\[(.+)\]r �'zWhen initializing mapper z, expression "relationship(z�)" seems to be using a generic class as the argument to relationship(); please state the generic argument using an annotation, e.g. "z : Mapped[r z['z']] = relationship()"z�When initializing mapper %s, expression %r failed to locate a name (%r). If this is a class name, consider adding this relationship() to the %r class after both dependent classes have been defined.) �re�match�group�stripr r= r� rt r� rI r+ )r\ ru r� Z generic_matchZclsargrD rD rE �_raise_for_name� s 2�� ���z _class_resolver._raise_for_namez Union[Table, Type[Any], _ModNS]rb c C s� | j }| j}d }z0|�d�D ] }|d kr2|| }q t||�}q W n` tk rn } z| �||� W 5 d }~X Y n` tk r� } z| �|jd |� W 5 d }~X Y n.X t|t �r�|j S t r�t|t t tf�s�t�|S d S )Nr0 r )r� r� r8 r� r6 r� rl �argsr3 r� r+ r �typer" rw r� )r\ ru �d�rvalrA r� �nrD rD rE � _resolve_name� s" $ z _class_resolver._resolve_namec C sh z.t | jt� | j�}t|t�r&|jW S |W S W n4 tk rb } z| �|j d |� W 5 d }~X Y nX d S )Nr ) �evalr� �globalsr� r3 r� r+ rl r� r� )r\ �xr� rD rD rE �__call__ s z_class_resolver.__call__N)F) rO r7 rP rR rs r_ r� r� r� r� rD rD rD rE r� � s �'r� r� �_fallback_dictr� zmTuple[Callable[[str], Callable[[], Union[Type[Any], Table, _ModNS]]], Callable[[str, bool], _class_resolver]])r+ r� r- c sx t d kr@dd l}ddlm} ddlm} t�|j��||d��a dddd d �� �fd d � }dd d�� �fdd� }||fS )Nr r )�foreign)�remote)r� r� Fr' rH r� )r� r� r- c s t � �t| |d�S )N)r� )r� r� )r� r� �r+ r� rD rE � resolve_arg0 s �z _resolver.<locals>.resolve_argz-Callable[[], Union[Type[Any], Table, _ModNS]])r� r- c s t � �t| �jS r` )r� r� r� )r� r� rD rE � resolve_name5 s z _resolver.<locals>.resolve_name)F) r� � sqlalchemy� r� r� r Z immutabledict�__dict__�union)r+ r� r� r� r� r� r� rD r� rE � _resolver! s �r� )?rQ � __future__r r� �typingr r r r r r r r r r r r r r r r r rV r� r r Zdescriptor_propsr � propertiesr r r r r Z sql.schemar Z util.typingr Z relationshipsr r! r" r# r' r� r( �setr& rs r; rG rN r% r4 r5 rw r� Z _inspectsr� r� r� r� r� rD rD rD rE �<module> sl 6! EG �