B
    .`5                 @   s   d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ dd	lm	Z	 e
eZd
d ZdZdZG dd deZG dd deZe e_G dd deZG dd deZG dd deZG dd deZG dd deeZdS )z@Path tracking utilities, representing mapper graph traversals.

    )chainN   )class_mapper   )exc)
inspection)utilc             C   s
   t | S )N)PathRegistrydeserialize)path r   K/home/dcms/DCMS/lib/python3.7/site-packages/sqlalchemy/orm/path_registry.py_unreduce_path   s    r   *Z_sa_defaultc               @   s   e Zd ZdZdZdZdZdd Zdd Zdd	 Z	d
d Z
d1ddZdd Zedd Zdd Zdd Zdd Zdd Zedd Zedd Zedd  Zed!d" Zd#d$ Zed%d& Zed'd( Zed)d* Zd+d, Zd-d. Zd/d0 ZdS )2r	   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.

    r   Fc             C   s@   y|d k	o| j |j kS  tk
r:   tdt|  dS X d S )Nz1Comparison of PathRegistry to %r is not supportedF)r   AttributeErrorr   warntype)selfotherr   r   r   __eq__>   s    zPathRegistry.__eq__c             C   s@   y|d kp| j |j kS  tk
r:   tdt|  dS X d S )Nz1Comparison of PathRegistry to %r is not supportedT)r   r   r   r   r   )r   r   r   r   r   __ne__H   s    zPathRegistry.__ne__c             C   s"   t d|| | |||| jf< d S )Nzset '%s' on path '%s' to '%s')logdebugnatural_path)r   
attributeskeyvaluer   r   r   setR   s    zPathRegistry.setc             C   s&   t d|| | ||| jf| d S )Nz$setdefault '%s' on path '%s' to '%s')r   r   
setdefaultr   )r   r   r   r   r   r   r   r   V   s    zPathRegistry.setdefaultNc             C   s"   || j f}||kr|| S |S d S )N)r   )r   r   r   r   r   r   r   getZ   s    
zPathRegistry.getc             C   s
   t | jS )N)lenr   )r   r   r   r   __len__a   s    zPathRegistry.__len__c             C   s
   t | jS )N)r    r   )r   r   r   r   lengthd   s    zPathRegistry.lengthc             c   s:   | j }x.tdt|dD ]}|| ||d  fV  qW d S )Nr   r   r   )r   ranger    )r   r   ir   r   r   pairsh   s    zPathRegistry.pairsc                sF   x@ fddt dt jdD D ]}|jr"||r"dS q"W dS d S )Nc                s   g | ]} j | qS r   )r   ).0r$   )r   r   r   
<listcomp>n   s    z0PathRegistry.contains_mapper.<locals>.<listcomp>r   r   TF)r#   r    r   	is_mapperisa)r   mapperZpath_mapperr   )r   r   contains_mapperm   s    &zPathRegistry.contains_mapperc             C   s   || j f|kS )N)r   )r   r   r   r   r   r   containst   s    zPathRegistry.containsc             C   s   t |  ffS )N)r   	serialize)r   r   r   r   
__reduce__w   s    zPathRegistry.__reduce__c                sR   t tdd  fddtdt dD D  fddtdt dD d g S )Nc             S   s   g | ]
}|j qS r   )class_)r&   mr   r   r   r'   ~   s    z0PathRegistry._serialize_path.<locals>.<listcomp>c                s   g | ]} | qS r   r   )r&   r$   )r   r   r   r'   ~   s    r   r   c                s   g | ]} | j qS r   )r   )r&   r$   )r   r   r   r'      s    r   )listzipr#   r    )clsr   r   )r   r   _serialize_pathz   s    &zPathRegistry._serialize_pathc             C   s6   t tdd |D  }|r2|d d kr2|dd }|S )Nc             S   s2   g | ]*\}}t ||d k	r(t |j| nd fqS )N)r   attrs)r&   mclsr   r   r   r   r'      s   z2PathRegistry._deserialize_path.<locals>.<listcomp>r   )tupler   )r3   r   pr   r   r   _deserialize_path   s    zPathRegistry._deserialize_pathc                s$    fddfdd|  D D S )Nc                s&   g | ]\\}}}|  |f|fqS r   )r4   )r&   r   r   r   )r3   r   r   r'      s   z7PathRegistry.serialize_context_dict.<locals>.<listcomp>c                s.   g | ]&\}}t |tr|d   kr||fqS )r   )
isinstancer8   )r&   kv)tokensr   r   r'      s   )items)r3   Zdict_r>   r   )r3   r>   r   serialize_context_dict   s    

z#PathRegistry.serialize_context_dictc                s   t  fdd|D S )Nc             3   s,   | ]$\\}}}|t  |f|fV  qd S )N)r8   r:   )r&   r   r   r   )r3   r   r   	<genexpr>   s   z8PathRegistry.deserialize_context_dict.<locals>.<genexpr>)r   OrderedDict)r3   Z
serializedr   )r3   r   deserialize_context_dict   s    
z%PathRegistry.deserialize_context_dictc             C   s   | j }| |S )N)r   r4   )r   r   r   r   r   r-      s    zPathRegistry.serializec             C   s    |d krd S |  |}| |S )N)r:   coerce)r3   r   r9   r   r   r   r
      s    
zPathRegistry.deserializec             C   s"   |j rt| j|S t| j|S d S )N)r(   CachingEntityRegistryrootSlotsEntityRegistry)r3   r*   r   r   r   
per_mapper   s    zPathRegistry.per_mapperc             C   s   t dd || jS )Nc             S   s   | | S )Nr   )prevnextr   r   r   <lambda>       z%PathRegistry.coerce.<locals>.<lambda>)r   reducerF   )r3   rawr   r   r   rD      s    zPathRegistry.coercec             C   sD   | dt rt| |S | dt r2t| j|S td| d S )N:zinvalid token: %s)endswith_WILDCARD_TOKENTokenRegistry_DEFAULT_TOKENrF   r   ArgumentError)r   tokenr   r   r   rU      s
    
zPathRegistry.tokenc             C   s   t dd |j| S )Nc             S   s   | | S )Nr   )rI   rJ   r   r   r   rK      rL   z&PathRegistry.__add__.<locals>.<lambda>)r   rM   r   )r   r   r   r   r   __add__   s    zPathRegistry.__add__c             C   s   d| j j| jf S )Nz%s(%r))	__class____name__r   )r   r   r   r   __repr__   s    zPathRegistry.__repr__)N)rX   
__module____qualname____doc__	__slots__is_tokenis_rootr   r   r   r   r   r!   propertyr"   r%   r+   r,   r.   classmethodr4   r:   r@   rC   r-   r
   rH   rD   rU   rV   rY   r   r   r   r   r	      s4   


	r	   c               @   s,   e Zd ZdZd ZZdZdZdZdd Z	dS )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    zRootRegistry.__getitem__N)
rX   rZ   r[   r\   r   r   
has_entityis_aliased_classr_   rd   r   r   r   r   rb      s   rb   c               @   s0   e Zd ZdZdd ZdZdZdd Zdd	 Zd
S )rR   )rU   parentr   r   c             C   s,   || _ || _|j|f | _|j|f | _d S )N)rU   rg   r   r   )r   rg   rU   r   r   r   __init__   s    zTokenRegistry.__init__FTc             c   s   | j js>| j js>xv| j j D ]}t| j j | | jV  qW nJ| j jr| j jjr| V  x0| j jj	D ]}t| j j | | jV  qbW n| V  d S )N)
rg   rf   r_   r*   Ziterate_to_rootrR   rU   rc   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   rc   r   r   r   rd      s    zTokenRegistry.__getitem__N)	rX   rZ   r[   r]   rh   re   r^   rj   rd   r   r   r   r   rR      s   rR   c               @   s\   e Zd ZdZdd Zdd Zejdd Zejdd	 Z	e
d
d Ze
dd Zdd ZdS )PropRegistryFc             C   s
  t |d }|}|jr|jr0|j|j  }}n|jr|jr|j|jkr|d |j}|j| }|jr||j|j }d| _q|}n2|jr|jr|j|jk	r|j	|jr|j|j }|| _
|| _|j|f | _|j|f | _d|j| j
j f| _| j
j| _d| jf| _d S )Nr7   Tloader)r   inspectrf   Z_use_mapper_pathrg   Zwith_polymorphic_mappersZ_entity_for_mapperr*   is_unnaturalr)   propr   r   Z_wildcard_tokenZ_wildcard_path_loader_keyZ_default_path_loader_keyZ_loader_key)r   rg   rp   ZinspZnatural_parentZsubclass_entityr   r   r   rh     s6    
 
zPropRegistry.__init__c             C   s   d dd | jD S )Nz -> c             s   s   | ]}t |V  qd S )N)str)r&   elemr   r   r   rA   J  s    z'PropRegistry.__str__.<locals>.<genexpr>)joinr   )r   r   r   r   __str__I  s    zPropRegistry.__str__c             C   s   t | jdS )Nr*   )hasattrrp   )r   r   r   r   re   L  s    zPropRegistry.has_entityc             C   s   | j jS )N)rp   r*   )r   r   r   r   rc   P  s    zPropRegistry.entityc             C   s   | j S )N)rc   )r   r   r   r   r*   T  s    zPropRegistry.mapperc             C   s
   | | j  S )N)rc   )r   r   r   r   entity_pathX  s    zPropRegistry.entity_pathc             C   s&   t |ttfr| j| S t| |S d S )N)r;   intslicer   rG   )r   rc   r   r   r   rd   \  s    
zPropRegistry.__getitem__N)rX   rZ   r[   ro   rh   rt   r   Zmemoized_propertyre   rc   r`   r*   rv   rd   r   r   r   r   rl      s   Hrl   c               @   sH   e Zd ZdZdZdd Zedd Zedd Zd	d
 Z	e	Z
