B
    /d7                 @   s   d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm	Z	 dd Z
dd Zd$d
dZdd Zdd Zdd Zdd Zdd Zdd Zdd ZG dd deZedd Zd%ddZd d! Zd"d# ZdS )&    N)chain)contextmanager)treec             C   sJ   | j }d|ksd|krdS tjtjdd}ttt	|d t
| S )Nzdist-packageszsite-packagesFlibpythonz\d.\d)partsospathjoinsysprefixboolrematchescapestr)r	   r   	base_path r   c/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/jedi/inference/helpers.pyis_stdlib_path   s
    r   c             C   s^   t  | }g }xD| jD ]:}t|tjr8t  |}||_nt|}||_|| qW ||_|S )zO
    Much, much faster than copy.deepcopy, but just for parser tree nodes.
    )copychildren
isinstancer   ZLeafparentdeep_ast_copyappend)objZnew_objZnew_childrenchild	new_childr   r   r   r      s    

r   Fc             C   sV  |j }|jdkr(ddlm} || jS |jdksJ||jd |jd fkr|dkrbddlm} |S |jdkrv| 	|S | 	|S |j }|j
|}|r|}n|d	 }|jd
kr|}	x"|	d	8 }	|j|	 }
|
jdkrP qW |j|	d	 | }n|jd }
|jd	| }|
dkr |d }
|d	d }| 	|
}ddlm} x|D ]}|| ||}q<W |S )a  
    Creates a "call" node that consist of all ``trailer`` and ``power``
    objects.  E.g. if you call it with ``append``::

        list([]).append(3) or None

    You would get a node with the content ``list([]).append`` back.

    This generates a copy of the original ast node.

    If you're using the leaf, e.g. the bracket `)` it will return ``list([])``.

    We use this function for two purposes. Given an expression ``bar.foo``,
    we may want to
      - infer the type of ``foo`` to offer completions after foo
      - infer the type of ``bar`` to be able to jump to the definition of foo
    The option ``cut_own_trailer`` must be set to true for the second purpose.
    Zfstringr   )compiledtrailer:)	NO_VALUESZatom   Z
error_nodeawaitN)infer_trailer)r   typeZjedi.inferencer   Zget_string_value_setinference_stater   Zjedi.inference.base_valuer#   Z
infer_nodeindexZjedi.inference.syntax_treer&   )contextleafZcut_own_trailerr    r   r#   powerr)   cutstartbaseZtrailersvaluesr&   r   r   r   infer_call_of_leaf-   sD    
"









r1   c             C   sP   y
| j }W n( tk
r2   | jdkr*| gS g S Y nX ttdd |D S d S )Nnamec             s   s   | ]}t |V  qd S )N)get_names_of_node).0cr   r   r   	<genexpr>x   s    z$get_names_of_node.<locals>.<genexpr>)r   AttributeErrorr'   listr   from_iterable)noder   r   r   r   r3   o   s    


r3   c             C   s   |   ot| jd dtS )N)default)Zis_compiledr   get_safe_valuer   )valuer   r   r   	is_string{   s    r>   c             C   s   t | pt| S )N)	is_numberr>   )r=   r   r   r   
is_literal   s    r@   c             C   s   | j d d} t| |r| S d S )N)r;   )r<   r   )r=   acceptr   r   r   _get_safe_value_or_none   s    
rB   c             C   s
   t | tS )N)rB   int)r=   r   r   r   get_int_or_none   s    rD   c             C   s
   t | tS )N)rB   r   )r=   r   r   r   get_str_or_none   s    rE   c             C   s   t | ttfd k	S )N)rB   rC   float)r=   r   r   r   r?      s    r?   c               @   s   e Zd ZdS )SimpleGetItemNotFoundN)__name__
__module____qualname__r   r   r   r   rG      s   rG   c           
   g   s:   y
d V  W n* | k
r4 } zt |W d d }~X Y nX d S )N)rG   )Zexception_classeser   r   r   reraise_getitem_errors   s    
rL   c             C   s   d}g }x| dd  D ]}|dkr6|s|t |j7 }q|jdkrtx.|jd d d D ]}|| ||krRP qRW qP q|jdkr|| ||krP q|dkr|sg }qP qW ||fS )Nr   r$   ).z...Zdotted_name   r2   ,)lenr=   r'   r   r   )ZnodesZis_import_fromZ
until_nodelevelnamesr:   nr   r   r   parse_dotted_names   s,    



rT   c             G   s   |  |d d |d S )Nr!   )import_moduleZpy__getattribute__)r(   rR   r   r   r   values_from_qualified_names   s    rV   c             C   s"   |   j}|d krdS |d dkS )NFr   )ZpandasnumpyZ
tensorflowZ
matplotlib)Zget_root_contextstring_names)r*   rX   r   r   r   is_big_annoying_library   s    
rY   )F)N)r   r   r   r   	itertoolsr   
contextlibr   Zparso.pythonr   r   r   r1   r3   r>   r@   rB   rD   rE   r?   	ExceptionrG   rL   rT   rV   rY   r   r   r   r   <module>   s*   
B
