3 \CS @sxdZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlZdddd d d d d ddddg ZeddZde_ddZddZd+dd Zd,dd Zed-ddZd.ddZejejjeddZGd d d ZGd!ddZy.ddlZdd"lmZd/d#d$ZejeeWne k r@YnXd%dZ!d0d&dZ"d'dZ#d(d Z$d)dZ%d*d Z&dS)1zUtilities to support packages.) namedtuple)singledispatchN) ModuleType get_importeriter_importers get_loader find_loader walk_packages iter_modulesget_data ImpImporter ImpLoader read_code extend_path ModuleInfozmodule_finder name ispkgz.A namedtuple with minimal info about a module.c CsLy |j}Wn4tk r>|j|}|dkr0dStjj||SX||SdS)z'Return the finder-specific module spec.N) find_specAttributeError find_module importlibutilspec_from_loader)findernamerloaderr/usr/lib64/python3.6/pkgutil.py _get_specs  rcCs6ddl}|jd}|tjjkr"dS|jd|j|S)Nr)marshalreadrr MAGIC_NUMBERload)streamrmagicrrrr(s    c #sifddxt||D]}|V|jryt|jWnNtk r\|dk rX||jYqtk r|dk r~||jnYqXttj|jddpg}fdd|D}t ||jd|EdHqWdS)aYields ModuleInfo for all modules recursively on path, or, if path is None, all accessible modules. 'path' should be either None or a list of paths to look for modules in. 'prefix' is a string to output on the front of every module name on output. Note that this function must import all *packages* (NOT all modules!) on the given path, in order to access the __path__ attribute to find submodules. 'onerror' is a function which gets called with one argument (the name of the package which was being imported) if any exception occurs while trying to import a package. If no onerror function is supplied, ImportErrors are caught and ignored, while all other exceptions are propagated, terminating the search. Examples: # list all modules python can access walk_packages() # list all submodules of ctypes walk_packages(ctypes.__path__, ctypes.__name__+'.') cSs||kr dSd||<dS)NTr)pmrrrseenRszwalk_packages..seenN__path__csg|]}|s|qSrr).0r&)r(rr isz!walk_packages...) r ispkg __import__r ImportError Exceptiongetattrsysmodulesr )pathprefixonerrorinfor)r(rr 5s   ccsf|dkrt}n tt|}i}xB|D]:}x4t||D]&\}}||kr4d||<t|||Vq4Wq$WdS)aYields ModuleInfo for all submodules on path, or, if path is None, all top-level modules on sys.path. 'path' should be either None or a list of paths to look for modules in. 'prefix' is a string to output on the front of every module name on output. N)rmapriter_importer_modulesr)r4r5Z importersyieldedirr-rrrr ns   cCst|dsgS|j|S)Nr )hasattrr )importerr5rrrr:s r:c cs0|jdkstjj|j rdSi}ddl}ytj|j}Wntk rRg}YnX|jx|D]}|j|}|dksb||krqbtjj|j|}d}| otjj|od|kr|}ytj|} Wntk rg} YnXx&| D]}|j|} | dkrd}PqWqb|rbd|krbd||<|||fVqbWdS)Nr__init__Fr,Tr8) r4osisdirinspectlistdirOSErrorsort getmodulenamejoin) r>r5r;rB filenamesfnmodnamer4r- dircontentssubnamerrr_iter_file_finder_moduless<       rMc Cs.tjtjdttjdaWdQRXdS)Nignoreimp)warningscatch_warnings simplefilterDeprecationWarningr import_modulerOrrrr _import_imps  rUc@s.eZdZdZd ddZd ddZd dd ZdS) r aPEP 302 Finder that wraps Python's "classic" import algorithm ImpImporter(dirname) produces a PEP 302 finder that searches that directory. ImpImporter(None) produces a PEP 302 finder that searches the current sys.path, plus any modules that are frozen or built-in. Note that ImpImporter does not currently support being used by placement on sys.meta_path. NcCstjdtt||_dS)Nz5This emulation is deprecated, use 'importlib' instead)rPwarnrSrUr4)selfr4rrrr?szImpImporter.__init__c Cs~|jdd}||kr$|jdkr$dS|jdkr4d}ntjj|jg}ytj||\}}}Wntk rndSXt||||S)Nr,r8)splitr4r@realpathrOrr/r )rWfullnamer4rLfilefilenameetcrrrrs zImpImporter.find_moduler%c cs0|jdkstjj|j rdSi}ddl}ytj|j}Wntk rRg}YnX|jx|D]}|j|}|dksb||krqbtjj|j|}d}| otjj|od|kr|}ytj|} Wntk rg} YnXx&| D]}|j|} | dkrd}PqWqb|rbd|krbd||<|||fVqbWdS)Nrr?Fr,Tr8) r4r@rArBrCrDrErFrG) rWr5r;rBrHrIrJr4r-rKrLrrrr s<       zImpImporter.iter_modules)N)N)r%)__name__ __module__ __qualname____doc__r?rr rrrrr s   c@sneZdZdZdZZddZddZddZd d Z d d Z d dZ dddZ dddZ ddZdddZdS)r zBPEP 302 Loader that wraps Python's "classic" import algorithm NcCs.tjdtt||_||_||_||_dS)Nz5This emulation is deprecated, use 'importlib' instead)rPrVrSrUr\r]r[r^)rWr[r\r]r^rrrr?szImpLoader.__init__c Cs:|jztj||j|j|j}Wd|jr4|jjX|S)N)_reopenrO load_moduler\r]r^close)rWr[modrrrrds  zImpLoader.load_modulec Cst|d }|jSQRXdS)Nrb)openr )rWpathnamer\rrrr "s zImpLoader.get_datacCsT|jrP|jjrP|jd}|tjkr2t|jd|_n|tjtjfkrPt|jd|_dS)Nrrg) r\closedr^rO PY_SOURCErhr] PY_COMPILED C_EXTENSION)rWmod_typerrrrc&s   zImpLoader._reopencCs0|dkr|j}n||jkr,td|j|f|S)Nz,Loader for module %s cannot handle module %s)r[r/)rWr[rrr _fix_name.s  zImpLoader._fix_namecCs|j|}|jdtjkS)Nrj)rqr^rO PKG_DIRECTORY)rWr[rrr is_package6s zImpLoader.is_packagec Cs|j|}|jdkr|jd}|tjkrD|j|}t||jd|_nJ|tjkrv|j zt |j |_Wd|j j Xn|tj kr|jj|_|jS)Nrjexec)rqcoder^rOrm get_sourcecompiler]rnrcrr\rerr _get_delegateget_code)rWr[rpsourcerrrry:s       zImpLoader.get_codec Cs|j|}|jdkr|jd}|tjkrP|jz|jj|_Wd|jjXnd|tj krt j j |j ddrt|j ddd}|j|_WdQRXn|tjkr|jj|_|jS)Nrjr8rkrXrX)rqrzr^rOrmrcr\r rernr@r4existsr]rhrrrxrv)rWr[rpfrrrrvKs      zImpLoader.get_sourcecCst|j}t|d}|jS)Nr?)r r]rr)rWrspecrrrrx]s  zImpLoader._get_delegatecCsH|j|}|jd}|tjkr*|jjS|tjtjtjfkrD|j SdS)Nrj) rqr^rOrrrx get_filenamermrnror])rWr[rprrrr~bs    zImpLoader.get_filename)N)N)N)r_r`rarbrurzr?rdr rcrqrsryrvrxr~rrrrr s    ) zipimporterc csttj|j}|j}t|}i}ddl}x|D]}|j|s@q0||djt j }t|dkr|djdr|d|krd||d<||ddfVt|dkrq0|j |d}|dkrq0|r0d|kr0||kr0d||<||dfVq0WdS) Nrrjr8z __init__.pyTr?r,F) sorted zipimport_zip_directory_cachearchiver5lenrB startswithrYr@seprF) r>r5Zdirlist_prefixZplenr;rBrIrJrrriter_zipimport_modulesps*     rcCsnytj|}WnZtk rhxDtjD]6}y||}tjj||PWq&tk rZYq&Xq&Wd}YnX|S)zRetrieve a finder for the given path item The returned finder is cached in sys.path_importer_cache if it was newly created by a path hook. The cache (or part of it) can be cleared manually if a rescan of sys.path_hooks is necessary. N)r2path_importer_cacheKeyError path_hooks setdefaultr/)Z path_itemr> path_hookrrrrs    ccs|jdrdj|}t|d|krV|jdd}tj|}t|dd}|dkrhdSntjEdHtj }x|D]}t |VqnWdS)aYield finders for the given module name If fullname contains a '.', the finders will be for the package containing fullname, otherwise they will be all registered top level finders (i.e. those on both sys.meta_path and sys.path_hooks). If the named module is in a package, that package is imported as a side effect of invoking this function. If no module name is specified, all top level finders are produced. r,z'Relative module name {!r} not supportedrr)N) rformatr/ rpartitionrrTr1r2 meta_pathr4r)r[msgZpkg_nameZpkgr4itemrrrrs      cCsn|tjkr tj|}|dkr dSt|trb|}t|dd}|dk rF|St|dddkrZdS|j}n|}t|S)zGet a "loader" object for module_or_name Returns None if the module cannot be found or imported. If the named module is not already imported, its containing package (if any) is imported, in order to establish the package __path__. N __loader____spec__)r2r3 isinstancerr1r_r)Zmodule_or_namemodulerr[rrrrs    cCs|jdrdj|}t|ytjj|}WnFttttfk rr}z"d}t|j|t |||WYdd}~XnX|dk r|j SdS)zFind a "loader" object for fullname This is a backwards compatibility wrapper around importlib.util.find_spec that converts most failures to ImportError and only returns the loader rather than the full spec r,z'Relative module name {!r} not supportedz,Error while finding loader for {!r} ({}: {})N) rrr/rrrr TypeError ValueErrortyper)r[rr}Zexrrrrs  *cCst|ts|S|d}|dd}|jd\}}}|rbytj|j}Wqhttfk r^|SXntj}x:|D]0}t|t sqpt |}|dk rg} t |dr|j |} | dk r| j pg} nt |dr|j|\}} x| D]} | |kr|j| qWtjj||} tjj| rpy t| } Wn8tk rT}ztjjd| |fWYdd}~XqpX| Bx:| D]2}|jd}| sb|jdrqb|j|qbWWdQRXqpW|S) aExtend a package's path. Intended use is to place the following code in a package's __init__.py: from pkgutil import extend_path __path__ = extend_path(__path__, __name__) This will add to the package's __path__ all subdirectories of directories on sys.path named after the package. This is useful if one wants to distribute different parts of a single logical package as multiple directories. It also looks for *.pkg files beginning where * matches the name argument. This feature is similar to *.pth files (see site.py), except that it doesn't special-case lines starting with 'import'. A *.pkg file is trusted at face value: apart from checking for duplicates, all entries found in a *.pkg file are added to the path, regardless of whether they are exist the filesystem. (This is a feature.) If the input path is not a list (as is the case for frozen packages) it is returned unchanged. The input path is not modified; an extended copy is returned. Items are only appended to the copy at the end. It is assumed that sys.path is a sequence. Items of sys.path that are not (unicode or 8-bit) strings referring to existing directories are ignored. Unicode items of sys.path that cause errors when used as filenames may cause this function to raise an exception (in line with os.path.isdir() behavior). z.pkgNr,rrzCan't open %s: %s  #)rlistrr2r3r)rrr4strrr=rsubmodule_search_locationsrappendr@rGisfilerhrDstderrwriterstripr)r4rZ sname_pkgZparent_package_Z final_nameZ search_pathdirrportionsr}ZportionZpkgfiler|rlinerrrrsP!            cCstjj|}|dkrdS|j}|dks2t|d r6dStjj|pLtjj |}|dksbt|d rfdS|j d}|j dt j j|jt j j|}|j|S)afGet a resource from a package. This is a wrapper round the PEP 302 loader get_data API. The package argument should be the name of a package, in standard module format (foo.bar). The resource argument should be in the form of a relative filename, using '/' as the path separator. The parent directory name '..' is not allowed, and nor is a rooted name (starting with a '/'). The function returns a binary string, which is the contents of the specified resource. For packages located in the filesystem, which have already been imported, this is the rough equivalent of d = os.path.dirname(sys.modules[package].__file__) data = open(os.path.join(d, resource), 'rb').read() If the package cannot be located or loaded, or it uses a PEP 302 loader which does not support get_data(), then None is returned. Nr __file__/r)rrrrr=r2r3get _bootstrap_loadrYinsertr@r4dirnamerrGr )packageZresourcer}rrfpartsZ resource_namerrrr Rs     )Nr%N)Nr%)r%)r%)r%)r%)'rb collectionsr functoolsrZ simplegenericrimportlib.utilimportlib.machineryr@Zos.pathr2typesrrP__all__rrrr r r:rMregister machinery FileFinderrUr r rrrr/rrrrrr rrrrsR     9   ( Jc   ^