Edit File: path_registry.cpython-37.pyc
B ��4].$ � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ dd lm Z e� e �Z d d � Z d Zd ZG dd� de�ZG dd� de�Ze� e_G dd� de�ZG dd� de�ZG dd� dee�ZdS )z@Path tracking utilities, representing mapper graph traversals. � )�chainN� )� class_mapper� )�exc)� inspection)�utilc C s t �| �S )N)� PathRegistry� deserialize)�path� r �O/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/path_registry.py�_unreduce_path s r �*Z _sa_defaultc @ s� e Zd ZdZdZdZdd� Zdd� Zdd� Zd&d d �Z d d � Z e dd� �Z dd� Z dd� Zdd� Zdd� Zdd� Zedd� �Zed d � �Zed d � �Zd d!� Zd"d#� Zd$d%� Zd S )'r a� Represent query load paths and registry functions. Basically represents structures like: (<User mapper>, "orders", <Order mapper>, "items", <Item mapper>) These structures are generated by things like query options (joinedload(), subqueryload(), etc.) and are used to compose keys stored in the query._attributes dictionary for various options. They are then re-composed at query compile/result row time as the query is formed and as rows are fetched, where they again serve to compose keys to look up options in the context.attributes dictionary, which is copied from query._attributes. The path structure has a limited amount of caching, where each "root" ultimately pulls from a fixed registry associated with the first mapper, that also contains elements for each of its property keys. However paths longer than two elements, which are the exception rather than the rule, are generated on an as-needed basis. Fc C s |d k o| j |j kS )N)r )�self�otherr r r �__eq__< s zPathRegistry.__eq__c C s" t �d|| |� |||| jf< d S )Nz set '%s' on path '%s' to '%s')�log�debug� natural_path)r � attributes�key�valuer r r �set? s zPathRegistry.setc C s& t �d|| |� |�|| jf|� d S )Nz$setdefault '%s' on path '%s' to '%s')r r � setdefaultr )r r r r r r r r C s zPathRegistry.setdefaultNc C s" || j f}||kr|| S |S d S )N)r )r r r r r r r �getG s zPathRegistry.getc C s t | j�S )N)�lenr )r r r r �__len__N s zPathRegistry.__len__c C s t | j�S )N)r r )r r r r �lengthQ s zPathRegistry.lengthc c s: | j }x.tdt|�d�D ]}|| ||d fV qW d S )Nr r r )r �ranger )r r �ir r r �pairsU s zPathRegistry.pairsc sF x@� fdd�t dt� j�d�D �D ]}|jr"|�|�r"dS q"W dS d S )Nc s g | ]}� j | �qS r )r )�.0r )r r r � <listcomp>[ s z0PathRegistry.contains_mapper.<locals>.<listcomp>r r TF)r r r Z is_mapper�isa)r �mapperZ path_mapperr )r r �contains_mapperZ s &z PathRegistry.contains_mapperc C s || j f|kS )N)r )r r r r r r �containsa s zPathRegistry.containsc C s t | �� ffS )N)r � serialize)r r r r � __reduce__d s zPathRegistry.__reduce__c sX | j � ttdd� � fdd�tdt� �d�D �D �� fdd�tdt� �d�D �d g ��S )Nc S s g | ] }|j �qS r )�class_)r" �mr r r r# k s z*PathRegistry.serialize.<locals>.<listcomp>c s g | ] }� | �qS r r )r" r )r r r r# k s r r c s g | ]}� | j �qS r )r )r" r )r r r r# l s r )r �list�zipr r )r r )r r r( g s &zPathRegistry.serializec C sH |d kr d S t tdd� |D �� �}|r>|d d kr>|dd� }| �|�S )Nc S s2 g | ]*\}}t |�|d k r(t |�j| nd f�qS )N)r �attrs)r" �mclsr r r r r# x s z,PathRegistry.deserialize.<locals>.<listcomp>���r )�tupler �coerce)�clsr �pr r r r p s zPathRegistry.deserializec C s t | j|�S )N)�EntityRegistry�root)r3 r% r r r � per_mapper� s zPathRegistry.per_mapperc C s t �dd� || j�S )Nc S s | | S )Nr )�prev�nextr r r �<lambda>� � z%PathRegistry.coerce.<locals>.<lambda>)r �reducer6 )r3 �rawr r r r2 � s zPathRegistry.coercec C sD |� dt �rt| |�S |� dt �r2t| j|�S t�d| ��d S )N�:zinvalid token: %s)�endswith�_WILDCARD_TOKEN� TokenRegistry�_DEFAULT_TOKENr6 r � ArgumentError)r �tokenr r r rD � s zPathRegistry.tokenc C s t �dd� |j| �S )Nc S s | | S )Nr )r8 r9 r r r r: � r; z&PathRegistry.__add__.<locals>.<lambda>)r r< r )r r r r r �__add__� s zPathRegistry.__add__c C s d| j j| jf S )Nz%s(%r))� __class__�__name__r )r r r r �__repr__� s zPathRegistry.__repr__)N)rG � __module__� __qualname__�__doc__�is_token�is_rootr r r r r �propertyr r! r& r' r) r( � classmethodr r7 r2 rD rE rH r r r r r s( r c @ s, e Zd ZdZd ZZdZdZdZdd� Z dS )� RootRegistryzXRoot registry, defers to mappers so that paths are maintained per-root-mapper. r FTc C s |j S )N)Z_path_registry)r �entityr r r � __getitem__� s zRootRegistry.__getitem__N) rG rI rJ rK r r � has_entity�is_aliased_classrM rR r r r r rP � s rP c @ s, e Zd Zdd� ZdZdZdd� Zdd� Zd S ) rA c C s, || _ || _|j|f | _|j|f | _d S )N)rD �parentr r )r rU rD r r r �__init__� s zTokenRegistry.__init__FTc c s� | j js>| j js>xv| j j�� D ]}t| j j | | j�V q W nJ| j jr�| j jjr�| V x0| j jj D ]}t| j j | | j�V qbW n| V d S )N) rU rT rM r% Ziterate_to_rootrA rD rQ Z_is_with_polymorphicZ_with_polymorphic_entities)r �entr r r �generate_for_superclasses� s z'TokenRegistry.generate_for_superclassesc C s t � �d S )N)�NotImplementedError)r rQ r r r rR � s zTokenRegistry.__getitem__N)rG rI rJ rV rS rL rX rR r r r r rA � s rA c @ sX e Zd Zdd� Zdd� Zejdd� �Zejdd� �Ze d d � �Z e d d � �Z d d� Z dS )� PropRegistryc C s� t �|d �}|jr|jr(|j|j }n>|jrf|jrf|j|jk rf|j|jkrf|d �|j�}|j| }|| _|| _|j |f | _ |j |f | _ d| jj | jj f| _ | jj | _ d| j f| _d S )Nr0 �loader)r �inspectrT Z_use_mapper_pathrU Zwith_polymorphic_mappersr% Z_entity_for_mapper�propr r Z_wildcard_tokenZ_wildcard_path_loader_keyZ_default_path_loader_keyZ _loader_key)r rU r] ZinspZsubclass_entityr r r rV � s zPropRegistry.__init__c C s d� dd� | jD ��S )Nz -> c s s | ]}t |�V qd S )N)�str)r" �elemr r r � <genexpr>� s z'PropRegistry.__str__.<locals>.<genexpr>)�joinr )r r r r �__str__� s zPropRegistry.__str__c C s t | jd�S )Nr% )�hasattrr] )r r r r rS � s zPropRegistry.has_entityc C s | j jS )N)r] r% )r r r r rQ � s zPropRegistry.entityc C s | j S )N)rQ )r r r r r% � s zPropRegistry.mapperc C s | | j S )N)rQ )r r r r � entity_path� s zPropRegistry.entity_pathc C s& t |ttf�r| j| S t| |�S d S )N)� isinstance�int�slicer r5 )r rQ r r r rR � s zPropRegistry.__getitem__N) rG rI rJ rV rb r Zmemoized_propertyrS rQ rN r% rd rR r r r r rZ � s rZ c @ sD e Zd ZdZdZdd� Zedd� �Zdd� ZeZ d d � Z d d � Z d S )r5 FTc C s� || _ || _|j| _|| _|j|f | _|jrr| jrr|j�|jd j�rZ|j|jf | _qz|j|jd jf | _n| j| _| | _d S )Nr0 ) r rU rT rQ r r% r$ r rd )r rU rQ r r r rV s zEntityRegistry.__init__c C s t �| j�jS )N)r r\ rQ r% )r r r r r% s zEntityRegistry.mapperc C s dS )NTr )r r r r �__bool__# s zEntityRegistry.__bool__c C s( t |ttf�r| j| S t�| |�S d S )N)re rf rg r �dictrR )r rQ r r r rR ( s zEntityRegistry.__getitem__c C s t | |� | |< }|S )N)rZ )r r �itemr r r � __missing__. s zEntityRegistry.__missing__N) rG rI rJ rT rS rV rN r% rh � __nonzero__rR rk r r r r r5 s r5 )rK � itertoolsr �logging�baser � r r r � getLoggerrG r r r@ rB �objectr rP r6 rA rZ ri r5 r r r r �<module> s ~ 5