Format: application/x-python-code
1 o
2
3 �hfC � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
4 Z
5 ddl m Z m
6 Z
7 ddlmZmZmZ ddlZe
8 rBddlmZ ed d
9 �Zg d �Zd d
10 gZdZdejjejjf Z G dd� de �Z G dd� d�Z dd� Z G dd� de �Z!G dd� d�Z"G dd� de"e!�Z#G d d � d e"e �Z$dEd d �Z%d!d"� Z&d#d$� Z'd%d&� Z(d'd(� Z)e
11 d)d*�d+d,� �Z*dFd-d-d.�d/d0�Z+dd1�d2d3�Z,dd1�d4d5�Z-d-dd6�d7d8�Z.dGd9d:�Z/d;d<� Z0d=d>� Z1d?d@� Z2dAdB� Z3dCdD� Z4dS )Hz�
12 Environments are a way to activate different Python versions or Virtualenvs for
13 static analysis. The Python binary in that environment is going to be executed.
14 � N)�
15 namedtuple)�which)�
16 TYPE_CHECKING)�memoize_method�
17 time_cache)�CompiledSubprocess�InferenceStateSameProcess�InferenceStateSubprocess)�InferenceState� VersionInfozmajor minor micro)z3.13z3.12z3.11z3.10z3.9z3.8z3.7z3.6z/usr/binz/usr/local/binZ CONDA_PREFIX�%s.%sc @ s e Zd ZdZdS )�InvalidPythonEnvironmentz�
18 If you see this exception, the Python executable or Virtualenv you have
19 been trying to use is probably not a correct Python version.
20 N)�__name__�
21 __module__� __qualname__�__doc__� r r �Jc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/api/environment.pyr
22 s r
23 c @ s$ e Zd Zedd� �Zedd� �ZdS )�_BaseEnvironmentc C s d| j j| j jf }tj|d�S )Nr ��version)� version_info�major�minor�parsoZ load_grammar)�self�version_stringr r r � get_grammar' s z _BaseEnvironment.get_grammarc C s. z| j W S ty t| j�| _ | j Y S w �N)�_hash�AttributeError�_calculate_sha256_for_file�
24 executable�r r r r �_sha256, s
25 �z_BaseEnvironment._sha256N)r r r r r �propertyr$ r r r r r & s
26
27 r c C s t jt jt jd d� fS �N� )�sysr" �prefixr r r r r � _get_info5 s �r* c @ sL e Zd ZdZdZddd�Zdd� Zdd� Zd d
28 d efd d
29 �Z e
30 dd� �Z dS )� Environmentz�
31 This class is supposed to be created by internal Jedi architecture. You
32 should not create it directly. Please use create_environment or the other
33 functions instead. It is then returned by that function.
34 Nc C s || _ || _| �� d S r )�_start_executable� _env_vars�_get_subprocess)r r" �env_varsr r r �__init__E s zEnvironment.__init__c
35 C s� | j d ur | j js | j S zt| j| jd�| _ | j �d t�}W n ty3 } z td| j|f ��d }~ww |d | _ |d | _
36 t |d � | _ | j S )N�r/ z,Could not get version information for %r: %rr � � )
37 � _subprocessZ
38 is_crashedr r, r- Z_sendr* � Exceptionr
39 r" �path� _VersionInfor )r �info�excr r r r. K s. �����
40
41 zEnvironment._get_subprocessc C s* d� dd� | jD ��}d| jj|| jf S )N�.c s s � | ]}t |�V qd S r )�str)�.0�ir r r � <genexpr>k s � z'Environment.__repr__.<locals>.<genexpr>z<%s: %s in %s>)�joinr � __class__r r6 )r r r r r �__repr__j s zEnvironment.__repr__�inference_stater
42 �returnc C s t || �� �S r )r r. �r rB r r r � get_inference_state_subprocessn s z*Environment.get_inference_state_subprocessc C s | � � �� S )z�
43 The sys path for this environment. Does not include potential
44 modifications from e.g. appending to :data:`sys.path`.
45
46 :returns: list of str
47 )r. � get_sys_pathr# r r r rF t s
48 zEnvironment.get_sys_pathr ) r r r r r4 r0 r. rA r rE r rF r r r r r+ = s
49 �
50 �r+ c @ s e Zd Zdd� ZdS )�_SameEnvironmentMixinc C s4 t j | _| _t j| _tt jd d� � | _d | _d S r& )r( r" r, r) r6 r7 r r- r# r r r r0 � s
51 z _SameEnvironmentMixin.__init__N)r r r r0 r r r r rG � s rG c @ s e Zd ZdS )�SameEnvironmentN)r r r r r r r rH � s rH c @ s& e Zd Zdddefdd�Zdd� ZdS ) �InterpreterEnvironmentrB r
52 rC c C s t |�S r )r rD r r r rE � s z5InterpreterEnvironment.get_inference_state_subprocessc C s t jS r )r( r6 r# r r r rF � s z#InterpreterEnvironment.get_sys_pathN)r r r r rE rF r r r r rI � s �
53 � rI � VIRTUAL_ENVc C sX t j�| �}|r*t j�|�t j�tj�krt� S zt|dd�W S t y) Y dS w dS )z�Get virtualenv environment from VIRTUAL_ENV environment variable.
54
55 It uses `safe=False` with ``create_environment``, because the environment
56 variable is considered to be safe / controlled by the user solely.
57 F��safeN)
58 �os�environ�getr6 �realpathr( r) �_try_get_same_env�create_environmentr
59 )�env_var�varr r r �_get_virtual_env_from_var� s ��rU c sb t �� }t| d�� � t� fdd�d�D ]}|�|� qW d � |�� S 1 s(w Y |�� S )N�rbc s � � tj�S r )�read�filecmp�BUFSIZEr ��fr r �<lambda>� s z,_calculate_sha256_for_file.<locals>.<lambda>� )�hashlib�sha256�open�iter�update� hexdigest)r6 r_ �blockr rZ r r! � s �
60 ��r! c C s, t � } | dur | S t t�}|dur|S t� S )a|
61 Tries to return an active Virtualenv or conda environment.
62 If there is no VIRTUAL_ENV variable or no CONDA_PREFIX variable set
63 set it will return the latest Python version installed on the system. This
64 makes it possible to use as many new Python features as possible when using
65 autocompletion and other functionality.
66
67 :returns: :class:`.Environment`
68 N)rU �
69 _CONDA_VARrQ )� virtual_env� conda_envr r r �get_default_environment� s
70 rh c C s� t � } tj�| j��� �d�sGtjdkrd}ndtj d tj
71 d f dtj d df}|D ]}tj� tj |�}tj�
72 |�rCt|� S q-t� S | S ) N�python�nt)zScripts\python.exe�
73 python.exezbin/python%s.%sr r2 z bin/python%sz
74 bin/python)rH rM r6 �basenamer" �lower�
75 startswith�namer( r r r? � exec_prefix�isfiler+ rI )�envZchecks�check�guessr r r rQ � s
76 � �rQ c C sN t j�d�p t j�t�} t� }| r%t j�| �t j�|j�kr%t�� t� S |S )NrJ )rM rN rO re � _get_cached_default_environmentr6 rP � clear_cache)rT � environmentr r r � get_cached_default_environment� s rx iX )�secondsc C s" zt � W S ty t� Y S w r )rh r
77 rI r r r r ru s
78
79 �ru T)rL �use_environment_varsc
80 c s� � | du rg } t � }|r-t� }|dur |V |�|j� tt�}|dur-|V |�|j� | D ]>}tj�|�s8q/tj�|�}t�|�D ])}tj� ||�}||v rQqC|�|� z
81 t
82 ||d�}t |�V W qC t yl Y qCw q/dS )a
83
84 :param paths: A list of paths in your file system to be scanned for
85 Virtualenvs. It will search in these paths and potentially execute the
86 Python binaries.
87 :param safe: Default True. In case this is False, it will allow this
88 function to execute potential `python` environments. An attacker might
89 be able to drop an executable in a path this function is searching by
90 default. If the executable has not been installed by root, it will not
91 be executed.
92 :param use_environment_vars: Default True. If True, the VIRTUAL_ENV
93 variable will be checked if it contains a valid VirtualEnv.
94 CONDA_PREFIX will be checked to see if it contains a valid conda
95 environment.
96
97 :yields: :class:`.Environment`
98 NrK )
99 �setrU �addr6 re rM �isdir�abspath�listdirr? �_get_executable_pathr+ r
100 ) �pathsrL rz Z _used_pathsrf rg � directoryr6 r" r r r �find_virtualenvs s< �
101 ���r� r1 c c s4 � t D ]}z t|| d�V W q ty Y qw dS )a&
102 Ignores virtualenvs and returns the Python versions that were installed on
103 your system. This might return nothing, if you're running Python e.g. from
104 a portable version.
105
106 The environments are sorted from latest to oldest Python version.
107
108 :yields: :class:`.Environment`
109 r1 N)�_SUPPORTED_PYTHONS�get_system_environmentr
110 )r/ r r r r �find_system_environmentsE s �
111 ��r� c C sp t d| �}|r|tjkrt� S t|�S tjdkr2t| �D ]}z t||d�W S ty1 Y q w td| ��)z�
112 Return the first Python environment found for a string of the form 'X.Y'
113 where X and Y are the major and minor versions of Python.
114
115 :raises: :exc:`.InvalidPythonEnvironment`
116 :returns: :class:`.Environment`
117 ri rj r1 z Cannot find executable python%s.) r r( r" rH r+ rM ro �&_get_executables_from_windows_registryr
118 )r r/ �exer r r r� X s
119
120 � r� )rL r/ c C s6 t j�| �rt| |� t| |d�S tt| |d�|d�S )z�
121 Make it possible to manually create an Environment object by specifying a
122 Virtualenv path or an executable path and optional environment variables.
123
124 :raises: :exc:`.InvalidPythonEnvironment`
125 :returns: :class:`.Environment`
126 r1 rK )rM r6 rq � _assert_safer+ r� )r6 rL r/ r r r rR o s
127 rR c C sp t jdkrt j�| dd�t j�| d�g}n t j�| dd�g}|D ]
128 }t j�|�r* nq td| ��t||� |S )z:
129 Returns None if it's not actually a virtual env.
130 rj �Scriptsrk �binri z%s seems to be missing.)rM ro r6 r? �existsr
131 r� )r6 rL Zpythonsri r r r r� } s
132 �
133 r� c
134 c s� � dd l }g d�}|j|jfD ]F}|D ]A}|j| d�}z0|�||�� }|�|d�d }tj�|d�}tj� |�r;|V W d � n1 sEw Y W q t
135 yT Y qw qd S )Nr )z0SOFTWARE\Python\PythonCore\{version}\InstallPathz<SOFTWARE\Wow6432Node\Python\PythonCore\{version}\InstallPathz3SOFTWARE\Python\PythonCore\{version}-32\InstallPathz?SOFTWARE\Wow6432Node\Python\PythonCore\{version}-32\InstallPathr � rk ) �winreg�HKEY_CURRENT_USER�HKEY_LOCAL_MACHINE�format�OpenKey� QueryValueExrM r6 r? rq � WindowsError)r r� Zsub_keysZroot_key�sub_key�keyr) r� r r r r� � s( � � �� ���r� c C s |r
136 t | �s td��d S d S )Nz(The python binary is potentially unsafe.)�_is_safer
137 )�executable_pathrL r r r r� � s
138 ��r� c C sL t j�| �}t|�r dS t� D ]}|j|kr dS |jt|�kr# dS qdS )NTF)rM r6 rP �_is_unix_safe_simpler� r" r$ r! )r� � real_pathrw r r r r� � s
139
140
141 �r� c s0 t � rt� fdd�tD ��S t�� �j}|dkS )Nc 3 s � | ]}� � |�V qd S r )rn )r< �p�r� r r r> � s � z'_is_unix_safe_simple.<locals>.<genexpr>r )�_is_unix_admin�any� _SAFE_PATHSrM �stat�st_uid)r� �uidr r� r r� � s
142 r� c C s$ zt �� dkW S ty Y dS w )Nr F)rM �getuidr r r r r r� � s
143 �r� )rJ r )T)5r rM r( r^ rX � collectionsr �shutilr �typingr Z
144 jedi.cacher r Z"jedi.inference.compiled.subprocessr r r r Zjedi.inferencer
145 r7 r� r� re r r r Z_CURRENT_VERSIONr5 r
146 r r* r+ rG rH rI rU r! rh rQ rx ru r� r� r� rR r� r� r� r� r� r� r r r r �<module> sT
147 G
148 (
149 6
150