From: Mark Hymers <mhy@debian.org>
Date: Sat, 1 Aug 2009 16:13:12 +0000 (+0100)
Subject: put holding routines into module
X-Git-Url: https://git.decadent.org.uk/gitweb/?a=commitdiff_plain;h=1359e943ab10977306602036675b22da89e9a964;p=dak.git

put holding routines into module

Signed-off-by: Mark Hymers <mhy@debian.org>
---

diff --git a/daklib/holding.py b/daklib/holding.py
new file mode 100755
index 00000000..d77eecde
--- /dev/null
+++ b/daklib/holding.py
@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+# vim:set et sw=4:
+
+"""
+Simple singleton class for storing info about Holding directory
+
+@contact: Debian FTP Master <ftpmaster@debian.org>
+@copyright: 2001 - 2006 James Troup <james@nocrew.org>
+@copyright: 2009  Joerg Jaspert <joerg@debian.org>
+@license: GNU General Public License version 2 or later
+"""
+
+# 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, or
+# (at your option) any later version.
+
+# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+###############################################################################
+
+import os
+from errno import ENOENT, EEXIST, EACCES
+
+from singleton import Singleton
+from config import Config
+from utils import fubar
+
+###############################################################################
+
+class Holding(Singleton):
+    def __init__(self, *args, **kwargs):
+        super(Holding, self).__init__(*args, **kwargs)
+
+    def _startup(self):
+        self.in_holding = {}
+        self.holding_dir = Config()["Dir::Queue::Holding"]
+
+    def copy_to_holding(self, filename):
+        base_filename = os.path.basename(filename)
+
+        dest = os.path.join(self.holding_dir, base_filename)
+        try:
+            fd = os.open(dest, os.O_RDWR | os.O_CREAT | os.O_EXCL, 0640)
+            os.close(fd)
+        except OSError, e:
+            # Shouldn't happen, but will if, for example, someone lists a
+            # file twice in the .changes.
+            if e.errno == EEXIST:
+                return "%s: already exists in holding area; can not overwrite." % (base_filename)
+
+        try:
+            shutil.copy(filename, dest)
+        except IOError, e:
+            # In either case (ENOENT or EACCES) we want to remove the
+            # O_CREAT | O_EXCLed ghost file, so add the file to the list
+            # of 'in holding' even if it's not the real file.
+            if e.errno == ENOENT:
+                os.unlink(dest)
+                return "%s: can not copy to holding area: file not found." % (base_filename)
+
+            elif e.errno == EACCES:
+                os.unlink(dest)
+                return "%s: can not copy to holding area: read permission denied." % (base_filename)
+
+        self.in_holding[base_filename] = ""
+
+        return None
+
+    def clean(self, filename):
+        cwd = os.getcwd()
+        os.chdir(self.holding_dir)
+        for f in self.in_holding.keys():
+            # TODO: Sanitize path in a much better manner...
+            if os.path.exists(f):
+                if f.find('/') != -1:
+                    fubar("WTF? clean_holding() got a file ('%s') with / in it!" % (f))
+                else:
+                    os.unlink(f)
+        self.in_holding = {}
+        os.chdir(cwd)
+