#!/usr/bin/env python # Manipulate suite tags # Copyright (C) 2000, 2001 James Troup # $Id: heidi,v 1.4 2001-03-02 02:24:33 troup Exp $ # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA ####################################################################################### # 8to6Guy: "Wow, Bob, You look rough!" # BTAF: "Mbblpmn..." # BTAF <.oO>: "You moron! This is what you get for staying up all night drinking vodka and salad dressing!" # BTAF <.oO>: "This coffee I.V. drip is barely even keeping me awake! I need something with more kick! But what?" # BTAF: "OMIGOD! I OVERDOSED ON HEROIN" # CoWorker#n: "Give him air!!" # CoWorker#n+1: "We need a syringe full of adrenaline!" # CoWorker#n+2: "Stab him in the heart!" # BTAF: "*YES!*" # CoWorker#n+3: "Bob's been overdosing quite a bit lately..." # CoWorker#n+4: "Third time this week." # -- http://www.angryflower.com/8to6.gif ####################################################################################### # Adds or removes packages from a suite. Takes the list of files # either from stdin or as a command line argument. Special action # "set", will reset the suite (!) and add all packages from scratch. ####################################################################################### import os, pg, string, sys; import apt_pkg; import utils, db_access; ####################################################################################### Cnf = None; projectB = None; ####################################################################################### def process_file (file, suite_id, action): if action == "set": projectB.query("DELETE FROM bin_associations WHERE suite = %s" % (suite_id)); projectB.query("DELETE FROM src_associations WHERE suite = %s" % (suite_id)); action = "add"; for line in file.readlines(): split_line = string.split(string.strip(line[:-1])); if len(split_line) != 3: sys.stderr.write("W: '%s' does not break into 'package version architecture'.\n" % (line[:-1])); continue; (package, version, architecture) = split_line; if architecture == "source": q = projectB.query("SELECT id FROM source WHERE source = '%s' AND version = '%s'" % (package, version)) else: 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)) ql = q.getresult(); if ql == []: sys.stderr.write("W: Couldn't find '%s~%s~%s'.\n" % (package, version, architecture)); continue; if len(ql) > 1: sys.stderr.write("E: Found more than one match for '%s~%s~%s'.\n" % (package, version, architecture)); continue; id = ql[0][0]; if architecture == "source": # Find the existing assoications ID, if any q = projectB.query("SELECT id FROM src_associations WHERE suite = %s and source = %s" % (suite_id, id)); ql = q.getresult(); if ql == []: assoication_id = None; else: assoication_id = ql[0][0]; # Take action if action == "add": if assoication_id != None: sys.stderr.write("W: '%s~%s~%s' already exists in suite %s.\n" % (package, version, architecture, suite_id)); continue; else: q = projectB.query("INSERT INTO src_associations (suite, source) VALUES (%s, %s)" % (suite_id, id)); elif action == "remove": if assoication_id == None: sys.stderr.write("W: '%s~%s~%s' doesn't exist in suite %s.\n" % (package, version, architecture, suite_id)); continue; else: q = projectB.query("DELETE FROM src_associations WHERE id = %s" % (assoication_id)); else: # Find the existing assoications ID, if any q = projectB.query("SELECT id FROM bin_associations WHERE suite = %s and bin = %s" % (suite_id, id)); ql = q.getresult(); if ql == []: assoication_id = None; else: assoication_id = ql[0][0]; # Take action if action == "add": if assoication_id != None: sys.stderr.write("W: '%s~%s~%s' already exists in suite %s.\n" % (package, version, architecture, suite_id)); continue; else: q = projectB.query("INSERT INTO bin_associations (suite, bin) VALUES (%s, %s)" % (suite_id, id)); elif action == "remove": if assoication_id == None: sys.stderr.write("W: '%s~%s~%s' doesn't exist in suite %s.\n" % (package, version, architecture, suite_id)); continue; else: q = projectB.query("DELETE FROM bin_associations WHERE id = %s" % (assoication_id)); ####################################################################################### def get_list (suite_id): # List binaries 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)); ql = q.getresult(); for i in ql: print string.join(i); # List source 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)); ql = q.getresult(); for i in ql: print string.join(i) + " source"; ####################################################################################### def main (): global Cnf, projectB; apt_pkg.init(); Cnf = apt_pkg.newConfiguration(); apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file()); Arguments = [('a',"add","Heidi::Options::Add", "HasArg"), ('d',"debug","Heidi::Options::Debug", "IntVal"), ('h',"help","Heidi::Options::Help"), ('l',"list","Heidi::Options::List","HasArg"), ('r',"remove", "Heidi::Options::Remove", "HasArg"), ('s',"set", "Heidi::Options::Set", "HasArg"), ('v',"version","Heidi::Options::Version")]; file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv); projectB = pg.connect('projectb', 'localhost'); db_access.init(Cnf, projectB); action = None; for i in ("add", "list", "remove", "set"): suite = Cnf["Heidi::Options::%s" % (i)]; if suite !="": if not Cnf.has_key("Suite::%s" % (suite)): sys.stderr.write("Unknown suite %s.\n" %(suite)); sys.exit(2); else: suite_id = db_access.get_suite_id(suite); if action != None: sys.stderr.write("Can only do one action at a time.\n"); sys.exit(2); action = i; # Need an action... if action == None: sys.stderr.write("No action specified.\n"); sys.exit(2); # Safety/Sanity check if action == "set" and suite != "testing": sys.stderr.write("Will not reset a suite other than testing...\n"); sys.exit(2); if action == "list": get_list(suite_id); else: if file_list != []: for file in file_list: process_file(utils.open_file(file,'r'), suite_id, action); else: process_file(sys.stdin, suite_id, action); ####################################################################################### if __name__ == '__main__': main()