from daklib.config import Config
from daklib import daklog
from daklib.changesutils import *
-from daklib.threadpool import ThreadPool
+from daklib.dakmultiprocessing import DakProcessPool, PROC_STATUS_SUCCESS, PROC_STATUS_SIGNALRAISED
+from multiprocessing import Manager, TimeoutError
# Globals
Cnf = None
Options = None
-sources = set()
+manager = Manager()
+sources = manager.list()
+htmlfiles_to_process = manager.list()
################################################################################
def do_pkg(changes_file):
+ changes_file = utils.validate_changes_file_arg(changes_file, 0)
+ if not changes_file:
+ return
+ print "\n" + changes_file
+
session = DBConn().session()
u = Upload()
u.pkg.changes_file = changes_file
- (u.pkg.changes["fingerprint"], rejects) = utils.check_signature(changes_file)
+ # We can afoord not to check the signature before loading the changes file
+ # as we've validated it already (otherwise it couldn't be in new)
+ # and we can more quickly skip over already processed files this way
u.load_changes(changes_file)
- new_queue = get_policy_queue('new', session );
- u.pkg.directory = new_queue.path
- u.update_subst()
+
origchanges = os.path.abspath(u.pkg.changes_file)
- files = u.pkg.files
- changes = u.pkg.changes
- htmlname = changes["source"] + "_" + changes["version"] + ".html"
- sources.add(htmlname)
- htmlfile = os.path.join(cnf["Show-New::HTMLPath"], htmlname)
+ # Still be cautious in case paring the changes file went badly
+ if u.pkg.changes.has_key('source') and u.pkg.changes.has_key('version'):
+ htmlname = u.pkg.changes["source"] + "_" + u.pkg.changes["version"] + ".html"
+ htmlfile = os.path.join(cnf["Show-New::HTMLPath"], htmlname)
+ htmlfiles_to_process.append(htmlfile)
+ else:
+ # Changes file was bad
+ print "Changes file %s missing source or version field" % changes_file
+ session.close()
+ return
+
+ # Have we already processed this?
if os.path.exists(htmlfile) and \
os.stat(htmlfile).st_mtime > os.stat(origchanges).st_mtime:
+ sources.append(htmlname)
session.close()
- return
+ return (PROC_STATUS_SUCCESS, '%s already up-to-date' % htmlfile)
+
+ # Now we'll load the fingerprint
+ (u.pkg.changes["fingerprint"], rejects) = utils.check_signature(changes_file, session=session)
+ new_queue = get_policy_queue('new', session );
+ u.pkg.directory = new_queue.path
+ u.update_subst()
+ files = u.pkg.files
+ changes = u.pkg.changes
+ sources.append(htmlname)
for deb_filename, f in files.items():
if deb_filename.endswith(".udeb") or deb_filename.endswith(".deb"):
outfile.close()
session.close()
+ htmlfiles_to_process.remove(htmlfile)
+ return (PROC_STATUS_SUCCESS, '%s already updated' % htmlfile)
+
################################################################################
def usage (exit_code=0):
examine_package.use_html=1
- threadpool = ThreadPool()
- for changes_file in changes_files:
- changes_file = utils.validate_changes_file_arg(changes_file, 0)
- if not changes_file:
- continue
- print "\n" + changes_file
- threadpool.queueTask(do_pkg, changes_file)
- threadpool.joinAll()
+ pool = DakProcessPool()
+ p = pool.map_async(do_pkg, changes_files)
+ pool.close()
+ try:
+ p.get(timeout=600)
+ except TimeoutError:
+ for htmlfile in htmlfiles_to_process:
+ with open(htmlfile, "w") as fd:
+ fd.write("Timed out while processing")
files = set(os.listdir(cnf["Show-New::HTMLPath"]))
- to_delete = filter(lambda x: x.endswith(".html"), files.difference(sources))
+ to_delete = filter(lambda x: x.endswith(".html"), files.difference(set(sources)))
for f in to_delete:
os.remove(os.path.join(cnf["Show-New::HTMLPath"],f))