B 4[~@sdZddlmZddlZddlZddlZyddl m Z Wn4e k rlddl m Z m Z GdddeZ YnXGdd d eZGd d d e ZdS) zTemporaryDirectory class, copied from Python 3 NamedFileInTemporaryDirectory and TemporaryWorkingDirectory from IPython, which uses the 3-clause BSD license. )print_functionN)TemporaryDirectory)mkdtemptemplatec@seZdZdZdedfddZddZdd d Zd d Zd dZ e e j Z e e jjZe e jjZe e jZe e jZe jZejZddZdS)raGCreate and return a temporary directory. This has the same behavior as mkdtemp but can be used as a context manager. For example: with TemporaryDirectory() as tmpdir: ... Upon exiting the context, the directory and everything contained in it are removed. NcCst||||_d|_dS)NF)rname_closed)selfsuffixprefixdirr /lib/python3.7/site-packages/testpath/tempdir.py__init__ szTemporaryDirectory.__init__cCs|jS)N)r)r r r r __enter__$szTemporaryDirectory.__enter__Fc Cs|jr|jsy||jWnJttfk rf}z(dt|kr@td||tj ddSd}~XYnXd|_|r| d|t dS)NNonez"ERROR: {!r} while cleaning up {!r})fileTzImplicitly cleaning up {!r}) rr_rmtree TypeErrorAttributeErrorstrprintformat_sysstderr_warnWarning)r rexr r rcleanup's     zTemporaryDirectory.cleanupcCs |dS)N)r)r excvaluetbr r r__exit__9szTemporaryDirectory.__exit__cCs|jdddS)NT)r)r)r r r r__del__<szTemporaryDirectory.__del__c Csx~||D]p}|||}y||}Wn|jk rDd}YnX|rV||q y||Wq |jk rzYq Xq Wy||Wn|jk rYnXdS)NF)_listdir _path_join_isdir _os_errorr_remove_rmdir)r pathrfullnameisdirr r rrMs     zTemporaryDirectory._rmtree)F)__name__ __module__ __qualname____doc__rrrrr"r# staticmethod_oslistdirr$r*joinr%r,r&remover(rmdirr)errorr' _warningswarnrrr r r rrs       rc@s6eZdZdZd ddZddZeZdd Zd d Zd S)NamedFileInTemporaryDirectoryaOpen a file named `filename` in a temporary directory. This context manager is preferred over :class:`tempfile.NamedTemporaryFile` when one needs to reopen the file, because on Windows only one handle on a file can be open at a time. You can close the returned handle explicitly inside the context without deleting the file, and the context manager will delete the whole directory when it exits. Arguments `mode` and `bufsize` are passed to `open`. Rest of the arguments are passed to `TemporaryDirectory`. Usage example:: with NamedFileInTemporaryDirectory('myfile', 'wb') as f: f.write('stuff') f.close() # You can now pass f.name to things that will re-open the file w+bcKs0tf||_tj|jj|}t||||_dS)N)r_tmpdirr2r*r4ropenr)r filenamemodebufsizekwdsr*r r rrvs z&NamedFileInTemporaryDirectory.__init__cCs|j|jdS)N)rcloser=r)r r r rr{s z%NamedFileInTemporaryDirectory.cleanupcCs|jS)N)r)r r r rrsz'NamedFileInTemporaryDirectory.__enter__cCs |dS)N)r)r typer tracebackr r rr"sz&NamedFileInTemporaryDirectory.__exit__N)r;r<) r-r.r/r0rrr#rr"r r r rr:cs  r:cs,eZdZdZfddZfddZZS)TemporaryWorkingDirectoryz Creates a temporary directory and sets the cwd to that directory. Automatically reverts to previous cwd upon cleanup. Usage example:: with TemporaryWorkingDirectory() as tmpdir: ... cs$t|_t|jtt|S)N)r2getcwdold_wdchdirrsuperrFr)r ) __class__r rrs  z#TemporaryWorkingDirectory.__enter__cs t|jtt||||S)N)r2rIrHrJrFr")r rr r!)rKr rr"s z"TemporaryWorkingDirectory.__exit__)r-r.r/r0rr" __classcell__r r )rKrrFs  rF)r0Z __future__rosr2warningsr8sysrZtempfiler ImportErrorrrobjectr:rFr r r rs O%