5 Import known_changes files
7 @contact: Debian FTP Master <ftpmaster@debian.org>
8 @copyright: 2009 Mike O'Connor <stew@debian.org>
9 @license: GNU General Public License version 2 or later
12 # This program is free software; you can redistribute it and/or modify
13 # it under the terms of the GNU General Public License as published by
14 # the Free Software Foundation; either version 2 of the License, or
15 # (at your option) any later version.
17 # This program is distributed in the hope that it will be useful,
18 # but WITHOUT ANY WARRANTY; without even the implied warranty of
19 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 # GNU General Public License for more details.
22 # You should have received a copy of the GNU General Public License
23 # along with this program; if not, write to the Free Software
24 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 ################################################################################
29 ################################################################################
35 from daklib.dbconn import DBConn,get_knownchange
36 from daklib.config import Config
38 from daklib.dak_exceptions import DBUpdateError, InvalidDscError, ChangesUnicodeError
39 from daklib.changes import Changes
40 from daklib.utils import parse_changes, warn, gpgv_get_status_output, process_gpgv_output
43 # where in dak.conf all of our configuration will be stowed
44 options_prefix = "KnownChanges"
45 options_prefix = "%s::Options" % options_prefix
47 log = logging.getLogger()
49 ################################################################################
52 def usage (exit_code=0):
53 print """Usage: dak import-known-changes [options]
57 run with n threads concurrently
60 show verbose information messages
63 supress all output but errors
68 def check_signature (sig_filename, data_filename=""):
72 "/home/joerg/keyring/keyrings/debian-keyring.gpg",
73 "/home/joerg/keyring/keyrings/debian-keyring.pgp",
74 "/home/joerg/keyring/keyrings/debian-maintainers.gpg",
75 "/home/joerg/keyring/keyrings/debian-role-keys.gpg",
76 "/home/joerg/keyring/keyrings/emeritus-keyring.pgp",
77 "/home/joerg/keyring/keyrings/emeritus-keyring.gpg",
78 "/home/joerg/keyring/keyrings/removed-keys.gpg",
79 "/home/joerg/keyring/keyrings/removed-keys.pgp"
82 keyringargs = " ".join(["--keyring %s" % x for x in keyrings ])
84 # Build the command line
85 status_read, status_write = os.pipe()
86 cmd = "gpgv --status-fd %s %s %s" % (status_write, keyringargs, sig_filename)
88 # Invoke gpgv on the file
89 (output, status, exit_status) = gpgv_get_status_output(cmd, status_read, status_write)
91 # Process the status-fd output
92 (keywords, internal_error) = process_gpgv_output(status)
94 # If we failed to parse the status-fd output, let's just whine and bail now
96 warn("Couldn't parse signature")
99 # usually one would check for bad things here. We, however, do not care.
101 # Next check gpgv exited with a zero return code
103 warn("Couldn't parse signature")
106 # Sanity check the good stuff we expect
107 if not keywords.has_key("VALIDSIG"):
108 warn("Couldn't parse signature")
110 args = keywords["VALIDSIG"]
112 warn("Couldn't parse signature")
114 fingerprint = args[0]
119 class EndOfChanges(object):
120 """something enqueued to signify the last change"""
124 class OneAtATime(object):
126 a one space queue which sits between multiple possible producers
127 and multiple possible consumers
130 self.next_in_line = None
131 self.next_lock = threading.Condition()
136 self.next_lock.notify()
138 def enqueue(self, next):
139 self.next_lock.acquire()
140 while self.next_in_line:
143 self.next_lock.wait()
145 assert( not self.next_in_line )
146 self.next_in_line = next
147 self.next_lock.notify()
148 self.next_lock.release()
151 self.next_lock.acquire()
152 while not self.next_in_line:
155 self.next_lock.wait()
157 result = self.next_in_line
159 self.next_in_line = None
160 self.next_lock.notify()
161 self.next_lock.release()
163 if isinstance(result, EndOfChanges):
168 class ChangesToImport(object):
169 """A changes file to be enqueued to be processed"""
170 def __init__(self, checkdir, changesfile, count):
171 self.dirpath = checkdir
172 self.changesfile = changesfile
176 return "#%d: %s in %s" % (self.count, self.changesfile, self.dirpath)
178 class ChangesGenerator(threading.Thread):
179 """enqueues changes files to be imported"""
180 def __init__(self, parent, queue):
181 threading.Thread.__init__(self)
183 self.session = DBConn().session()
193 for directory in [ "Accepted", "Byhand", "Done", "New", "ProposedUpdates", "OldProposedUpdates" ]:
194 checkdir = cnf["Dir::Queue::%s" % (directory) ]
195 if os.path.exists(checkdir):
196 print "Looking into %s" % (checkdir)
198 for dirpath, dirnames, filenames in os.walk(checkdir, topdown=False):
200 # Empty directory (or only subdirectories), next
203 for changesfile in filenames:
205 if not changesfile.endswith(".changes"):
206 # Only interested in changes files.
210 if not get_knownchange(changesfile, self.session):
211 to_import = ChangesToImport(dirpath, changesfile, count)
214 self.queue.enqueue(to_import)
215 except KeyboardInterrupt:
216 print("got Ctrl-c in enqueue thread. terminating")
220 self.queue.enqueue(EndOfChanges())
222 class ImportThread(threading.Thread):
223 def __init__(self, parent, queue):
224 threading.Thread.__init__(self)
226 self.session = DBConn().session()
238 to_import = self.queue.dequeue()
242 print( "Directory %s, file %7d, (%s)" % (to_import.dirpath[-10:], to_import.count, to_import.changesfile) )
245 changes.changes_file = to_import.changesfile
246 changesfile = os.path.join(to_import.dirpath, to_import.changesfile)
247 changes.changes = parse_changes(changesfile, signing_rules=-1)
248 changes.changes["fingerprint"] = check_signature(changesfile)
249 changes.add_known_changes(to_import.dirpath, self.session)
250 self.session.commit()
252 except InvalidDscError, line:
253 warn("syntax error in .dsc file '%s', line %s." % (f, line))
255 except ChangesUnicodeError:
256 warn("found invalid changes file, not properly utf-8 encoded")
259 except KeyboardInterrupt:
260 print("Caught C-c; on ImportThread. terminating.")
264 traceback.print_exc()
268 class ImportKnownChanges(object):
269 def __init__(self,num_threads):
270 self.queue = OneAtATime()
271 self.threads = [ ChangesGenerator(self,self.queue) ]
273 for i in range(num_threads):
274 self.threads.append( ImportThread(self,self.queue) )
277 for thread in self.threads:
280 except KeyboardInterrupt:
281 print("Caught C-c; terminating.")
282 utils.warn("Caught C-c; terminating.")
286 traceback.print_stack90
287 for thread in self.threads:
288 print( "STU: before ask %s to die" % thread )
290 print( "STU: after ask %s to die" % thread )
299 arguments = [('h',"help", "%s::%s" % (options_prefix,"Help")),
300 ('j',"concurrency", "%s::%s" % (options_prefix,"Concurrency"),"HasArg"),
301 ('q',"quiet", "%s::%s" % (options_prefix,"Quiet")),
302 ('v',"verbose", "%s::%s" % (options_prefix,"Verbose")),
305 args = apt_pkg.ParseCommandLine(cnf.Cnf, arguments,sys.argv)
312 if cnf.has_key("%s::%s" % (options_prefix,"Help")):
316 if cnf.has_key("%s::%s" % (options_prefix,"Quiet")):
319 elif cnf.has_key("%s::%s" % (options_prefix,"Verbose")):
323 logging.basicConfig( level=level,
324 format='%(asctime)s %(levelname)s %(message)s',
325 stream = sys.stderr )
327 if Config().has_key( "%s::%s" %(options_prefix,"Concurrency")):
328 num_threads = int(Config()[ "%s::%s" %(options_prefix,"Concurrency")])
330 ImportKnownChanges(num_threads)
335 if __name__ == '__main__':