| Index: recipe_modules/tempfile/api.py
|
| diff --git a/recipe_modules/tempfile/api.py b/recipe_modules/tempfile/api.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..1bbbf407013c9e5fb6e6efdf5533277b4fcc0144
|
| --- /dev/null
|
| +++ b/recipe_modules/tempfile/api.py
|
| @@ -0,0 +1,19 @@
|
| +# Copyright 2016 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import contextlib
|
| +
|
| +from recipe_engine import recipe_api
|
| +
|
| +
|
| +class TempfileApi(recipe_api.RecipeApi):
|
| + @contextlib.contextmanager
|
| + def temp_dir(self, prefix):
|
| + path = None
|
| + try:
|
| + path = self.m.path.mkdtemp(prefix)
|
| + yield path
|
| + finally:
|
| + if path:
|
| + self.m.shutil.rmtree(path, infra_step=True)
|
|
|