daklib/holding.py

   1 #!/usr/bin/env python
   2 # vim:set et sw=4:
   3
   4 """
   5 Simple singleton class for storing info about Holding directory
   6
   7 @contact: Debian FTP Master <ftpmaster@debian.org>
   8 @copyright: 2001 - 2006 James Troup <james@nocrew.org>
   9 @copyright: 2009  Joerg Jaspert <joerg@debian.org>
  10 @license: GNU General Public License version 2 or later
  11 """
  12
  13 # This program is free software; you can redistribute it and/or modify
  14 # it under the terms of the GNU General Public License as published by
  15 # the Free Software Foundation; either version 2 of the License, or
  16 # (at your option) any later version.
  17
  18 # This program is distributed in the hope that it will be useful,
  19 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  20 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21 # GNU General Public License for more details.
  22
  23 # You should have received a copy of the GNU General Public License
  24 # along with this program; if not, write to the Free Software
  25 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  26
  27 ###############################################################################
  28
  29 import os
  30 from errno import ENOENT, EEXIST, EACCES
  31
  32 from singleton import Singleton
  33 from config import Config
  34 from utils import fubar
  35
  36 ###############################################################################
  37
  38 class Holding(Singleton):
  39     def __init__(self, *args, **kwargs):
  40         super(Holding, self).__init__(*args, **kwargs)
  41
  42     def _startup(self):
  43         self.in_holding = {}
  44         self.holding_dir = Config()["Dir::Queue::Holding"]
  45
  46     def copy_to_holding(self, filename):
  47         base_filename = os.path.basename(filename)
  48
  49         dest = os.path.join(self.holding_dir, base_filename)
  50         try:
  51             fd = os.open(dest, os.O_RDWR | os.O_CREAT | os.O_EXCL, 0640)
  52             os.close(fd)
  53         except OSError, e:
  54             # Shouldn't happen, but will if, for example, someone lists a
  55             # file twice in the .changes.
  56             if e.errno == EEXIST:
  57                 return "%s: already exists in holding area; can not overwrite." % (base_filename)
  58
  59         try:
  60             shutil.copy(filename, dest)
  61         except IOError, e:
  62             # In either case (ENOENT or EACCES) we want to remove the
  63             # O_CREAT | O_EXCLed ghost file, so add the file to the list
  64             # of 'in holding' even if it's not the real file.
  65             if e.errno == ENOENT:
  66                 os.unlink(dest)
  67                 return "%s: can not copy to holding area: file not found." % (base_filename)
  68
  69             elif e.errno == EACCES:
  70                 os.unlink(dest)
  71                 return "%s: can not copy to holding area: read permission denied." % (base_filename)
  72
  73         self.in_holding[base_filename] = ""
  74
  75         return None
  76
  77     def clean(self, filename):
  78         cwd = os.getcwd()
  79         os.chdir(self.holding_dir)
  80         for f in self.in_holding.keys():
  81             # TODO: Sanitize path in a much better manner...
  82             if os.path.exists(f):
  83                 if f.find('/') != -1:
  84                     fubar("WTF? clean_holding() got a file ('%s') with / in it!" % (f))
  85                 else:
  86                     os.unlink(f)
  87         self.in_holding = {}
  88         os.chdir(cwd)
  89