f98ee40f4b
This is an ongoing series of commits that are reformatting our Python code. This catches the last of the python files to reformat. Since they where so few I bunched them together. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, #libc, Mordante, sivachandra Differential Revision: https://reviews.llvm.org/D150784
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
# DExTer : Debugging Experience Tester
|
|
# ~~~~~~ ~ ~~ ~ ~~
|
|
#
|
|
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
# See https://llvm.org/LICENSE.txt for license information.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
"""Create/set a temporary working directory for some operations."""
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import time
|
|
|
|
from dex.utils.Exceptions import Error
|
|
|
|
|
|
class WorkingDirectory(object):
|
|
def __init__(self, context, *args, **kwargs):
|
|
self.context = context
|
|
self.orig_cwd = os.getcwd()
|
|
|
|
dir_ = kwargs.get("dir", None)
|
|
if dir_ and not os.path.isdir(dir_):
|
|
os.makedirs(dir_, exist_ok=True)
|
|
self.path = tempfile.mkdtemp(*args, **kwargs)
|
|
|
|
def __enter__(self):
|
|
os.chdir(self.path)
|
|
return self
|
|
|
|
def __exit__(self, *args):
|
|
os.chdir(self.orig_cwd)
|
|
if self.context.options.save_temps:
|
|
self.context.o.blue('"{}" left in place [--save-temps]\n'.format(self.path))
|
|
return
|
|
|
|
for _ in range(100):
|
|
try:
|
|
shutil.rmtree(self.path)
|
|
return
|
|
except OSError:
|
|
time.sleep(0.1)
|
|
|
|
self.context.logger.warning(
|
|
f'"{self.path}" left in place (couldn\'t delete)', enable_prefix=True
|
|
)
|
|
return
|