daklib/changesutils.py

   1 #!/usr/bin/env python
   2 # vim:set et ts=4 sw=4:
   3
   4 """Utilities for handling changes files
   5
   6 @contact: Debian FTP Master <ftpmaster@debian.org>
   7 @copyright: 2001, 2002, 2003, 2004, 2005, 2006  James Troup <james@nocrew.org>
   8 @copyright: 2009 Joerg Jaspert <joerg@debian.org>
   9 @copyright: 2009 Frank Lichtenheld <djpig@debian.org>
  10 @license: GNU General Public License version 2 or later
  11 """
  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.
  16
  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.
  21
  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
  25
  26 ################################################################################
  27
  28 import copy
  29 import os
  30 import stat
  31 import apt_pkg
  32
  33 from daklib.dbconn import *
  34 from daklib.queue import *
  35 from daklib import utils
  36 from daklib.config import Config
  37
  38 ################################################################################
  39
  40 __all__ = []
  41
  42 ################################################################################
  43
  44 def indiv_sg_compare (a, b):
  45     """Sort by source name, source, version, 'have source', and
  46        finally by filename."""
  47     # Sort by source version
  48     q = apt_pkg.VersionCompare(a["version"], b["version"])
  49     if q:
  50         return -q
  51
  52     # Sort by 'have source'
  53     a_has_source = a["architecture"].get("source")
  54     b_has_source = b["architecture"].get("source")
  55     if a_has_source and not b_has_source:
  56         return -1
  57     elif b_has_source and not a_has_source:
  58         return 1
  59
  60     return cmp(a["filename"], b["filename"])
  61
  62 __all__.append('indiv_sg_compare')
  63
  64 ############################################################
  65
  66 def sg_compare (a, b):
  67     a = a[1]
  68     b = b[1]
  69     """Sort by have note, source already in database and time of oldest upload."""
  70     # Sort by have note
  71     a_note_state = a["note_state"]
  72     b_note_state = b["note_state"]
  73     if a_note_state < b_note_state:
  74         return -1
  75     elif a_note_state > b_note_state:
  76         return 1
  77     # Sort by source already in database (descending)
  78     source_in_database = cmp(a["source_in_database"], b["source_in_database"])
  79     if source_in_database:
  80         return -source_in_database
  81
  82     # Sort by time of oldest upload
  83     return cmp(a["oldest"], b["oldest"])
  84
  85 __all__.append('sg_compare')
  86
  87 def sort_changes(changes_files, session, binaries = None):
  88     """Sort into source groups, then sort each source group by version,
  89     have source, filename.  Finally, sort the source groups by have
  90     note, time of oldest upload of each source upload."""
  91     if len(changes_files) == 1:
  92         return changes_files
  93
  94     sorted_list = []
  95     cache = {}
  96     # Read in all the .changes files
  97     for filename in changes_files:
  98         u = Upload()
  99         try:
 100             u.pkg.changes_file = filename
 101             u.load_changes(filename)
 102             u.update_subst()
 103             cache[filename] = copy.copy(u.pkg.changes)
 104             cache[filename]["filename"] = filename
 105         except:
 106             sorted_list.append(filename)
 107             break
 108     # Divide the .changes into per-source groups
 109     per_source = {}
 110     for filename in cache.keys():
 111         source = cache[filename]["source"]
 112         if not per_source.has_key(source):
 113             per_source[source] = {}
 114             per_source[source]["list"] = []
 115         per_source[source]["list"].append(cache[filename])
 116     # Determine oldest time and have note status for each source group
 117     for source in per_source.keys():
 118         q = session.query(DBSource).filter_by(source = source).all()
 119         per_source[source]["source_in_database"] = binaries and -(len(q)>0) or len(q)>0
 120         source_list = per_source[source]["list"]
 121         first = source_list[0]
 122         oldest = os.stat(first["filename"])[stat.ST_MTIME]
 123         have_note = 0
 124         for d in per_source[source]["list"]:
 125             mtime = os.stat(d["filename"])[stat.ST_MTIME]
 126             if mtime < oldest:
 127                 oldest = mtime
 128             have_note += has_new_comment(d["source"], d["version"], session)
 129         per_source[source]["oldest"] = oldest
 130         if not have_note:
 131             per_source[source]["note_state"] = 0; # none
 132         elif have_note < len(source_list):
 133             per_source[source]["note_state"] = 1; # some
 134         else:
 135             per_source[source]["note_state"] = 2; # all
 136         per_source[source]["list"].sort(indiv_sg_compare)
 137     per_source_items = per_source.items()
 138     per_source_items.sort(sg_compare)
 139     for i in per_source_items:
 140         for j in i[1]["list"]:
 141             sorted_list.append(j["filename"])
 142     return sorted_list
 143
 144 __all__.append('sort_changes')
 145
 146 ################################################################################
 147
 148 def changes_to_queue(upload, srcqueue, destqueue, session):
 149     """Move a changes file to a different queue and mark as approved for the
 150        source queue"""
 151
 152     try:
 153         chg = session.query(DBChange).filter_by(changesname=os.path.basename(upload.pkg.changes_file)).one()
 154     except NoResultFound:
 155         return False
 156
 157     chg.approved_for_id = srcqueue.policy_queue_id
 158
 159     for f in chg.files:
 160         # update the changes_pending_files row
 161         f.queue = destqueue
 162         # Only worry about unprocessed files
 163         if not f.processed:
 164             utils.move(os.path.join(srcqueue.path, f.filename), destqueue.path, perms=int(destqueue.perms, 8))
 165
 166     utils.move(os.path.join(srcqueue.path, upload.pkg.changes_file), destqueue.path, perms=int(destqueue.perms, 8))
 167     chg.in_queue = destqueue
 168     session.commit()
 169
 170     return True
 171
 172 __all__.append('changes_to_queue')
 173
 174 def new_accept(upload, dry_run, session):
 175     print "ACCEPT"
 176
 177     if not dry_run:
 178         cnf = Config()
 179
 180         (summary, short_summary) = upload.build_summaries()
 181         destqueue = get_policy_queue('newstage', session)
 182
 183         srcqueue = get_policy_queue_from_path(upload.pkg.directory, session)
 184
 185         if not srcqueue:
 186             # Assume NEW and hope for the best
 187             srcqueue = get_policy_queue('new', session)
 188
 189         changes_to_queue(upload, srcqueue, destqueue, session)
 190
 191 __all__.append('new_accept')