diff --git a/cruft/_commands/utils/iohelper.py b/cruft/_commands/utils/iohelper.py index 2cbabae..49757d0 100644 --- a/cruft/_commands/utils/iohelper.py +++ b/cruft/_commands/utils/iohelper.py @@ -1,15 +1,23 @@ from tempfile import TemporaryDirectory from time import sleep +import sys class AltTemporaryDirectory: def __init__(self): self.tmpdir = TemporaryDirectory() + self._extended_path = False + name = self.tmpdir.name + if name not in sys.path: + self._extended_path = True + sys.path.append(name) def __enter__(self): return self.tmpdir.name def cleanup(self, cnt=0): + if self._extended_path: + sys.path.remove(self.tmpdir.name) if cnt >= 5: # pragma: no cover raise RuntimeError("Could not delete TemporaryDirectory!") try: