Merge pull request #1554 from clinton-hall/fix/cleanup

Add exception handling for failure to return to original directory
This commit is contained in:
Labrys of Knossos 2019-02-18 06:33:34 -05:00 committed by GitHub
commit 506ede833e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -33,7 +33,17 @@ class WorkingDirectory(object):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
os.chdir(self.original_directory)
try:
os.chdir(self.original_directory)
except OSError as error:
print(
'Unable to return to {original_directory}: {error}\n'
'Continuing in {working_directory}'.format(
original_directory=self.original_directory,
error=error,
working_directory=self.working_directory,
)
)
def module_path(module=__file__, parent=False):