#!/usr/bin/env python # rhona, cleans up unassociated binary and source packages # Copyright (C) 2000 James Troup # $Id: rhona,v 1.6 2000-12-20 08:15:35 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 ################################################################################################### # 07:05| well.. *shrug*.. no, probably not.. but to fix it, # | we're going to have to implement reference counting # | through dependencies.. do we really want to go down # | that road? # # 07:05| elmo: Augh! ################################################################################################### import os, pg, stat, string, sys, time import apt_pkg import utils ################################################################################################### projectB = None Cnf = None delete_date = None; overrides = {}; ################################################################################################### def usage (exit_code): print """Usage: rhona [OPTION]... [CHANGES]... -D, --debug=VALUE debug -n, --no-action don't do anything -v, --verbose be verbose -V, --version display version number and exit""" sys.exit(exit_code) ################################################################################################### # See if a given package is in the override file. Caches and only loads override files on demand. def in_override_p (package): global overrides; if overrides == {}: filename = Cnf["Dir::OverrideDir"] + Cnf["Rhona::OverrideFilename"]; file = utils.open_file(filename, 'r'); for line in file.readlines(): line = string.strip(utils.re_comments.sub('', line)) if line != "": overrides[line] = 1 file.close() return overrides.get(package, None); ################################################################################################### def check_binaries(): global delete_date; print "Checking for orphaned binary packages..." # A nicer way to do this would be `SELECT bin FROM # bin_associations EXCEPT SELECT id from binaries WHERE # last_update IS NULL', but it seems postgresql can't handle that # query as it hadn't return after I left it running for 20 minutes # on auric. linked_binaries = {}; q = projectB.query("SELECT bin FROM bin_associations"); ql = q.getresult(); for i in ql: linked_binaries[i[0]] = ""; all_binaries = {}; q = projectB.query("SELECT b.id, b.file FROM binaries b, files f WHERE f.last_used IS NULL AND f.id = b.file") ql = q.getresult(); for i in ql: all_binaries[i[0]] = i[1]; projectB.query("BEGIN WORK"); for id in all_binaries.keys(): if not linked_binaries.has_key(id): projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (delete_date, all_binaries[id])) projectB.query("COMMIT WORK"); # Check for any binaries which are marked for eventual deletion but are now used again. all_marked_binaries = {}; q = projectB.query("SELECT b.id, b.file FROM binaries b, files f WHERE f.last_used IS NOT NULL AND f.id = b.file") ql = q.getresult(); for i in ql: all_marked_binaries[i[0]] = i[1]; projectB.query("BEGIN WORK"); for id in all_marked_binaries.keys(): if linked_binaries.has_key(id): # Can't imagine why this would happen, so warn about it for now. print "W: %s has released %s from the target list." % (id, all_marked_binaries[id]); projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (all_marked_binaries[id])); projectB.query("COMMIT WORK"); def check_sources(): global delete_date; print "Checking for orphaned source packages..." # A nicer way to do this would be using `EXCEPT', but see the # comment in check_binaries(). linked_sources = {}; q = projectB.query("SELECT source FROM binaries WHERE source is not null"); ql = q.getresult(); for i in ql: linked_sources[i[0]] = ""; all_sources = {}; all_sources_package = {}; q = projectB.query("SELECT s.id, s.file, s.source FROM source s, files f WHERE f.last_used IS NULL AND f.id = s.file"); ql = q.getresult(); for i in ql: all_sources[i[0]] = i[1]; all_sources_package[i[0]] = i[2]; projectB.query("BEGIN WORK"); for id in all_sources.keys(): if not linked_sources.has_key(id): # Is this a known source-only package? if in_override_p(all_sources_package[id]): continue; # Then check to see if the source is still in any suites untouchable = 0; q = projectB.query("SELECT su.suite_name FROM src_associations sa, suite su WHERE sa.source = %s and su.id = sa.suite" % (id)); for i in q.getresult(): if Cnf.Find("Suite::%s::Untouchable" % (i[0])): untouchable = 1; else: if not Cnf["Rhona::Options::No-Action"]: projectB.query("DELETE FROM src_associations WHERE source = %s" % (id)); # We can't delete binary-less source-only packages if # they're in an untouchable suite (i.e. stable)... if untouchable: continue; projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (delete_date, all_sources[id])) # Delete all other files references by .dsc too if they're not used by anyone else q = projectB.query("SELECT f.id FROM files f, dsc_files d WHERE d.source = %d AND d.file = f.id" % (id)); for i in q.getresult(): q = projectB.query("SELECT id FROM dsc_files d WHERE file = %s" % (i[0])); ql = q.getresult(); if len(ql) == 1: projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (delete_date, i[0])); # If the file is used by another source package # (e.g. because it's an .orig.tar.gz) We need to delete # this source package's reference to it from dsc_files. # So just clear out all references to the source file in # dsc_files now. if not Cnf["Rhona::Options::No-Action"]: projectB.query("DELETE FROM dsc_files WHERE source = %s" % (id)); projectB.query("COMMIT WORK"); # Check for any sources which are marked for eventual deletion but are now used again. all_marked_sources = {}; q = projectB.query("SELECT s.id, s.file FROM source s, files f WHERE f.last_used IS NOT NULL AND f.id = s.file"); ql = q.getresult(); for i in ql: all_marked_sources[i[0]] = i[1]; projectB.query("BEGIN WORK"); for id in all_marked_sources.keys(): if linked_sources.has_key(id): # Can't imagine why this would happen, so warn about it for now. print "W: %s has released %s from the target list." % (id, all_marked_sources[id]); projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (all_marked_sources[id])); # Unmark all other files references by .dsc too q = projectB.query("SELECT file FROM dsc_files WHERE source = %d" % (id)); ql = q.getresult(); for i in ql: projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (i[0])); projectB.query("COMMIT WORK"); # Whee, check for any source files (i.e. non .dsc) which are # marked for eventual deletion but are now used by a source # package again. all_marked_dsc_files = {} q = projectB.query("SELECT s.id, s.file FROM source s, files f, dsc_files d WHERE f.last_used IS NOT NULL AND f.id = d.file AND d.source = s.id"); ql = q.getresult(); for i in ql: all_marked_dsc_files[i[0]] = i[1]; projectB.query("BEGIN WORK"); for id in all_marked_dsc_files.keys(): if linked_sources.has_key(id): # Can't imagine why this would happen, so warn about it for now. print "W: %s has released %s from the target list." % (id, all_marked_sources[id]); projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (all_marked_sources[id])); # Unmark all other files references by .dsc too q = projectB.query("SELECT file FROM dsc_files WHERE source = %d" % (id)); ql = q.getresult(); for i in ql: projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (i[0])); projectB.query("COMMIT WORK"); def check_files(): global delete_date; print "Checking for unused files..." # Check for files not references in either binaries or dsc_files used = {}; q = projectB.query("SELECT file FROM binaries"); for i in q.getresult(): used[i[0]] = ""; q = projectB.query("SELECT file FROM dsc_files"); for i in q.getresult(): used[i[0]] = ""; all = {}; q = projectB.query("SELECT f.id, l.path, f.filename FROM files f, location l WHERE f.location = l.id;"); for i in q.getresult(): all[i[0]] = i[1] + i[2]; projectB.query("BEGIN WORK"); for id in all.keys(): if not used.has_key(id): projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (delete_date, id)); projectB.query("COMMIT WORK"); def clean_binaries(): global delete_date; # We do this here so that the binaries we remove will have their # source also removed (if possible). print "Cleaning binaries from the DB..." if not Cnf["Rhona::Options::No-Action"]: before = time.time(); sys.stdout.write("[Deleting from binaries table... "); projectB.query("DELETE FROM binaries WHERE EXISTS (SELECT id FROM files WHERE binaries.file = files.id AND files.last_used <= '%s')" % (delete_date)); sys.stdout.write("done. (%d)]\n" % (int(time.time()-before))); def clean(): global delete_date; count = 0; size = 0; print "Cleaning out packages..." # Ensure destination directory exists dest = Cnf["Dir::Morgue"] + '/' + Cnf["Rhona::MorgueSubDir"]; if not os.path.exists(dest): os.mkdir(dest); # Delete from source (dsc_file should already be done!) if not Cnf["Rhona::Options::No-Action"]: before = time.time(); sys.stdout.write("[Deleting from source table... "); projectB.query("DELETE FROM source WHERE EXISTS (SELECT id FROM files WHERE source.file = files.id AND files.last_used <= '%s')" % (delete_date)); sys.stdout.write("done. (%d)]\n" % (int(time.time()-before))); # Delete files from the pool q = projectB.query("SELECT l.path, f.filename FROM location l, files f WHERE f.last_used <= '%s' AND l.id = f.location" % (delete_date)); for i in q.getresult(): filename = i[0] + i[1]; if not os.path.exists(filename): sys.stderr.write("E: can not find %s.\n" % (filename)); continue; if os.path.isfile(filename): if os.path.islink(filename): count = count + 1; if Cnf["Rhona::Options::No-Action"]: print "Removing symlink %s..." % (filename); else: os.unlink(filename); else: size = size + os.stat(filename)[stat.ST_SIZE]; count = count + 1; if Cnf["Rhona::Options::No-Action"]: print "Cleaning %s to %s..." % (filename, dest); else: utils.move(filename, dest); else: sys.stderr.write("%s is neither symlink nor file?!\n" % (filename)); sys.exit(1); # delete from files if not Cnf["Rhona::Options::No-Action"]: before = time.time(); sys.stdout.write("[Deleting from files table... "); projectB.query("DELETE FROM files WHERE last_used <= '%s'" % (delete_date)); sys.stdout.write("done. (%d)]\n" % (int(time.time()-before))); if count > 0: sys.stderr.write("Cleaned %d files, %s.\n" % (count, utils.size_type(size))); def clean_maintainers(): print "Cleaning out unused Maintainer entries..." used = {}; q = projectB.query("SELECT maintainer FROM binaries WHERE maintainer IS NOT NULL"); for i in q.getresult(): used[i[0]] = ""; q = projectB.query("SELECT maintainer FROM source WHERE maintainer IS NOT NULL"); for i in q.getresult(): used[i[0]] = ""; all = {}; q = projectB.query("SELECT id, name FROM maintainer"); for i in q.getresult(): all[i[0]] = i[1]; count = 0; projectB.query("BEGIN WORK"); for id in all.keys(): if not used.has_key(id): if not Cnf["Rhona::Options::No-Action"]: projectB.query("DELETE FROM maintainer WHERE id = %s" % (id)); count = count + 1; projectB.query("COMMIT WORK"); if count > 0: sys.stderr.write("Cleared out %d maintainer entries.\n" % (count)); def main(): global Cnf, projectB, delete_date; projectB = pg.connect('projectb', 'localhost'); apt_pkg.init(); Cnf = apt_pkg.newConfiguration(); apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file()); Arguments = [('D',"debug","Rhona::Options::Debug", "IntVal"), ('h',"help","Rhona::Options::Help"), ('n',"no-action","Rhona::Options::No-Action"), ('V',"version","Rhona::Options::Version")]; apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv); if Cnf["Rhona::Options::Help"]: usage(0); if Cnf["Rhona::Options::Version"]: print "rhona version 0.0000000000"; usage(0); delete_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time())); check_binaries(); clean_binaries(); check_sources(); check_files(); clean(); clean_maintainers(); if __name__ == '__main__': main()