From 50a948a728a0768907700befe501bb743828b67b Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Thu, 29 Sep 2011 10:51:04 +0100 Subject: Initial import. --- morphlib/tempdir.py | 58 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 58 insertions(+) create mode 100644 morphlib/tempdir.py (limited to 'morphlib/tempdir.py') diff --git a/morphlib/tempdir.py b/morphlib/tempdir.py new file mode 100644 index 00000000..7f58cdc3 --- /dev/null +++ b/morphlib/tempdir.py @@ -0,0 +1,58 @@ +# Copyright (C) 2011 Codethink Limited +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License along +# with this program; if not, write to the Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + + +import logging +import os +import shutil +import tempfile + + +class Tempdir(object): + + '''Temporary file handling for morph.''' + + def __init__(self, parent=None): + self.dirname = tempfile.mkdtemp(dir=parent) + logging.debug('Created temporary directory %s' % self.dirname) + + def remove(self): + '''Remove the temporary directory.''' + logging.debug('Removing temporary directory %s' % self.dirname) + shutil.rmtree(self.dirname) + self.dirname = None + + def clear(self): + '''Clear temporary directory of everything.''' + for x in os.listdir(self.dirname): + filename = self.join(x) + if os.path.isdir(filename): + shutil.rmtree(filename) + else: + os.remove(filename) + + def join(self, relative): + '''Return full path to file in temporary directory. + + The relative path is given appended to the name of the + temporary directory. If the relative path is actually absolute, + it is forced to become relative. + + The returned path is normalized. + + ''' + + return os.path.normpath(os.path.join(self.dirname, './' + relative)) + -- cgit v1.2.1