]> git.decadent.org.uk Git - dak.git/blob - daklib/holding.py
Prefer the "Borg" pattern over the GoF singleton pattern.
[dak.git] / 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 import shutil
32
33 from config import Config
34 from utils import fubar
35
36 ###############################################################################
37
38 class Holding(object):
39     __shared_state = {}
40
41     def __init__(self, *args, **kwargs):
42         self.__dict__ = self.__shared_state
43
44         if not getattr(self, 'initialised', False):
45             self.initialised = True
46
47             self.in_holding = {}
48             self.holding_dir = Config()["Dir::Queue::Holding"]
49
50     def copy_to_holding(self, filename):
51         base_filename = os.path.basename(filename)
52
53         dest = os.path.join(self.holding_dir, base_filename)
54         try:
55             fd = os.open(dest, os.O_RDWR | os.O_CREAT | os.O_EXCL, 0640)
56             os.close(fd)
57         except OSError, e:
58             # Shouldn't happen, but will if, for example, someone lists a
59             # file twice in the .changes.
60             if e.errno == EEXIST:
61                 return "%s: already exists in holding area; can not overwrite." % (base_filename)
62
63         try:
64             shutil.copy(filename, dest)
65         except IOError, e:
66             # In either case (ENOENT or EACCES) we want to remove the
67             # O_CREAT | O_EXCLed ghost file, so add the file to the list
68             # of 'in holding' even if it's not the real file.
69             if e.errno == ENOENT:
70                 os.unlink(dest)
71                 return "%s: can not copy to holding area: file not found." % (base_filename)
72
73             elif e.errno == EACCES:
74                 os.unlink(dest)
75                 return "%s: can not copy to holding area: read permission denied." % (base_filename)
76
77         self.in_holding[base_filename] = ""
78
79         return None
80
81     def clean(self):
82         cwd = os.getcwd()
83         os.chdir(self.holding_dir)
84         for f in self.in_holding.keys():
85             # TODO: Sanitize path in a much better manner...
86             if os.path.exists(f):
87                 if f.find('/') != -1:
88                     fubar("WTF? clean_holding() got a file ('%s') with / in it!" % (f))
89                 else:
90                     os.unlink(f)
91         self.in_holding = {}
92         os.chdir(cwd)
93