o
    Zh7                     @   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|v sd|v 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   M/var/www/html/lang_env/lib/python3.10/site-packages/jedi/inference/helpers.pyis_stdlib_path   s
   r   c                 C   sZ   t  | }g }| jD ]}t|tjrt  |}||_nt|}||_|| q
||_|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   sL  |j }|jdkrddlm} || jS |jdks%||jd |jd fvr@|dkr1ddlm} |S |jdkr;| 	|S | 	|S |j }|j
|}|rN|}n|d	 }|jd
krt|}		 |	d	8 }	|j|	 }
|
jdkrinqZ|j|	d	 | }n|jd }
|jd	| }|
dkr|d }
|d	d }| 	|
}ddlm} |D ]}|| ||}q|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_nodeT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-   sF   
"







r1   c                 C   sN   z| j }W n ty   | jdkr| g Y S g  Y S w ttdd |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   r4   o   s   


r4   c                 C   s   |   ot| jd dtS N)default)Zis_compiledr   get_safe_valuer   valuer   r   r   	is_string{   s   rA   c                 C   s   t | pt| S r3   )	is_numberrA   r?   r   r   r   
is_literal   s   rC   c                 C   s   | j d d} t| |r| S d S r<   )r>   r   )r@   acceptr   r   r   _get_safe_value_or_none   s   
rE   c                 C   
   t | tS r3   )rE   intr?   r   r   r   get_int_or_none      
rH   c                 C   rF   r3   )rE   r   r?   r   r   r   get_str_or_none   rI   rJ   c                 C   s   t | ttfd uS r3   )rE   rG   floatr?   r   r   r   rB      s   rB   c                   @   s   e Zd ZdS )SimpleGetItemNotFoundN)__name__
__module____qualname__r   r   r   r   rL      s    rL   c               
   g   s0    zd V  W d S  | y } zt |d }~ww r3   )rL   )Zexception_classeser   r   r   reraise_getitem_errors   s   rQ   c                 C   s   d}g }| dd  D ]R}|dv r|s|t |j7 }q
|jdkr;|jd d d D ]}|| ||u r4 nq'q
 ||fS |jdkrO|| ||u rN ||fS q
|dkrX|sWg }q
 ||fS ||fS )Nr   r$   ).z...Zdotted_name   r2   ,)lenr@   r'   r   r   )nodesZis_import_fromZ
until_nodelevelnamesr;   nr   r   r   parse_dotted_names   s8   



rZ   c                 G   s   |  |d d |d S )Nr!   )import_moduleZpy__getattribute__)r(   rX   r   r   r   values_from_qualified_names   s   r\   c                 C   s"   |   j}|d u rdS |d dv S )NFr   )ZpandasnumpyZ
tensorflowZ
matplotlib)Zget_root_contextstring_names)r*   r^   r   r   r   is_big_annoying_library   s   
r_   )Fr3   )r   r   r   r   	itertoolsr   
contextlibr   Zparso.pythonr   r   r   r1   r4   rA   rC   rE   rH   rJ   rB   	ExceptionrL   rQ   rZ   r\   r_   r   r   r   r   <module>   s.    
B

