X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=rhona;h=fe887747cd9e1cfbe711e831639cb5b8295312ac;hb=e4f2f33a24646716adedd63afa87a672cd6ed599;hp=7b0d931037ffa5288b257e15b9da665945f4fb8a;hpb=ba18f507ebb85f05a410e94c76cbc13add116d0e;p=dak.git diff --git a/rhona b/rhona index 7b0d9310..fe887747 100755 --- a/rhona +++ b/rhona @@ -1,8 +1,8 @@ #!/usr/bin/env python -# rhona, cleans up unassociated binary (and source) packages +# rhona, cleans up unassociated binary and source packages # Copyright (C) 2000 James Troup -# $Id: rhona,v 1.2 2000-11-24 04:04:23 troup Exp $ +# $Id: rhona,v 1.7 2001-01-10 06:08:03 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 @@ -18,6 +18,8 @@ # 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 @@ -25,20 +27,64 @@ # # 07:05| elmo: Augh! -import pg, string, os, sys, time +################################################################################################### + +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"]; + try: + file = utils.open_file(filename, 'r'); + except utils.cant_open_exc: + pass; + else: + 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(); @@ -54,8 +100,7 @@ def check_binaries(): projectB.query("BEGIN WORK"); for id in all_binaries.keys(): if not linked_binaries.has_key(id): - date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-(4*(24*60*60)))) - projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (date, all_binaries[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. @@ -65,6 +110,7 @@ def check_binaries(): 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): @@ -74,8 +120,12 @@ def check_binaries(): 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 - # commeint in process_binary. + # comment in check_binaries(). linked_sources = {}; q = projectB.query("SELECT source FROM binaries WHERE source is not null"); @@ -84,24 +134,51 @@ def check_sources(): linked_sources[i[0]] = ""; all_sources = {}; - q = projectB.query("SELECT s.id, s.file FROM source s, files f WHERE f.last_used IS NULL AND f.id = s.file") + 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): - date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-(4*(24*60*60)))) - projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (date, all_sources[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)); - ql = q.getresult(); - for i in ql: - q_others = projectB.query("SELECT file FROM dsc_files d WHERE file = %s" % (i[0])); - ql_others = q.getresult(); + 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" % (date, i[0])); + 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. @@ -144,23 +221,126 @@ def check_sources(): 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(): - date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-int(Cnf["Rhona::StayOfExecution"]))); - q = projectB.query("SELECT l.path, f.filename FROM location l, files f WHERE f.last_used < '%s' AND l.id = f.location" % (date)) - ql = q.getresult(); - for i in ql: + 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]; - dest = Cnf["Rhona::Morgue"]+os.path.basename(filename); if not os.path.exists(filename): sys.stderr.write("E: can not find %s.\n" % (filename)); continue; - print "Cleaning %s to %s..." % (filename, dest) - #utils.move(filename, dest); - #projectB.query("DELETE FROM binaries WHERE id = %s" % (i[0])); - #FIXME: need to remove from "or source" + files + dsc_files.. etc. + 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; + global Cnf, projectB, delete_date; projectB = pg.connect('projectb', 'localhost'); @@ -169,12 +349,31 @@ def main(): Cnf = apt_pkg.newConfiguration(); apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file()); - print "Checking for orphaned binary packages..." + 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); + + override_filename = Cnf["Dir::OverrideDir"] + Cnf["Rhona::OverrideFilename"]; + if not os.access(override_filename, os.R_OK): + sys.stderr.write("W: Could not find source-only override file '%s'.\n" % (override_filename)); + + delete_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time())); + check_binaries(); - print "Checking for orphaned source packages..." + clean_binaries(); check_sources(); - print "Cleaning orphaned packages..." + check_files(); clean(); + clean_maintainers(); if __name__ == '__main__': main()