ndZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z d dZ dZ d d Z y) zUtilities for local mode.)absolute_import)division)unicode_literalsN)filescH|tj}tj|tjtjtj dd5}|r*|j j|jd|j jtj|jd}|D]#}|j||j%|j ddd|jS#1swYjSxYw)aFExecutes shell commands in subprocess. Executes the supplied command with the supplied standard input string, streams the output to stdout, and returns the process's return code. Args: cmd: (List[str]) Strings to send in as the command. input_str: (str) if supplied, it will be passed as stdin to the supplied command. if None, stdin will get closed immediately. file: optional file-like object (stream), the output from the executed process's stdout will get sent to this stream. Defaults to sys.stdout. Returns: return code of the process NF)stdinstdoutstderruniversal_newlinesbufsizezutf-8)newline)sysr subprocessPopenPIPESTDOUTr writeencodecloseio TextIOWrapperflush returncode)cmd input_strfilepoutlines ;lib/googlecloudsdk/command_lib/ai/custom_jobs/local_util.pyExecuteCommandr#s  \ ::D OO __     ggmmI$$W-.GGMMO  188R 0C jj jjl  jjl%( )( s B D  D!c\|jdtjjdzS)zConverts the supplied python module into corresponding python file. Args: module_name: (str) A python module name (separated by dots) Returns: A string representing a python file path. .z.py)replaceospathsep) module_names r" ModuleToPathr+Gs$   S"''++ . 66c|xstj}d}tj|D]}tjj ||}tjj |r|dk(sHtj|d}`tjj|\}}|dvstj|d}|S)z@Removes generic `__pycache__` folder and '*.pyc' '*.pyo' files.F __pycache__T)z.pycz.pyo) rGetCWDr'listdirr(joinisdirRmTreesplitextremove)root_dir is_cleanednameitem_exts r" ClearPyCacher<Ss  '(*jj"d 77<<$ 'D ww}}T   T ww%fa   $ # r,)NN)N)__doc__ __future__rrrrr'rrgooglecloudsdk.core.utilrr#r+r<r,r"rAs2 &'  *'T 7r,