B &dA@sdZddlmZddlZddlZddlZyddl m Z Wn4e k rlddl m Z m Z GdddeZ YnXGdd d eZGd d d e ZdS) zTemporaryDirectory class, copied from Python 3 This is copied from the stdlib and will be standard in Python 3.2 and onwards. )print_functionN)TemporaryDirectory)mkdtemptemplatec@seZdZdZdedfddZddZdd d Zd d Zd dZ e e j Z e e jjZe e jjZe e jZe e jZe jZejZddZdS)raFCreate 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 everthing contained in it are removed. NcCst||||_d|_dS)NF)rname_closed)selfsuffixprefixdirr e/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/ipython_genutils/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__<szTemporaryDirectory.__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 rrPs     zTemporaryDirectory._rmtree)F)__name__ __module__ __qualname____doc__rrrrr"r# staticmethod_oslistdirr$r*joinr%r,r&remover(rmdirr)errorr' _warningswarnrrr r r rrs       rc@s2eZdZd ddZddZeZddZd d Zd S) NamedFileInTemporaryDirectoryw+bcKs0tf||_tj|jj|}t||||_dS)aN Open a file named `filename` in a temporary directory. This context manager is preferred over `NamedTemporaryFile` in stdlib `tempfile` when one needs to reopen the file. Arguments `mode` and `bufsize` are passed to `open`. Rest of the arguments are passed to `TemporaryDirectory`. N)r_tmpdirr2r*r4ropenr)r filenamemodebufsizekwdsr*r r rris z&NamedFileInTemporaryDirectory.__init__cCs|j|jdS)N)rcloser=r)r r r rrxs z%NamedFileInTemporaryDirectory.cleanupcCs|jS)N)r)r r r rr~sz'NamedFileInTemporaryDirectory.__enter__cCs |dS)N)r)r typer tracebackr r rr"sz&NamedFileInTemporaryDirectory.__exit__N)r;r<)r-r.r/rrr#rr"r r r rr:gs  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)r0 __future__rosr2warningsr8sysrtempfiler ImportErrorrrobjectr:rFr r r rs P