3 # Manipulate suite tags
4 # Copyright (C) 2000, 2001 James Troup <james@nocrew.org>
5 # $Id: heidi,v 1.9 2001-09-27 01:22:51 troup Exp $
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 2 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 #######################################################################################
23 # 8to6Guy: "Wow, Bob, You look rough!"
25 # BTAF <.oO>: "You moron! This is what you get for staying up all night drinking vodka and salad dressing!"
26 # BTAF <.oO>: "This coffee I.V. drip is barely even keeping me awake! I need something with more kick! But what?"
27 # BTAF: "OMIGOD! I OVERDOSED ON HEROIN"
28 # CoWorker#n: "Give him air!!"
29 # CoWorker#n+1: "We need a syringe full of adrenaline!"
30 # CoWorker#n+2: "Stab him in the heart!"
32 # CoWorker#n+3: "Bob's been overdosing quite a bit lately..."
33 # CoWorker#n+4: "Third time this week."
35 # -- http://www.angryflower.com/8to6.gif
37 #######################################################################################
39 # Adds or removes packages from a suite. Takes the list of files
40 # either from stdin or as a command line argument. Special action
41 # "set", will reset the suite (!) and add all packages from scratch.
43 #######################################################################################
45 import os, pg, string, sys;
47 import utils, db_access, logging;
49 #######################################################################################
55 ################################################################################
57 def usage (exit_code=0):
58 print """Usage: heidi [OPTIONS] [FILE]
59 Display or alter the contents of a suite using FILE(s), or stdin.
61 -a, --add=SUITE add to SUITE
62 -l, --list=SUITE list the contents of SUITE
63 -r, --remove=SUITE remove from SUITE
64 -s, --set=SUITE set SUITE
65 -h, --help show this help and exit"""
69 #######################################################################################
71 def get_id (package, version, architecture):
72 if architecture == "source":
73 q = projectB.query("SELECT id FROM source WHERE source = '%s' AND version = '%s'" % (package, version))
75 q = projectB.query("SELECT b.id FROM binaries b, architecture a WHERE b.package = '%s' AND b.version = '%s' AND (a.arch_string = '%s' OR a.arch_string = 'all') AND b.architecture = a.id" % (package, version, architecture))
79 utils.warn("Couldn't find '%s~%s~%s'." % (package, version, architecture));
82 utils.warn("Found more than one match for '%s~%s~%s'." % (package, version, architecture));
87 #######################################################################################
89 def set_suite (file, suite_id):
90 lines = file.readlines();
92 projectB.query("BEGIN WORK");
94 # Build up a dictionary of what is currently in the suite
96 q = projectB.query("SELECT b.package, b.version, a.arch_string, ba.id FROM binaries b, bin_associations ba, architecture a WHERE ba.suite = %s AND ba.bin = b.id AND b.architecture = a.id" % (suite_id));
99 key = string.join(i[:3]);
101 q = projectB.query("SELECT s.source, s.version, sa.id FROM source s, src_associations sa WHERE sa.suite = %s AND sa.source = s.id" % (suite_id));
104 key = string.join(i[:2]) + " source";
107 # Build up a dictionary of what should be in the suite
110 split_line = string.split(string.strip(line[:-1]));
111 if len(split_line) != 3:
112 utils.warn("'%s' does not break into 'package version architecture'." % (line[:-1]));
114 key = string.join(split_line);
117 # Check to see which packages need removed and remove them
118 for key in current.keys():
119 if not desired.has_key(key):
120 (package, version, architecture) = string.split(key);
122 if architecture == "source":
123 q = projectB.query("DELETE FROM src_associations WHERE id = %s" % (id));
125 q = projectB.query("DELETE FROM bin_associations WHERE id = %s" % (id));
126 Logger.log(["removed",key,id]);
128 # Check to see which packages need added and add them
129 for key in desired.keys():
130 if not current.has_key(key):
131 (package, version, architecture) = string.split(key);
132 id = get_id (package, version, architecture);
135 if architecture == "source":
136 q = projectB.query("INSERT INTO src_associations (suite, source) VALUES (%s, %s)" % (suite_id, id));
138 q = projectB.query("INSERT INTO bin_associations (suite, bin) VALUES (%s, %s)" % (suite_id, id));
139 Logger.log(["added",key,id]);
141 projectB.query("COMMIT WORK");
143 #######################################################################################
145 def process_file (file, suite_id, action):
148 set_suite (file, suite_id);
151 lines = file.readlines();
153 projectB.query("BEGIN WORK");
156 split_line = string.split(string.strip(line[:-1]));
157 if len(split_line) != 3:
158 utils.warn("'%s' does not break into 'package version architecture'." % (line[:-1]));
161 (package, version, architecture) = split_line;
163 id = get_id(package, version, architecture);
167 if architecture == "source":
168 # Find the existing assoications ID, if any
169 q = projectB.query("SELECT id FROM src_associations WHERE suite = %s and source = %s" % (suite_id, id));
172 assoication_id = None;
174 assoication_id = ql[0][0];
177 if assoication_id != None:
178 utils.warn("'%s~%s~%s' already exists in suite %s." % (package, version, architecture, suite_id));
181 q = projectB.query("INSERT INTO src_associations (suite, source) VALUES (%s, %s)" % (suite_id, id));
182 elif action == "remove":
183 if assoication_id == None:
184 utils.warn("'%s~%s~%s' doesn't exist in suite %s." % (package, version, architecture, suite_id));
187 q = projectB.query("DELETE FROM src_associations WHERE id = %s" % (assoication_id));
189 # Find the existing assoications ID, if any
190 q = projectB.query("SELECT id FROM bin_associations WHERE suite = %s and bin = %s" % (suite_id, id));
193 assoication_id = None;
195 assoication_id = ql[0][0];
198 if assoication_id != None:
199 utils.warn("'%s~%s~%s' already exists in suite %s." % (package, version, architecture, suite_id));
202 q = projectB.query("INSERT INTO bin_associations (suite, bin) VALUES (%s, %s)" % (suite_id, id));
203 elif action == "remove":
204 if assoication_id == None:
205 utils.warn("'%s~%s~%s' doesn't exist in suite %s." % (package, version, architecture, suite_id));
208 q = projectB.query("DELETE FROM bin_associations WHERE id = %s" % (assoication_id));
210 projectB.query("COMMIT WORK");
212 #######################################################################################
214 def get_list (suite_id):
216 q = projectB.query("SELECT b.package, b.version, a.arch_string FROM binaries b, bin_associations ba, architecture a WHERE ba.suite = %s AND ba.bin = b.id AND b.architecture = a.id" % (suite_id));
219 print string.join(i);
222 q = projectB.query("SELECT s.source, s.version FROM source s, src_associations sa WHERE sa.suite = %s AND sa.source = s.id" % (suite_id));
225 print string.join(i) + " source";
227 #######################################################################################
230 global Cnf, projectB, Logger;
234 Cnf = apt_pkg.newConfiguration();
235 apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file());
237 Arguments = [('a',"add","Heidi::Options::Add", "HasArg"),
238 ('h',"help","Heidi::Options::Help"),
239 ('l',"list","Heidi::Options::List","HasArg"),
240 ('r',"remove", "Heidi::Options::Remove", "HasArg"),
241 ('s',"set", "Heidi::Options::Set", "HasArg")];
243 for i in ["add", "help", "list", "remove", "set", "version" ]:
244 Cnf["Heidi::Options::%s" % (i)] = "";
246 file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
247 Options = Cnf.SubTree("Heidi::Options")
252 projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"],int(Cnf["DB::Port"]));
254 db_access.init(Cnf, projectB);
258 for i in ("add", "list", "remove", "set"):
259 suite = Cnf["Heidi::Options::%s" % (i)];
261 if not Cnf.has_key("Suite::%s" % (suite)):
262 utils.fubar("Unknown suite '%s'." %(suite));
264 suite_id = db_access.get_suite_id(suite);
266 utils.fubar("Can only perform one action at a time.");
271 utils.fubar("No action specified.");
273 # Safety/Sanity check
274 if action == "set" and suite != "testing":
275 utils.fubar("Will not reset a suite other than testing.");
280 Logger = logging.Logger(Cnf, "heidi");
282 for file in file_list:
283 process_file(utils.open_file(file,'r'), suite_id, action);
285 process_file(sys.stdin, suite_id, action);
288 #######################################################################################
290 if __name__ == '__main__':