eG ddlmZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z ejd dZej d ddZej d dd Zej d dd Zddd Zy)) annotationsN) GeneratorIterator)copy)Path)TextIOc#K|xt_t_ dtjt_tjt_y#tjt_tjt_wxYww)z+Patch and subsequently reset a text stream.N)sysstderrstdout __stderr__ __stdout__)outs 8/usr/lib/python3/dist-packages/pylint/testutils/utils.py_patch_streamsrsM"!CJ$ ^^ ^^ ^^ ^^ sBA3B4BBc#Ktj} |t|t_d|t_y#|t_wxYwwN)r pathr)replacement_sys_path original_paths r_test_sys_pathrs:HHM!  +01CH  =sA : A  AA c#Ktj} |tj|dtj|y#tj|wxYwwr)osgetcwdchdir)current_working_directory original_dirs r _test_cwdr(sF99;L $ 0 HH. /  sA#A A# A  A#c#Ktjjd}|r|tjd<n||tjd= d||tjd<ydtjvrtjd=yy#||tjd<wdtjvrtjd=wwxYww)N PYTHONPATH)renvironget)new_pythonpathoriginal_pythonpaths r_test_environ_pythonpathr%5s**..6#1 <  $7$C JJ| $)  *':BJJ| $ RZZ ' <((  *':BJJ| $ RZZ ' <((sA C B ;C  . :param list paths: list of relative paths to files or directories :param str chroot: the root directory in which paths will be created >>> from os.path import isdir, isfile >>> isdir('/tmp/a') False >>> create_files(['a/b/foo.py', 'a/b/c/', 'a/b/c/d/e.py'], '/tmp') >>> isdir('/tmp/a') True >>> isdir('/tmp/a/b/c') True >>> isfile('/tmp/a/b/c/d/e.py') True >>> isfile('/tmp/a/b/foo.py') True wzutf-8)encodingN) setrrjoinbasenameadddirnameisdirmakedirsopen)pathschrootdirsfilesrfilenamedirpathfilepaths r create_filesr8Hs&%%D ww||FD)77##D) HHTN HHRWW__T* + IIdO !ww}}W% KK ! (C' 2      s =D  D )rrreturnzIterator[None]r)rzlist[str] | Noner9Generator[None, None, None])rzstr | Path | Noner9r:)r#z str | Noner9r:).)r1z list[str]r2strr9None) __future__r contextlibrr collections.abcrrrpathlibrtypingrcontextmanagerrrrr%r8rrFs # / $$ -1 !* !  ! ! 37 0     !%)) ))$#rE