]> git.decadent.org.uk Git - dak.git/blob - daklib/urgencylog.py
move UrgencyLog class to daklib
[dak.git] / daklib / urgencylog.py
1 #!/usr/bin/env python
2 # vim:set et sw=4:
3
4 """
5 Urgency Logger class for dak
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 from singleton import Singleton
30 from config import Config
31 from utils import warn, open_file, move
32
33 ###############################################################################
34
35 class UrgencyLog(Singleton):
36     "Urgency Logger object"
37     def __init__(self, *args, **kwargs):
38         super(UrgencyLog, self).__init__(*args, **kwargs)
39
40     def _startup(self):
41         "Initialize a new Urgency Logger object"
42
43         self.timestamp = time.strftime("%Y%m%d%H%M%S")
44
45         # Create the log directory if it doesn't exist
46         self.log_dir = Config()["Dir::UrgencyLog"]
47
48         if not os.path.exists(self.log_dir) or not os.access(self.log_dir, os.W_OK):
49             warn("UrgencyLog directory %s does not exist or is not writeable, using /srv/ftp.debian.org/tmp/ instead" % (self.log_dir))
50             self.log_dir = '/srv/ftp.debian.org/tmp/'
51
52         # Open the logfile
53         self.log_filename = "%s/.install-urgencies-%s.new" % (self.log_dir, self.timestamp)
54         self.log_file = open_file(self.log_filename, 'w')
55         self.writes = 0
56
57     def log(self, source, version, urgency):
58         "Log an event"
59         self.log_file.write(" ".join([source, version, urgency])+'\n')
60         self.log_file.flush()
61         self.writes += 1
62
63     def close(self):
64         "Close a Logger object"
65         self.log_file.flush()
66         self.log_file.close()
67
68         if self.writes:
69             new_filename = "%s/install-urgencies-%s" % (self.log_dir, self.timestamp)
70             move(self.log_filename, new_filename)
71         else:
72             os.unlink(self.log_filename)
73