U nu f!@s*ddlZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZddlmZddlmZdd lmZmZd d Zd d ZddZddZd/ddZddZddZd0ddZddZddZddZd d!Z d1d#d$Z!d2d%d&Z"d'd(Z#d)d*Z$d+d,Z%Gd-d.d.Z&dS)3N)Path)ZipFile) zipimporterZipImportError) all_suffixes)access)debug) parser_utils)KnownContentFileIO ZipFileIOcCstjSN)syspathrrP/tmp/pip-unpacked-wheel-ol4ehs9f/jedi/inference/compiled/subprocess/functions.py get_sys_pathsrcKstj|f|Sr )r load_module)inference_statekwargsrrrrsrcOs|j|}t|j|||Sr )Zcompiled_subprocessZget_access_handlegetattrr)rid attributeargsrhandlerrrget_compiled_method_returns rcCs t||Sr )rZcreate_access_path)robjrrrcreate_simple_objectsrcKsd|dk r|tjt_}z8ztfd|i|WW Stk rJYW dSXW5|dk r^|t_XdS)zK Returns Tuple[Union[NamespaceInfo, FileIO, None], Optional[bool]] N full_name)NN)r r _find_module ImportError)rsys_pathrrtemprrrget_module_info!sr"cCstjSr )r builtin_module_names)rrrrget_builtin_module_names0sr$cCs|dS)zE Raise an error to simulate certain problems for unit tests. Nr)rZexception_typerrr_test_raise_error4sr%cCs>|dk r t|tjdtj|dk r:t|tjdS)zL Force some prints in the subprocesses. This exists for unit tests. N)file)printr stderrflushstdout)rr(r*rrr _test_print;s  r+cCs6tD]*}tj|d|}tj|r|SqdS)z^ The __init__ file can be searched in a directory. If found return it, else None. __init__N)rosrjoinexists)Zdirectory_pathsuffixrrrr_get_init_pathGs    r1cCs t|Sr )r safe_literal_eval)rvaluerrrr2Ssr2cOstt||Sr )list_iter_module_names)rrrrriter_module_namesWsr6c cs|D]}zddt|D}WnVtk rvzt|}t|j|j}Wn&tk rpt d|YYqYnXYnX|D]`\}}|r|dkr| r|Vq|| dr|dd}n t |}|r|d|kr||dkr||Vq|qdS) Ncss|]}|j|fVqdSr )nameis_dir).0entryrrr _sz%_iter_module_names..z"Not possible to list directory: %s __pycache__z.pyi.r,)r-scandirOSErrorr_zip_list_subdirectoryarchiveprefixrrwarning isidentifierendswithinspect getmodulename)rpathsrZ dir_entriesZzip_import_infor7r8modnamerrrr5[s.     r5Tc Csd}d}tjD]}|r(|tjjkr(d}n|}z |j}Wntk rNYqYnX|||}|dk r|jdkrnq|j}|dkr|j s|s|n|}t ||j j } | dfSqqt |||S)a Provides information about a module. This function isolates the differences in importing libraries introduced with python 3.3 on; it gets a module name and optionally a path. It will return a tuple containin an open file for the module (if not builtin), the filename or the name of the module if it is a builtin one and a boolean indicating if the module is contained in a package. NfrozenT)r meta_path importlib machinery PathFinder find_specAttributeErrororiginloader has_locationImplicitNSInfosubmodule_search_locations_path_find_module_py33) stringrris_global_searchspecrSfinderprPZimplicit_ns_inforrrr~s*        rc Cs|s"tjj||}|dk r"|j}|dkr|dkrztj|}|dk rN|j}Wn2tk r}ztdt|W5d}~XYnX|dkrtd |t ||S)Nz Originally zCouldn't find a loader for {}) rMrNrOrPrSutil ValueErrorrreprformat _from_loader)rYrrSrrZr[errrrXs  "rXcCsz |j}Wntk r"d}Yn X||}z |j}Wntk rRd|fYSX||}zt|j}Wntk rtdYnX|tjjjk r||}n t ||}|dkrd|fSt |t rt ||t |j|fSt|||fS)NFz$get_source was not defined on loader) is_packagerQ get_filenametype get_sourcerrMrNSourceFileLoader _get_source isinstancerr rrBr )rSrYZis_package_methodrdre module_pathfcoderrrrbs,      rbcCs<||}z ||WStk r6td|dYnXdS)z This method is here as a replacement for SourceLoader.get_source. That method returns unicode, but we prefer bytes. z'source not available through get_data())r7N)reget_datar@r)rSfullnamerrrrris  riccsTt|}t|}|}|D]2}t|}|j|kr||}|j|dfVqdS)N/)rrnamelistparent relative_tor7rF)zip_pathZzip_subdir_pathzip_fileZzip_content_file_pathsZ raw_file_name file_pathrrrrAs  rAc@seZdZdZddZdS)rUz;Stores information returned from an implicit namespace speccCs||_||_dSr )r7rI)selfr7rIrrrr,szImplicitNSInfo.__init__N)__name__ __module__ __qualname____doc__r,rrrrrUsrU)NN)NN)NNT)NNNT)'r r-rGrMpathlibrzipfiler zipimportrrZimportlib.machineryrZjedi.inference.compiledrZjedirr Z jedi.file_ior r rrrrr"r$r%r+r1r2r6r5rrXrbrirArUrrrrs:         # * '