Format: application/x-python-code
1 o
2
3 �hc( � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
4 Z
5 m Z d dl m
6 Z
7 d dlmZmZ d d lmZ g d
8 �Zd d � Zd
9 d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdeee fdd�Z e� d#d d ��Z d d � Z G d!d"� d"e
10 �Z dS )$� N)�List)�Path)�search_ancestor)� inference_state_method_cache)� goto_import�load_module_from_path)�ParserTreeFilter)� NO_VALUES�ValueSet)�infer_call_of_leaf))�_pytestZ monkeypatch)r �capture)r �logging)r Ztmpdir)r Zpytesterc � � fdd�}|S )Nc s( | � � dkr| j� � dkrtS � | |�S )N�fixturez_pytest.fixtures)Z
11 py__name__�parent_contextr )�value� arguments��callback� �Ic:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/plugins/pytest.py�wrapper s
12 zexecute.<locals>.wrapperr )r r r r r �execute s r c s dd� �� �fdd�}|S )Nc S s^ | j jd ur | �� }tdd� |D ��r t�dd� |D ��S |S | �� }|�� r+|�� S |� � S )Nc s s" � | ] }|j jd d�dkV qdS )T�Zinclude_module_names)�typing� GeneratorN)�name�get_qualified_names��.0�vr r r � <genexpr>% s � � �z=infer_anonymous_param.<locals>.get_returns.<locals>.<genexpr>c s s � | ]
13 }|� d ��� V qdS )�__next__N)Zpy__getattribute__Zexecute_annotationr r r r r" ( s
14 � �
15 �)
16 � tree_node�
17 annotationZexecute_with_values�anyr
18 � from_sets�
19 as_contextZ is_generatorZmerge_yield_valuesZget_return_values)r �resultZfunction_contextr r r � get_returns" s �
20 �z*infer_anonymous_param.<locals>.get_returnsc sZ | j r� | �S t| �\}}|r)| �� }t|| j|d�}|r)t��fdd�|D ��S � | �S )N��skip_own_modulec 3 s& � | ]}|� � D ]}� |�V qqd S �N)�infer)r r r )r* r r r" G s � �� �z9infer_anonymous_param.<locals>.wrapper.<locals>.<genexpr>)Zannotation_node� _is_a_pytest_param_and_inherited�get_root_context�_goto_pytest_fixture� string_namer
21 r' )�
22 param_name�is_pytest_param�param_name_is_function_name�module�fixtures��funcr* r r r 7 s ���z&infer_anonymous_param.<locals>.wrapperr �r9 r r r8 r �infer_anonymous_param! s r; c r )Nc s4 t | �\}}|rt| �� | j|d�}|r|S � | �S �Nr+ )r/ r1 r0 r2 )r3 r4 r5 �names�r9 r r r Q s ��z%goto_anonymous_param.<locals>.wrapperr r: r r> r �goto_anonymous_paramP s r? c r )Nc sH | � � }t||�r g }t|�D ]
23 }|t|��� 7 }q|r |S � | ||�S r- )r0 �_is_pytest_func�_iter_pytest_modules�
24 FixtureFilter�values)�context� func_name�decorator_nodes�module_contextr= r> r r r a s
25 z%complete_param_names.<locals>.wrapperr r: r r> r �complete_param_names` s rH c C s0 t | |d�D ]} t| ��|�}|r| S qd S r< )rA rB �get)rG r r, r= r r r r1 m s ��r1 c C s< t | jd�}|du r dS |�� }t|jj|�|jj| jkfS )z�
26 Pytest params are either in a `test_*` function or have a pytest fixture
27 with the decorator @pytest.fixture.
28
29 This is a heuristic and will work in most cases.
30 �funcdefN)FF)r � tree_nameZget_decoratorsr@ r r r2 )r3 rJ �
31 decoratorsr r r r/ t s �r/ c C s | � d�p
32 tdd� |D ��S )N�testc s s � | ] }d |� � v V qdS )r N)�get_code)r �nr r r r" � � � z"_is_pytest_func.<locals>.<genexpr>)�
33 startswithr& )rE rF r r r r@ � s
34 �r@ �returnc C s� t jdkr7ddlm} t jdkr| dd�}n| � �dd�}t jdkr)d d
35 � |D �S d d
36 � |D �}d d
37 � |D �S dd
38 lm} dd
39 � |dd�D �S )z�
40 Finds pytest plugin modules hooked by setuptools entry points
41
42 See https://docs.pytest.org/en/stable/how-to/writing_plugins.html#setuptools-entry-points
43 )� � r )� entry_points)rS �
44 Zpytest11)�groupr )rS � c S � g | ]}|j �d ��qS ��.)r6 �split�r �epr r r �
45 <listcomp>� � z/_find_pytest_plugin_modules.<locals>.<listcomp>c S s g | ] }|j �|j��qS r )�pattern�matchr r] r r r r_ � s ��c S s g | ] }|r|� d ��d��qS )r6 r[ )rW r\ )r �xr r r r_ � s )�iter_entry_pointsc S rY rZ )� module_namer\ r] r r r r_ � r` )�sys� version_info�importlib.metadatarU rI �
46 pkg_resourcesrd )rU Zpytest_entry_points�matchesrd r r r �_find_pytest_plugin_modules� s
47
48
49 � rk Fc # s2 � |s| V | � � j}|d ur�|�� � | j�� }d }t� fdd�|D ��r�� �d�}t|j�| � � krez(t
50 | j|�}|� � }|V |j �
51 � �d�}|rZ|�|d �}t| |�E d H W n tyd Y nw � �� � |d urt� j|jkrtn
52 � }t� fdd�|D ��s%tt� D ]} | j�| �D ]}
53 |
54 � � V q�q�d S )Nc 3 s � | ] }� j �|�V qd S r- )�pathrQ )r �p��folderr r r" � rP z'_iter_pytest_modules.<locals>.<genexpr>z conftest.pyZpytest_pluginsr )� get_value�file_ioZget_parent_folder�inference_stateZ get_sys_pathr& Z get_file_ior rl Z
55 py__file__r r( r$ Zget_used_namesrI Z create_name�_load_pytest_plugins�FileNotFoundError�_PYTEST_FIXTURE_MODULESrk �
56 import_module) rG r, rq �sys_pathZ last_folder�mZconftest_moduleZ plugins_listr r= � module_valuer rn r rA � sB �
57
58
59 � �� ��rA c c sn � ddl m} |�� D ])}|�� D ]"}|�� D ]}||�}|r2|�d�}| j�|�D ]}|�� V q*qqq d S )Nr )�get_str_or_noner[ )�jedi.inference.helpersrz r. Z
60 py__iter__r\ rr rv r( ) rG r rz �inferredZ seq_valuer Zfq_namer= ry r r r rs � s �
61 ����rs c s$ e Zd Z� fdd�Zdd� Z� ZS )rB c # sb � t � �|�D ]'}|jjdkr$t� j|�}t� fdd�|D ��r#|V q� �� j|�r.|V qd S )NZ import_fromc 3 s& � | ]}|j r� �|j|j �V qd S r- )rK � _is_fixturer )r Ziname��selfr r r" � s � �
62 �z(FixtureFilter._filter.<locals>.<genexpr>)�super�_filter�parent�typer r r& r} )r r= r Zimported_names�� __class__r~ r r� � s � ����zFixtureFilter._filterc C s� |j }|jdkr
63 dS |j }|jdkrdS |jd }|jdkr"|j}n|g}|D ]D}|jd }d|�� v rk|jdkrT|jd }|�� } | d
64 krNt|| d d �}
65 n |�|�}
66 n|�|�}
67 |
68 D ]} | jjd d
69 �dkrj d S q[q'dS )NrJ F� decoratedr rL � r Z atom_expr������)T)Zcut_own_trailerr )r r7 r ) r� r� �childrenrN Z
70 get_last_leafr Z
71 infer_noder r ) r rD r rJ r� rL � decoratorZ dotted_nameZ last_trailerZ last_leafrC r r r r r} � s<
72
73
74
75
76
77
78 �
79 ���zFixtureFilter._is_fixture)�__name__�
80 __module__� __qualname__r� r} �
81 __classcell__r r r� r rB � s rB )F) rf r r �pathlibr Z
82 parso.treer Zjedi.inference.cacher Zjedi.inference.importsr r Zjedi.inference.filtersr Zjedi.inference.base_valuer r
83 r{ r ru r r; r? rH r1 r/ r@ �strrk rA rs rB r r r r �<module> s, /
84 &
85