dak/clean_suites.py

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