o ȷeA@sdZddlmZddlZddlZddlZzddl m Z Wne y5ddl m Z m Z GdddeZ YnwGdd 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|_dSNF)rname_closed)selfsuffixprefixdirrO/var/www/ideatree/venv/lib/python3.10/site-packages/ipython_genutils/tempdir.py__init__#s zTemporaryDirectory.__init__cC|jSN)rr rrr __enter__'zTemporaryDirectory.__enter__Fc Cs|jrF|jsHz||jWn'ttfy5}zdt|vrtd||tj dWYd}~dSd}~wwd|_|rJ| d|t dSdSdSdS)NNonez"ERROR: {!r} while cleaning up {!r}fileTzImplicitly cleaning up {!r}) rr _rmtree TypeErrorAttributeErrorstrprintformat_sysstderr_warnWarning)r r!exrrrcleanup*s&      zTemporaryDirectory.cleanupcC |dSrr$r excvaluetbrrr__exit__< zTemporaryDirectory.__exit__cCs|jdddS)NT)r!r&rrrr__del__?szTemporaryDirectory.__del__c Cs||D]5}|||}z||}Wn |jy d}Ynw|r)||qz||Wq|jy:Yqwz||WdS|jyNYdSwr)_listdir _path_join_isdir _os_errorr_remove_rmdir)r pathrfullnameisdirrrrrPs&  zTemporaryDirectory._rmtree)F)__name__ __module__ __qualname____doc__rrrr$r+r- staticmethod_oslistdirr.r4joinr/r6r0remover2rmdirr3errorr1 _warningswarnr!rrrrrrs       rc@s2eZdZd ddZddZeZddZd d Zd S) NamedFileInTemporaryDirectoryw+bcKs4tdi||_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`. Nr)r_tmpdirr<r4r>ropenr)r filenamemodebufsizekwdsr4rrrris z&NamedFileInTemporaryDirectory.__init__cCs|j|jdSr)rcloserGr$rrrrr$xs z%NamedFileInTemporaryDirectory.cleanupcCrrrrrrrr~rz'NamedFileInTemporaryDirectory.__enter__cCr%rr&)r typer) tracebackrrrr+r,z&NamedFileInTemporaryDirectory.__exit__N)rErF)r7r8r9rr$r-rr+rrrrrDgs   rDcs,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|Sr)r<getcwdold_wdchdirrsuperrPrr __class__rrrs  z#TemporaryWorkingDirectory.__enter__cs t|jtt||||Sr)r<rSrRrTrPr+r'rUrrr+s z"TemporaryWorkingDirectory.__exit__)r7r8r9r:rr+ __classcell__rrrUrrPs rP)r: __future__rosr<warningsrBsysrtempfiler ImportErrorrrobjectrDrPrrrrs  S