summaryrefslogtreecommitdiffstats
path: root/ipapython/dnssec/temp.py
blob: 23ee377bece178ecfb1a0cdaa1168d03fb0b3177 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#!/usr/bin/python
#
# Copyright (C) 2014  FreeIPA Contributors see COPYING for license
#

import errno
import shutil
import tempfile

class TemporaryDirectory(object):
    def __init__(self, root):
        self.root = root

    def __enter__(self):
        self.name = tempfile.mkdtemp(dir=self.root)
        return self.name

    def __exit__(self, exc_type, exc_value, traceback):
        try:
            shutil.rmtree(self.name)
        except OSError as e:
            if e.errno != errno.ENOENT:
                raise