3 """ Manipulate suite tags """
4 # Copyright (C) 2000, 2001, 2002, 2003, 2004, 2005, 2006 James Troup <james@nocrew.org>
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 #######################################################################################
22 # 8to6Guy: "Wow, Bob, You look rough!"
24 # BTAF <.oO>: "You moron! This is what you get for staying up all night drinking vodka and salad dressing!"
25 # BTAF <.oO>: "This coffee I.V. drip is barely even keeping me awake! I need something with more kick! But what?"
26 # BTAF: "OMIGOD! I OVERDOSED ON HEROIN"
27 # CoWorker#n: "Give him air!!"
28 # CoWorker#n+1: "We need a syringe full of adrenaline!"
29 # CoWorker#n+2: "Stab him in the heart!"
31 # CoWorker#n+3: "Bob's been overdosing quite a bit lately..."
32 # CoWorker#n+4: "Third time this week."
34 # -- http://www.angryflower.com/8to6.gif
36 #######################################################################################
38 # Adds or removes packages from a suite. Takes the list of files
39 # either from stdin or as a command line argument. Special action
40 # "set", will reset the suite (!) and add all packages from scratch.
42 #######################################################################################
47 from daklib.config import Config
48 from daklib.dbconn import *
49 from daklib import daklog
50 from daklib import utils
52 #######################################################################################
56 ################################################################################
58 def usage (exit_code=0):
59 print """Usage: dak control-suite [OPTIONS] [FILE]
60 Display or alter the contents of a suite using FILE(s), or stdin.
62 -a, --add=SUITE add to SUITE
63 -h, --help show this help and exit
64 -l, --list=SUITE list the contents of SUITE
65 -r, --remove=SUITE remove from SUITE
66 -s, --set=SUITE set SUITE
67 -b, --britney generate changelog entry for britney runs"""
71 #######################################################################################
73 def get_id(package, version, architecture, session):
74 if architecture == "source":
75 q = session.execute("SELECT id FROM source WHERE source = :package AND version = :version",
76 {'package': package, 'version': version})
78 q = session.execute("""SELECT b.id FROM binaries b, architecture a
79 WHERE b.package = :package AND b.version = :version
80 AND (a.arch_string = :arch OR a.arch_string = 'all')
81 AND b.architecture = a.id""",
82 {'package': package, 'version': version, 'arch': architecture})
86 utils.warn("Couldn't find '%s_%s_%s'." % (package, version, architecture))
90 utils.warn("Found more than one match for '%s_%s_%s'." % (package, version, architecture))
95 #######################################################################################
97 def britney_changelog(packages, suite, session):
102 q = session.execute("""SELECT s.source, s.version, sa.id
103 FROM source s, src_associations sa
104 WHERE sa.suite = :suiteid
105 AND sa.source = s.id""", {'suiteid': suite.suite_id})
107 for p in q.fetchall():
109 for p in packages.keys():
115 for p in current.keys():
117 if apt_pkg.VersionCompare(current[p], old[p]) > 0:
118 new[p] = [current[p], old[p]]
120 new[p] = [current[p], 0]
122 query = "SELECT source, changelog FROM changelogs WHERE"
124 query += " source = '%s' AND version > '%s' AND version <= '%s'" \
125 % (p, new[p][1], new[p][0])
126 query += " AND architecture LIKE '%source%' AND distribution in \
127 ('unstable', 'experimental', 'testing-proposed-updates') OR"
128 query += " False ORDER BY source, version DESC"
129 q = session.execute(query)
132 brit = utils.open_file(Config()["Changelogs::Britney"], 'w')
135 if pu and pu != u[0]:
137 brit.write("%s\n" % u[1])
139 if q.rowcount: brit.write("\n\n\n")
141 for p in list(set(old.keys()).difference(current.keys())):
142 brit.write("REMOVED: %s %s\n" % (p, old[p]))
147 #######################################################################################
149 def set_suite(file, suite, session, britney=False):
150 suite_id = suite.suite_id
151 lines = file.readlines()
153 # Our session is already in a transaction
155 # Build up a dictionary of what is currently in the suite
157 q = session.execute("""SELECT b.package, b.version, a.arch_string, ba.id
158 FROM binaries b, bin_associations ba, architecture a
159 WHERE ba.suite = :suiteid
160 AND ba.bin = b.id AND b.architecture = a.id""", {'suiteid': suite_id})
161 for i in q.fetchall():
162 key = " ".join(i[:3])
165 q = session.execute("""SELECT s.source, s.version, sa.id
166 FROM source s, src_associations sa
167 WHERE sa.suite = :suiteid
168 AND sa.source = s.id""", {'suiteid': suite_id})
169 for i in q.fetchall():
170 key = " ".join(i[:2]) + " source"
173 # Build up a dictionary of what should be in the suite
176 split_line = line.strip().split()
177 if len(split_line) != 3:
178 utils.warn("'%s' does not break into 'package version architecture'." % (line[:-1]))
180 key = " ".join(split_line)
183 # Check to see which packages need removed and remove them
184 for key in current.keys():
185 if not desired.has_key(key):
186 (package, version, architecture) = key.split()
188 if architecture == "source":
189 session.execute("""DELETE FROM src_associations WHERE id = :pkid""", {'pkid': pkid})
191 session.execute("""DELETE FROM bin_associations WHERE id = :pkid""", {'pkid': pkid})
192 Logger.log(["removed", key, pkid])
194 # Check to see which packages need added and add them
195 for key in desired.keys():
196 if not current.has_key(key):
197 (package, version, architecture) = key.split()
198 pkid = get_id (package, version, architecture, session)
201 if architecture == "source":
202 session.execute("""INSERT INTO src_associations (suite, source)
203 VALUES (:suiteid, :pkid)""", {'suiteid': suite_id, 'pkid': pkid})
205 session.execute("""INSERT INTO bin_associations (suite, bin)
206 VALUES (:suiteid, :pkid)""", {'suiteid': suite_id, 'pkid': pkid})
207 Logger.log(["added", key, pkid])
212 britney_changelog(current, suite, session)
214 #######################################################################################
216 def process_file(file, suite, action, session, britney=False):
218 set_suite(file, suite, session, britney)
221 suite_id = suite.suite_id
223 lines = file.readlines()
225 # Our session is already in a transaction
227 split_line = line.strip().split()
228 if len(split_line) != 3:
229 utils.warn("'%s' does not break into 'package version architecture'." % (line[:-1]))
232 (package, version, architecture) = split_line
234 pkid = get_id(package, version, architecture, session)
238 if architecture == "source":
239 # Find the existing association ID, if any
240 q = session.execute("""SELECT id FROM src_associations
241 WHERE suite = :suiteid and source = :pkid""",
242 {'suiteid': suite_id, 'pkid': pkid})
245 association_id = None
247 association_id = ql[0][0]
252 utils.warn("'%s_%s_%s' already exists in suite %s." % (package, version, architecture, suite))
255 session.execute("""INSERT INTO src_associations (suite, source)
256 VALUES (:suiteid, :pkid)""",
257 {'suiteid': suite_id, 'pkid': pkid})
258 elif action == "remove":
259 if association_id == None:
260 utils.warn("'%s_%s_%s' doesn't exist in suite %s." % (package, version, architecture, suite))
263 session.execute("""DELETE FROM src_associations WHERE id = :pkid""", {'pkid': association_id})
265 # Find the existing associations ID, if any
266 q = session.execute("""SELECT id FROM bin_associations
267 WHERE suite = :suiteid and bin = :pkid""",
268 {'suiteid': suite_id, 'pkid': pkid})
271 association_id = None
273 association_id = ql[0][0]
278 utils.warn("'%s_%s_%s' already exists in suite %s." % (package, version, architecture, suite))
281 session.execute("""INSERT INTO bin_associations (suite, bin)
282 VALUES (:suiteid, :pkid)""",
283 {'suiteid': suite_id, 'pkid': pkid})
284 elif action == "remove":
285 if association_id == None:
286 utils.warn("'%s_%s_%s' doesn't exist in suite %s." % (package, version, architecture, suite))
289 session.execute("""DELETE FROM bin_associations WHERE id = :pkid""", {'pkid': association_id})
293 #######################################################################################
295 def get_list(suite, session):
296 suite_id = suite.suite_id
298 q = session.execute("""SELECT b.package, b.version, a.arch_string
299 FROM binaries b, bin_associations ba, architecture a
300 WHERE ba.suite = :suiteid
301 AND ba.bin = b.id AND b.architecture = a.id""", {'suiteid': suite_id})
302 for i in q.fetchall():
306 q = session.execute("""SELECT s.source, s.version
307 FROM source s, src_associations sa
308 WHERE sa.suite = :suiteid
309 AND sa.source = s.id""", {'suiteid': suite_id})
310 for i in q.fetchall():
311 print " ".join(i) + " source"
313 #######################################################################################
320 Arguments = [('a',"add","Control-Suite::Options::Add", "HasArg"),
321 ('b',"britney","Control-Suite::Options::Britney"),
322 ('h',"help","Control-Suite::Options::Help"),
323 ('l',"list","Control-Suite::Options::List","HasArg"),
324 ('r',"remove", "Control-Suite::Options::Remove", "HasArg"),
325 ('s',"set", "Control-Suite::Options::Set", "HasArg")]
327 for i in ["add", "britney", "help", "list", "remove", "set", "version" ]:
328 if not cnf.has_key("Control-Suite::Options::%s" % (i)):
329 cnf["Control-Suite::Options::%s" % (i)] = ""
332 file_list = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv);
333 except SystemError, e:
336 Options = cnf.SubTree("Control-Suite::Options")
341 session = DBConn().session()
345 for i in ("add", "list", "remove", "set"):
346 if cnf["Control-Suite::Options::%s" % (i)] != "":
347 suite_name = cnf["Control-Suite::Options::%s" % (i)]
348 suite = get_suite(suite_name, session=session)
350 utils.fubar("Unknown suite '%s'." % (suite_name))
353 utils.fubar("Can only perform one action at a time.")
358 utils.fubar("No action specified.")
360 # Safety/Sanity check
361 # XXX: This should be stored in the database
362 if action == "set" and suite_name not in ["testing"]:
363 utils.fubar("Will not reset suite %s" % (suite_name))
366 if action == "set" and cnf["Control-Suite::Options::Britney"]:
370 get_list(suite, session)
372 Logger = daklog.Logger(cnf.Cnf, "control-suite")
375 process_file(utils.open_file(f), suite, action, session, britney)
377 process_file(sys.stdin, suite, action, session, britney)
380 #######################################################################################
382 if __name__ == '__main__':