Edit File: clsregistry.cpython-37.pyc
B ��4]++ � @ s d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd l m Z dd l m Z dd l m Z dd l mZ e� Zd d� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Ze�e�dd� � G dd� de�Zdd � ZG d d � d e�Zd d � Zd!d"� Z dS )#z�Routines to handle the string class registry used by declarative. This system allows specification of classes and expressions used in :func:`.relationship` using strings. � N� )�exc)� inspection)�util)� class_mapper)� interfaces)�ColumnProperty)�RelationshipProperty)�SynonymProperty)�_get_table_keyc C s� | |j kr6|j | }t|t�s@t||g� }|j | < n ||j | < y|j d }W n( tk rv tdd� |j d< }Y nX |j�d�}x@|r�|�d�}|�|�}x|D ]}|�|�}q�W |� | |� q�W dS )z^Add a class to the _decl_class_registry associated with the given declarative class. �_sa_module_registryN�.r ) �_decl_class_registry� isinstance�_MultipleClassMarker�KeyError� _ModuleMarker� __module__�split�pop� get_module� add_class)� classname�cls�existingZ root_module�tokens�token�module� r �Y/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/ext/declarative/clsregistry.pyr ! s r c @ s> e Zd ZdZdZddd�Zdd� Zdd � Zd d � Zd d � Z dS )r zRrefers to multiple classes of the same name within _decl_class_registry. )� on_remove�contents� __weakref__Nc s, |� _ t� fdd�|D ��� _t�� � d S )Nc s g | ]}t �|� j��qS r )�weakref�ref� _remove_item)�.0�item)�selfr r � <listcomp>T s z1_MultipleClassMarker.__init__.<locals>.<listcomp>)r �setr! � _registries�add)r( �classesr r )r( r �__init__Q s z _MultipleClassMarker.__init__c C s dd� | j D �S )Nc s s | ] }|� V qd S )Nr )r& r$ r r r � <genexpr>Y s z0_MultipleClassMarker.__iter__.<locals>.<genexpr>)r! )r( r r r �__iter__X s z _MultipleClassMarker.__iter__c C sV t | j�dkr*t�dd�||g � ��n(t| j�d }|� }|d krNt|��|S d S )N� zxMultiple classes found for path "%s" in the registry of this declarative base. Please use a fully module-qualified path.r r )�lenr! r �InvalidRequestError�join�list� NameError)r( �path�keyr$ r r r r � attempt_get[ s z _MultipleClassMarker.attempt_getc C s. | j �|� | j s*t�| � | jr*| �� d S )N)r! �remover+ �discardr )r( r$ r r r r% j s z!_MultipleClassMarker._remove_itemc C sX t dd� dd� | jD �D ��}|j|kr>t�d|j|jf � | j�t�|| j �� d S )Nc S s g | ]}|d k r|j �qS )N)r )r& r r r r r) w s z1_MultipleClassMarker.add_item.<locals>.<listcomp>c S s g | ] }|� �qS r r )r& r$ r r r r) x s 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.) r* r! r r �warn�__name__r, r# r$ r% )r( r' �modulesr r r �add_itemq s z _MultipleClassMarker.add_item)N) r= r � __qualname__�__doc__� __slots__r. r0 r9 r% r? r r r r r I s r c @ sL e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dS )r z?"refers to a module name within _decl_class_registry. )�parent�namer! �mod_nsr7 r" c C sJ || _ || _i | _t| �| _| j r6| j j| jg | _ng | _t�| � d S )N)rC rD r! �_ModNSrE r7 r+ r, )r( rD rC r r r r. � s z_ModuleMarker.__init__c C s || j kS )N)r! )r( rD r r r � __contains__� s z_ModuleMarker.__contains__c C s | j | S )N)r! )r( rD r r r � __getitem__� s z_ModuleMarker.__getitem__c C s: | j �|d � | j s6| jd k r6| j�| j� t�| � d S )N)r! r rC r% rD r+ r; )r( rD r r r r% � s z_ModuleMarker._remove_itemc C s t | j|�S )N)�getattrrE )r( r8 r r r � resolve_attr� s z_ModuleMarker.resolve_attrc C s. || j kr t|| �}|| j |< n | j | }|S )N)r! r )r( rD �markerr r r r � s z_ModuleMarker.get_modulec sF � �j kr �j � }|�|� n"t|g� �fdd�d� }�j � < d S )Nc s �� � �S )N)r% r )rD r( r r �<lambda>� � z)_ModuleMarker.add_class.<locals>.<lambda>)r )r! r? r )r( rD r r r )rD r( r r � s z_ModuleMarker.add_classN) r= r r@ rA rB r. rG rH r% rJ r r r r r r r � s r c @ s e Zd ZdZdd� Zdd� ZdS )rF )Z__parentc C s || _ d S )N)�_ModNS__parent)r( rC r r r r. � s z_ModNS.__init__c C st y| 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) rN r! r r r rE r �AssertionErrorr9 r7 �AttributeErrorrD )r( r8 �valuer r r � __getattr__� s z_ModNS.__getattr__N)r= r r@ rB r. rR r r r r rF � s rF c @ s e Zd ZdZdd� Zdd� ZdS )� _GetColumns)r c C s || _ d S )N)r )r( r r r r r. � s z_GetColumns.__init__c C s� t | jdd�}|rv||jkr0t�d| j|f ��|j| }|jtjkrv|j}t |t �r^|j }nt |t �svt�d| ��t | j|�S )NF)Z 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 r3 Zextension_typer Z NOT_EXTENSION�propertyr r rD r rI )r( r8 �mp�desc�propr r r rR � s z_GetColumns.__getattr__N)r= r r@ rB r. rR r r r r rS � s rS c C s t �| j�S )N)r �inspectr )�targetr r r rL � rM rL c @ s e Zd ZdZdd� Zdd� ZdS )� _GetTable)r8 �metadatac C s || _ || _d S )N)r8 r[ )r( r8 r[ r r r r. � s z_GetTable.__init__c C s | j jt|| j� S )N)r[ �tablesr r8 )r( r8 r r r rR � s z_GetTable.__getattr__N)r= r r@ rB r. rR r r r r rZ � s rZ c C s t |t�r|�g | �}t|�S )N)r r r9 rS )r8 rQ r r r �_determine_container� s r] c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_class_resolverc C s6 || _ || _| | _| _|| _t�| j�| _d| _ d S )Nr ) r rW �argZ_declarative_arg�fallbackr Z PopulateDict� _access_cls�_dict� _resolvers)r( r rW r` r_ r r r r. s z_class_resolver.__init__c C s� | j }||jkr t||j| �S ||jjkr8|jj| S ||jjkrPt||j�S d|jkr|||jd kr||jd }|�|�S | jr�x"| jD ]}||�}|d k r�|S q�W | j | S )Nr ) r r r] r[ r\ Z_schemasrZ rJ rc r` )r( r8 r �registryZresolvrQ r r r ra s" z_class_resolver._access_clsc C sv y*t | jt� | j�}t|t�r$|jS |S W nF tk rp } z(t� d| j j | j|j d | jf ��W d d }~X Y nX d S )Nz�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.r ) �evalr_ �globalsrb r rS r r6 r r3 rW rC �args)r( �x�nr r r �__call__$ s z_class_resolver.__call__N)r= r r@ r. ra rj r r r r r^ s r^ c sF dd l }ddlm}m} |j�� ���||d�� � ��fdd�}|S )Nr )�foreign�remotec s t � ��| �S )N)r^ )r_ )r r` rW r r � resolve_arg= s z _resolver.<locals>.resolve_arg)� sqlalchemyZsqlalchemy.ormrk rl �__dict__�copy�update)r rW rn rk rl rm r )r r` rW r � _resolver6 s rr c C s� t |t�r�t| |�}x2dD ]*}t||�}t |tj�rt||||�� qW |jr�t |jt�r�|j\}}x4dD ],}||krjt || tj�rj||| �||< qjW |S )N)�argument�order_by� primaryjoin� secondaryjoin� secondaryZ_user_defined_foreign_keys� remote_side)ru rv rw Z foreign_keysrx rt ) r r rr rI r � string_types�setattrZbackref�tuple)r rW rm �attr�vr8 �kwargsr r r �_deferred_relationshipC s r ) rA r# � r r r Zormr r Zorm.propertiesr r r Zschemar r* r+ r �objectr r rF rS Z _inspectsrZ r] r^ rr r r r r r �<module> s. (=4 1