o O%@sdZddlmZddlmZddlmZddlZddlZddlZddlZ ddl Z ddl Z ddl Z ddl Z ddlmZddZd d Zd d Zd dZddZddZddZd"ddZddZddZd"ddZejddZd#d d!ZdS)$z0Utilities for accessing local package resources.)absolute_import)division)unicode_literalsN)filescCs"|d}|dkr|d|SdS)z+Returns package name for given module name..rN)rfind) module_name last_dot_idxr C/tmp/google-cloud-sdk/lib/googlecloudsdk/core/util/pkg_resources.py_GetPackageName"s  r cCstt||S)zCGet a resource as a byte string for given resource in same package.)pkgutilget_datar )r resource_namer r r GetResource*srcCsJtj|r t|Sttj|}t|dr| |St d |)aAGets the given resource as a byte string. This is similar to GetResource(), but uses file paths instead of module names. Args: path: str, filesystem like path to a file/resource. Returns: The contents of the resource as a byte string. Raises: IOError: if resource is not found under given path. rzFile not found {0}) ospathisfilerReadBinaryFileContentsr get_importerdirnamehasattrrIOErrorformat)rimporterr r r GetResourceFromFile/s    rcCstj|r6tjtj|ds|tjvStj||}tj|r.tjtj|dStj|dS|d}t tjj|g|ddR}|sPdSt |d}|rZ|j n|j }||dduS)zChecks if given name can be imported at given path. Args: name: str, module name without '.' or suffixes. path: str, filesystem path to location of the module. Returns: True, if name is importable. __init__.py.pyrNF find_spec) rrisdirrjoinsysexistssplitrrrr find_module)namer name_pathrfind_spec_exists find_methodr r r IsImportableGs    " r+cCstjtj|drtj|tj|d}ntj|dr*tj||d}nt||Stj|}|t j |<|j ||S)aLoads module at given path under given name. Note that it also updates sys.modules with name_to_give. Args: name_to_give: str, name to assign to loaded module module_path: str, python path to location of the module, this is either filesystem path or path into egg or zip package Returns: Imported module Raises: ImportError: if module cannot be imported. rr) rrrr" importlibutilspec_from_file_location_GetModuleFromPathViaPkgutilmodule_from_specr#modulesloader exec_module) name_to_give module_pathspecmoduler r r GetModuleFromPathes   r8cCsnttj|}|std|t|d}|r|jn|j }tj |}||r0t ||||Std|)z5Loads module by using pkgutil.get_importer mechanism.z {0} not foundr ) rrrrr ImportErrorrrr r&basename _LoadModule)r5r4rr)r*r r r r r/s  r/cCs\||}t|}||r|g|_tj|d|_n|d|_t ||j |t j |<|S)z-Loads the module or package under given name.z __init__.pycz.pyc) get_codetypes ModuleType is_package__path__rrr"__file__exec__dict__r#r1)rr5r r4coder7r r r r;s      r;c cst}|dur g}|durd}|D]s}||sq|t|dtj}t|dkrG|ddrG|d|vrG||d|ddfVt|dkrNqtj|d}tj |\}}|dksj|d krk||vrkq|d krq|n|} d |vr| |vr|| | d fVqdS) zDYields module names from given list of file paths with given prefix.NrrrT__init__rrF) set startswithlenr%rsepaddrr:splitext) file_listextra_extensionsprefixyielded file_pathfile_path_partsfilenamemodnameextto_yieldr r r _IterModuless8     rXcCsjg}t|D]+}tjtj||r||qtj||d}tj|r2|tj|dq|S)z=List packages or modules which can be imported at given path.r)rlistdirrrr"append)r importablesrTpkg_init_filepathr r r _ListPackagesAndFiless  r]cCs t|dr |j}|S|}|S)z&Get a list of files from the importer._files)rr^ _get_files)r files_listr r r _GetFilesLists racCs~g}tj|rtt||}nt|}tt|||j}gg}}|D]\}}|r1| |q%| |q%t |t |fS)a$Returns list of packages and modules in given path. Args: path: str, filesystem path extra_extensions: [str], The list of file extra extensions that should be considered modules for the purposes of listing (in addition to .py). Returns: tuple([packages], [modules]) ) rrr!rXr]rrrarPrZsorted)rrO iter_modulesrpackagesr1r'ispkgr r r ListPackages       rfccsXtj|rt|}z |VW|dS|wt|}t|dtj VdS)z!Get a file reader for given path.zutf-8N) rrrr FileReadercloserstrr%linesep)rf byte_stringr r r GetFileTextReaderByLines  rm*.*cCstj|rt|d|St|}t|ds!td|g}t |D]2}| |j s0q'|t |j d tj}t |dkrDq't|d|rY|tj||dq'|S)a,Get files from a given directory that match a pattern. Args: path_dir: str, filesystem path to directory filter_pattern: str, pattern to filter files to retrieve. Returns: List of filtered files from a directory. Raises: IOError: if resource is not found under given path. /rzPath not found {0}NrFr)rrr!globrrrrrrarIrPrJr%rKfnmatchrZr")path_dirfilter_patternrfiltered_filesrRrSr r r GetFilesFromDirectorys$      ru)N)rn)__doc__ __future__rrr contextlibrqrpimportlib.utilr,rrr#r=googlecloudsdk.core.utilrr rrr+r8r/r;rXr]rarfcontextmanagerrmrur r r r s6    ( "