daklib/queue_install.py

   1 #!/usr/bin/env python
   2 # vim:set et sw=4:
   3
   4 """
   5 Utility functions for process-upload
   6
   7 @contact: Debian FTP Master <ftpmaster@debian.org>
   8 @copyright: 2000, 2001, 2002, 2003, 2004, 2005, 2006  James Troup <james@nocrew.org>
   9 @copyright: 2009  Joerg Jaspert <joerg@debian.org>
  10 @copyright: 2009  Mark Hymers <mhy@debian.org>
  11 @license: GNU General Public License version 2 or later
  12 """
  13
  14 # This program is free software; you can redistribute it and/or modify
  15 # it under the terms of the GNU General Public License as published by
  16 # the Free Software Foundation; either version 2 of the License, or
  17 # (at your option) any later version.
  18
  19 # This program is distributed in the hope that it will be useful,
  20 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  21 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22 # GNU General Public License for more details.
  23
  24 # You should have received a copy of the GNU General Public License
  25 # along with this program; if not, write to the Free Software
  26 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  27
  28 import os
  29
  30 from daklib import utils
  31 from daklib.dbconn import *
  32 from daklib.config import Config
  33
  34 ################################################################################
  35
  36 def package_to_suite(u, suite_name, session):
  37     if not u.pkg.changes["distribution"].has_key(suite_name):
  38         return False
  39
  40     ret = True
  41
  42     if not u.pkg.changes["architecture"].has_key("source"):
  43         q = session.query(SrcAssociation.sa_id)
  44         q = q.join(Suite).filter_by(suite_name=suite_name)
  45         q = q.join(DBSource).filter_by(source=u.pkg.changes['source'])
  46         q = q.filter_by(version=u.pkg.changes['version']).limit(1)
  47
  48         # NB: Careful, this logic isn't what you would think it is
  49         # Source is already in the target suite so no need to go to policy
  50         # Instead, we don't move to the policy area, we just do an ACCEPT
  51         if q.count() > 0:
  52             ret = False
  53
  54     return ret
  55
  56 def package_to_queue(u, summary, short_summary, queue, chg, session, announce=None):
  57     cnf = Config()
  58     dir = queue.path
  59
  60     print "Moving to %s policy queue" % queue.queue_name.upper()
  61     u.logger.log(["Moving to %s" % queue.queue_name, u.pkg.changes_file])
  62
  63     u.move_to_queue(queue)
  64     chg.in_queue_id = queue.policy_queue_id
  65     session.add(chg)
  66     session.commit()
  67
  68     # Check for override disparities
  69     u.check_override()
  70
  71     # Send accept mail, announce to lists and close bugs
  72     if announce:
  73         template = os.path.join(cnf["Dir::Templates"], announce)
  74         u.update_subst()
  75         mail_message = utils.TemplateSubst(u.Subst, template)
  76         utils.send_mail(mail_message)
  77         u.announce(short_summary, True)
  78
  79 ################################################################################
  80
  81 def is_unembargo(u):
  82    session = DBConn().session()
  83    cnf = Config()
  84
  85    # If we dont have the disembargo queue we are not on security and so not interested
  86    # in doing any security queue handling
  87    if not get_policy_queue("disembargo"):
  88        return False
  89
  90    q = session.execute("SELECT package FROM disembargo WHERE package = :source AND version = :version",
  91                        {'source': u.pkg.changes["source"],
  92                         'version': u.pkg.changes["version"]})
  93    if q.rowcount > 0:
  94        session.close()
  95        return True
  96
  97    oldcwd = os.getcwd()
  98    os.chdir(cnf["Dir::Queue::Disembargo"])
  99    disdir = os.getcwd()
 100    os.chdir(oldcwd)
 101
 102    ret = False
 103
 104    if u.pkg.directory == disdir:
 105        if u.pkg.changes["architecture"].has_key("source"):
 106            session.execute("INSERT INTO disembargo (package, version) VALUES (:package, :version)",
 107                            {'source': u.pkg.changes["source"],
 108                             'version': u.pkg.changes["version"]})
 109            session.commit()
 110
 111            ret = True
 112
 113    session.close()
 114
 115    return ret
 116
 117 def queue_unembargo(u, summary, short_summary, session=None):
 118     return package_to_queue(u, summary, short_summary,
 119                             get_policy_queue('disembargo'), chg, session,
 120                             announce=None)
 121 #
 122 #################################################################################
 123 #
 124 def is_embargo(u):
 125    # if we are the security archive, we always have a embargo queue and its the
 126    # last in line, so if that exists, return true
 127    if get_policy_queue('embargo'):
 128        return True
 129
 130 def queue_embargo(u, summary, short_summary, session=None):
 131     return package_to_queue(u, summary, short_summary,
 132                             get_policy_queue('embargo'), chg, session,
 133                             announce=None)
 134
 135 ################################################################################
 136
 137 def is_autobyhand(u):
 138     cnf = Config()
 139
 140     all_auto = 1
 141     any_auto = 0
 142     for f in u.pkg.files.keys():
 143         if u.pkg.files[f].has_key("byhand"):
 144             any_auto = 1
 145
 146             # filename is of form "PKG_VER_ARCH.EXT" where PKG, VER and ARCH
 147             # don't contain underscores, and ARCH doesn't contain dots.
 148             # further VER matches the .changes Version:, and ARCH should be in
 149             # the .changes Architecture: list.
 150             if f.count("_") < 2:
 151                 all_auto = 0
 152                 continue
 153
 154             (pckg, ver, archext) = f.split("_", 2)
 155             if archext.count(".") < 1 or u.pkg.changes["version"] != ver:
 156                 all_auto = 0
 157                 continue
 158
 159             ABH = cnf.SubTree("AutomaticByHandPackages")
 160             if not ABH.has_key(pckg) or \
 161               ABH["%s::Source" % (pckg)] != u.pkg.changes["source"]:
 162                 print "not match %s %s" % (pckg, u.pkg.changes["source"])
 163                 all_auto = 0
 164                 continue
 165
 166             (arch, ext) = archext.split(".", 1)
 167             if arch not in u.pkg.changes["architecture"]:
 168                 all_auto = 0
 169                 continue
 170
 171             u.pkg.files[f]["byhand-arch"] = arch
 172             u.pkg.files[f]["byhand-script"] = ABH["%s::Script" % (pckg)]
 173
 174     return any_auto and all_auto
 175
 176 def do_autobyhand(u, summary, short_summary, chg, session):
 177     print "Attempting AUTOBYHAND."
 178     byhandleft = False
 179     for f, entry in u.pkg.files.items():
 180         byhandfile = f
 181
 182         if not entry.has_key("byhand"):
 183             continue
 184
 185         if not entry.has_key("byhand-script"):
 186             byhandleft = True
 187             continue
 188
 189         os.system("ls -l %s" % byhandfile)
 190
 191         result = os.system("%s %s %s %s %s" % (
 192                 entry["byhand-script"],
 193                 byhandfile,
 194                 u.pkg.changes["version"],
 195                 entry["byhand-arch"],
 196                 os.path.abspath(u.pkg.changes_file)))
 197
 198         if result == 0:
 199             os.unlink(byhandfile)
 200             del u.pkg.files[f]
 201         else:
 202             print "Error processing %s, left as byhand." % (f)
 203             byhandleft = True
 204
 205     if byhandleft:
 206         do_byhand(u, summary, short_summary, chg, session)
 207     else:
 208         u.accept(summary, short_summary, session)
 209         u.check_override()
 210
 211 ################################################################################
 212
 213 def is_byhand(u):
 214     for f in u.pkg.files.keys():
 215         if u.pkg.files[f].has_key("byhand"):
 216             return True
 217     return False
 218
 219 def do_byhand(u, summary, short_summary, chg, session):
 220     return package_to_queue(u, summary, short_summary,
 221                             get_policy_queue('byhand'), chg, session,
 222                             announce=None)
 223
 224 ################################################################################
 225
 226 def is_new(u):
 227     for f in u.pkg.files.keys():
 228         if u.pkg.files[f].has_key("new"):
 229             return True
 230     return False
 231
 232 def acknowledge_new(u, summary, short_summary, chg, session):
 233     cnf = Config()
 234
 235     print "Moving to NEW queue."
 236     u.logger.log(["Moving to new", u.pkg.changes_file])
 237
 238     q = get_policy_queue('new', session)
 239
 240     u.move_to_queue(q)
 241     chg.in_queue_id = q.policy_queue_id
 242     session.add(chg)
 243     session.commit()
 244
 245     print "Sending new ack."
 246     template = os.path.join(cnf["Dir::Templates"], 'process-unchecked.new')
 247     u.update_subst()
 248     u.Subst["__SUMMARY__"] = summary
 249     new_ack_message = utils.TemplateSubst(u.Subst, template)
 250     utils.send_mail(new_ack_message)
 251
 252 ################################################################################
 253
 254 # q-unapproved hax0ring
 255 QueueInfo = {
 256     "new": { "is": is_new, "process": acknowledge_new },
 257     "autobyhand" : { "is" : is_autobyhand, "process": do_autobyhand },
 258     "byhand" : { "is": is_byhand, "process": do_byhand },
 259     "embargoed" : { "is": is_embargo, "process": do_embargo },
 260     "unembargoed" : { "is": is_unembargo, "process": do_unembargo },
 261 }
 262
 263 def determine_target(u):
 264     cnf = Config()
 265
 266     # Statically handled queues
 267     target = None
 268
 269     for q in ["autobyhand", "byhand", "new", "unembargoed", "embargoed"]:
 270         if QueueInfo[q]["is"](u):
 271             target = q
 272             break
 273
 274     return target
 275
 276 ###############################################################################
 277