rhona

   1 #!/usr/bin/env python
   2
   3 # rhona, cleans up unassociated binary and source packages
   4 # Copyright (C) 2000, 2001, 2002  James Troup <james@nocrew.org>
   5 # $Id: rhona,v 1.24 2002-05-23 12:18:32 troup Exp $
   6
   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.
  11
  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.
  16
  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
  20
  21 ################################################################################
  22
  23 # 07:05|<elmo> well.. *shrug*.. no, probably not.. but to fix it,
  24 #      |       we're going to have to implement reference counting
  25 #      |       through dependencies.. do we really want to go down
  26 #      |       that road?
  27 #
  28 # 07:05|<Culus> elmo: Augh! <brain jumps out of skull>
  29
  30 ################################################################################
  31
  32 import os, pg, stat, sys, time
  33 import apt_pkg
  34 import utils
  35
  36 ################################################################################
  37
  38 projectB = None;
  39 Cnf = None;
  40 Options = None;
  41 now_date = None;     # mark newly "deleted" things as deleted "now"
  42 delete_date = None;  # delete things marked "deleted" earler than this
  43
  44 ################################################################################
  45
  46 def usage (exit_code=0):
  47     print """Usage: rhona [OPTIONS]
  48 Clean old packages from suites.
  49
  50   -n, --no-action            don't do anything
  51   -h, --help                 show this help and exit"""
  52     sys.exit(exit_code)
  53
  54 ################################################################################
  55
  56 def check_binaries():
  57     global delete_date, now_date;
  58
  59     print "Checking for orphaned binary packages..."
  60
  61     # Get the list of binary packages not in a suite and mark them for
  62     # deletion.
  63
  64     q = projectB.query("""
  65 SELECT b.file FROM binaries b WHERE NOT EXISTS
  66         (SELECT ba.bin FROM bin_associations ba WHERE ba.bin = b.id)""");
  67     ql = q.getresult();
  68
  69     projectB.query("BEGIN WORK");
  70     for i in ql:
  71         file_id = i[0];
  72         projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s AND last_used IS NULL" % (now_date, file_id))
  73     projectB.query("COMMIT WORK");
  74
  75     # Check for any binaries which are marked for eventual deletion
  76     # but are now used again.
  77
  78     q = projectB.query("""
  79 SELECT b.file FROM binaries b, files f
  80    WHERE f.last_used IS NOT NULL AND f.id = b.file AND
  81       EXISTS (SELECT suite FROM bin_associations ba WHERE ba.bin = b.id)""");
  82     ql = q.getresult();
  83     projectB.query("BEGIN WORK");
  84     for i in ql:
  85         file_id = i[0];
  86         projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (file_id));
  87     projectB.query("COMMIT WORK");
  88
  89 def check_sources():
  90     global delete_date, now_date;
  91
  92     print "Checking for orphaned source packages..."
  93
  94     # Get the list of source packages not in a suite.
  95
  96     q = projectB.query("""
  97 SELECT s.id, s.file FROM source s
  98   WHERE NOT EXISTS
  99    (SELECT sa.suite FROM src_associations sa WHERE sa.source = s.id)
 100   AND NOT EXISTS (SELECT b.id FROM binaries b WHERE b.source = s.id)""");
 101
 102     #### XXX: this should ignore cases where the files for the binary b
 103     ####      have been marked for deletion (so the delay between bins go
 104     ####      byebye and sources go byebye is 0 instead of StayOfExecution)
 105
 106     ql = q.getresult();
 107
 108     projectB.query("BEGIN WORK");
 109     for i in ql:
 110         source_id = i[0];
 111         dsc_file_id = i[1];
 112
 113         # Mark the .dsc file for deletion
 114         projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s AND last_used IS NULL" % (now_date, dsc_file_id))
 115         # Mark all other files references by .dsc too if they're not used by anyone else
 116         x = projectB.query("SELECT f.id FROM files f, dsc_files d WHERE d.source = %s AND d.file = f.id" % (source_id));
 117         for j in x.getresult():
 118             file_id = j[0];
 119             y = projectB.query("SELECT id FROM dsc_files d WHERE d.file = %s" % (file_id));
 120             if len(y.getresult()) == 1:
 121                 projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s AND last_used IS NULL" % (now_date, file_id));
 122     projectB.query("COMMIT WORK");
 123
 124     # Check for any sources which are marked for deletion but which
 125     # are now used again.
 126
 127     q = projectB.query("""
 128 SELECT f.id FROM source s, files f, dsc_files df
 129   WHERE f.last_used IS NOT NULL AND s.id = df.source AND df.file = f.id
 130     AND ((EXISTS (SELECT sa.suite FROM src_associations sa WHERE sa.source = s.id))
 131       OR (EXISTS (SELECT b.id FROM binaries b WHERE b.source = s.id)))""");
 132
 133     #### XXX: this should also handle deleted binaries specially (ie, not
 134     ####      reinstate sources because of them
 135
 136     ql = q.getresult();
 137     # Could be done in SQL; but left this way for hysterical raisins
 138     # [and freedom to innovate don'cha know?]
 139     projectB.query("BEGIN WORK");
 140     for i in ql:
 141         file_id = i[0];
 142         projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (file_id));
 143     projectB.query("COMMIT WORK");
 144
 145 def check_files():
 146     global delete_date, now_date;
 147
 148     # FIXME: this is evil; nothing should ever be in this state.  if
 149     # they are, it's a bug and the files should not be auto-deleted.
 150
 151     return;
 152
 153     print "Checking for unused files..."
 154
 155     q = projectB.query("""
 156 SELECT id FROM files f
 157   WHERE NOT EXISTS (SELECT id FROM binaries b WHERE b.file = f.id)
 158     AND NOT EXISTS (SELECT id FROM dsc_files df WHERE df.file = f.id)""");
 159
 160     projectB.query("BEGIN WORK");
 161     for i in q.getresult():
 162         file_id = i[0];
 163         projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (now_date, file_id));
 164     projectB.query("COMMIT WORK");
 165
 166 def clean_binaries():
 167     global delete_date, now_date;
 168
 169     # We do this here so that the binaries we remove will have their
 170     # source also removed (if possible).
 171
 172     # XXX: why doesn't this remove the files here as well? I don't think it
 173     #      buys anything keeping this separate
 174     print "Cleaning binaries from the DB..."
 175     if not Options["No-Action"]:
 176         before = time.time();
 177         sys.stdout.write("[Deleting from binaries table... ");
 178         projectB.query("DELETE FROM binaries WHERE EXISTS (SELECT id FROM files WHERE binaries.file = files.id AND files.last_used <= '%s')" % (delete_date));
 179         sys.stdout.write("done. (%d seconds)]\n" % (int(time.time()-before)));
 180
 181 def clean():
 182     global delete_date, now_date;
 183     count = 0;
 184     size = 0;
 185
 186     print "Cleaning out packages..."
 187
 188     date = time.strftime("%Y-%m-%d", time.localtime(time.time()));
 189     dest = Cnf["Dir::Morgue"] + '/' + Cnf["Rhona::MorgueSubDir"] + '/' + date;
 190     if not os.path.exists(dest):
 191         os.mkdir(dest);
 192
 193     # Delete from source
 194     if not Options["No-Action"]:
 195         before = time.time();
 196         sys.stdout.write("[Deleting from source table... ");
 197         projectB.query("DELETE FROM dsc_files WHERE EXISTS (SELECT df.id FROM source s, files f, dsc_files df WHERE f.last_used <= '%s' AND s.file = f.id AND s.id = df.source AND df.id = dsc_files.id)" % (delete_date));
 198         projectB.query("DELETE FROM source WHERE EXISTS (SELECT id FROM files WHERE source.file = files.id AND files.last_used <= '%s')" % (delete_date));
 199         sys.stdout.write("done. (%d seconds)]\n" % (int(time.time()-before)));
 200
 201     # Delete files from the pool
 202     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));
 203     for i in q.getresult():
 204         filename = i[0] + i[1];
 205         if not os.path.exists(filename):
 206             utils.warn("can not find '%s'." % (filename));
 207             continue;
 208         if os.path.isfile(filename):
 209             if os.path.islink(filename):
 210                 count = count + 1;
 211                 if Options["No-Action"]:
 212                     print "Removing symlink %s..." % (filename);
 213                 else:
 214                     os.unlink(filename);
 215             else:
 216                 size = size + os.stat(filename)[stat.ST_SIZE];
 217                 count = count + 1;
 218
 219                 dest_filename = dest + '/' + os.path.basename(filename);
 220                 # If the destination file exists; try to find another filename to use
 221                 if os.path.exists(dest_filename):
 222                     dest_filename = utils.find_next_free(dest_filename);
 223
 224                 if Options["No-Action"]:
 225                     print "Cleaning %s -> %s ..." % (filename, dest_filename);
 226                 else:
 227                     utils.move(filename, dest_filename);
 228         else:
 229             utils.fubar("%s is neither symlink nor file?!" % (filename));
 230
 231     # Delete from the 'files' table
 232     if not Options["No-Action"]:
 233         before = time.time();
 234         sys.stdout.write("[Deleting from files table... ");
 235         projectB.query("DELETE FROM files WHERE last_used <= '%s'" % (delete_date));
 236         sys.stdout.write("done. (%d seconds)]\n" % (int(time.time()-before)));
 237     if count > 0:
 238         sys.stderr.write("Cleaned %d files, %s.\n" % (count, utils.size_type(size)));
 239
 240 ################################################################################
 241
 242 def clean_maintainers():
 243     print "Cleaning out unused Maintainer entries..."
 244
 245     q = projectB.query("""
 246 SELECT m.id FROM maintainer m
 247   WHERE NOT EXISTS (SELECT id FROM binaries b WHERE b.maintainer = m.id)
 248     AND NOT EXISTS (SELECT id FROM source s WHERE s.maintainer = m.id)""");
 249     ql = q.getresult();
 250
 251     count = 0;
 252     projectB.query("BEGIN WORK");
 253     for i in ql:
 254         maintainer_id = i[0];
 255         if not Options["No-Action"]:
 256             projectB.query("DELETE FROM maintainer WHERE id = %s" % (maintainer_id));
 257             count = count + 1;
 258     projectB.query("COMMIT WORK");
 259
 260     if count > 0:
 261         sys.stderr.write("Cleared out %d maintainer entries.\n" % (count));
 262
 263 ################################################################################
 264
 265 def clean_fingerprints():
 266     print "Cleaning out unused fingerprint entries..."
 267
 268     q = projectB.query("""
 269 SELECT f.id FROM fingerprint f
 270   WHERE NOT EXISTS (SELECT id FROM binaries b WHERE b.sig_fpr = f.id)
 271     AND NOT EXISTS (SELECT id FROM source s WHERE s.sig_fpr = f.id)""");
 272     ql = q.getresult();
 273
 274     count = 0;
 275     projectB.query("BEGIN WORK");
 276     for i in ql:
 277         fingerprint_id = i[0];
 278         if not Options["No-Action"]:
 279             projectB.query("DELETE FROM fingerprint WHERE id = %s" % (fingerprint_id));
 280             count = count + 1;
 281     projectB.query("COMMIT WORK");
 282
 283     if count > 0:
 284         sys.stderr.write("Cleared out %d fingerprint entries.\n" % (count));
 285
 286 ################################################################################
 287
 288 def clean_accepted_autobuild():
 289     global now_date;
 290
 291     if not Cnf.ValueList("Dinstall::AcceptedAutoBuildSuites") or Options["No-Action"]:
 292         return;
 293
 294     print "Cleaning out accepted autobuild symlinks..."
 295
 296     our_delete_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-int(Cnf["Rhona::AcceptedAutoBuildStayOfExecution"])));
 297     count = 0;
 298
 299     q = projectB.query("SELECT filename FROM accepted_autobuild WHERE last_used <= '%s'" % (our_delete_date));
 300     for i in q.getresult():
 301         filename = i[0];
 302         if not os.path.exists(filename):
 303             utils.warn("%s (from accepted_autobuild) doesn't exist." % (filename));
 304             continue;
 305         if not Cnf.FindB("Dinstall::SecurityAcceptedAutoBuild") and not os.path.islink(filename):
 306             utils.fubar("%s (from accepted_autobuild) should be a symlink but isn't." % (filename));
 307         os.unlink(filename);
 308         count = count + 1;
 309     projectB.query("DELETE FROM accepted_autobuild WHERE last_used <= '%s'" % (our_delete_date));
 310
 311     if count:
 312         sys.stderr.write("Cleaned %d accepted-autobuild files.\n" % (count));
 313
 314 ################################################################################
 315
 316 def main():
 317     global Cnf, Options, projectB, delete_date, now_date;
 318
 319     Cnf = utils.get_conf()
 320     for i in ["Help", "No-Action" ]:
 321         if not Cnf.has_key("Rhona::Options::%s" % (i)):
 322             Cnf["Rhona::Options::%s" % (i)] = "";
 323
 324     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
 325
 326     Arguments = [('h',"help","Rhona::Options::Help"),
 327                  ('n',"no-action","Rhona::Options::No-Action")];
 328
 329     apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
 330     Options = Cnf.SubTree("Rhona::Options")
 331
 332     if Options["Help"]:
 333         usage();
 334
 335     now_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()));
 336     delete_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-int(Cnf["Rhona::StayOfExecution"])));
 337
 338     check_binaries();
 339     clean_binaries();
 340     check_sources();
 341     check_files();
 342     clean();
 343     clean_maintainers();
 344     clean_fingerprints();
 345     clean_accepted_autobuild();
 346
 347 ################################################################################
 348
 349 if __name__ == '__main__':
 350     main()
 351