dd ZdS )AbstractEntityRegistryr   Tc             C   s   || _ || _|j| _|| _|j|f | _|jrx| js:|jrx|j|jd jr`|j|jf | _q|j|jd jf | _n| j| _d S )Nr7   )	r   rg   rf   rc   r   ro   r*   r)   r   )r   rg   rc   r   r   r   rh   h  s    zAbstractEntityRegistry.__init__c             C   s   | S )Nr   )r   r   r   r   rv     s    z"AbstractEntityRegistry.entity_pathc             C   s   t | jjS )N)r   rn   rc   r*   )r   r   r   r   r*     s    zAbstractEntityRegistry.mapperc             C   s   dS )NTr   )r   r   r   r   __bool__  s    zAbstractEntityRegistry.__bool__c             C   s&   t |ttfr| j| S t| |S d S )N)r;   rw   rx   r   rl   )r   rc   r   r   r   rd     s    
z"AbstractEntityRegistry.__getitem__N)rX   rZ   r[   r]   re   rh   r`   rv   r*   rz   __nonzero__rd   r   r   r   r   ry   c  s    ry   c               @   s   e Zd ZdZdS )rG   )r   rg   rf   rc   r   r   N)rX   rZ   r[   r]   r   r   r   r   rG     s   rG   c               @   s   e Zd Zdd Zdd ZdS )rE   c             C   s(   t |ttfr| j| S t| |S d S )N)r;   rw   rx   r   dictrd   )r   rc   r   r   r   rd     s    
z!CachingEntityRegistry.__getitem__c             C   s   t | | | |< }|S )N)rl   )r   r   itemr   r   r   __missing__  s    z!CachingEntityRegistry.__missing__N)rX   rZ   r[   rd   r~   r   r   r   r   rE     s   rE   )r\   	itertoolsr   loggingbaser    r   r   r   	getLoggerrX   r   r   rQ   rS   objectr	   rb   rF   rR   rl   ry   rG   r|   rE   r   r   r   r   <module>	   s&   
 /e9