3 # rhona, cleans up unassociated binary and source packages
4 # Copyright (C) 2000, 2001, 2002, 2003 James Troup <james@nocrew.org>
5 # $Id: rhona,v 1.29 2005-11-25 06:59:45 ajt Exp $
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.
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.
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
21 ################################################################################
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
28 # 07:05|<Culus> elmo: Augh! <brain jumps out of skull>
30 ################################################################################
32 import os, pg, stat, sys, time
36 ################################################################################
41 now_date = None; # mark newly "deleted" things as deleted "now"
42 delete_date = None; # delete things marked "deleted" earler than this
44 ################################################################################
46 def usage (exit_code=0):
47 print """Usage: rhona [OPTIONS]
48 Clean old packages from suites.
50 -n, --no-action don't do anything
51 -h, --help show this help and exit"""
54 ################################################################################
57 global delete_date, now_date;
59 print "Checking for orphaned binary packages..."
61 # Get the list of binary packages not in a suite and mark them for
63 q = projectB.query("""
64 SELECT b.file FROM binaries b, files f
65 WHERE f.last_used IS NULL AND b.file = f.id
66 AND NOT EXISTS (SELECT 1 FROM bin_associations ba WHERE ba.bin = b.id)""");
69 projectB.query("BEGIN WORK");
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");
75 # Check for any binaries which are marked for eventual deletion
76 # but are now used again.
77 q = projectB.query("""
78 SELECT b.file FROM binaries b, files f
79 WHERE f.last_used IS NOT NULL AND f.id = b.file
80 AND EXISTS (SELECT 1 FROM bin_associations ba WHERE ba.bin = b.id)""");
83 projectB.query("BEGIN WORK");
86 projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (file_id));
87 projectB.query("COMMIT WORK");
89 ########################################
92 global delete_date, now_date;
94 print "Checking for orphaned source packages..."
96 # Get the list of source packages not in a suite and not used by
98 q = projectB.query("""
99 SELECT s.id, s.file FROM source s, files f
100 WHERE f.last_used IS NULL AND s.file = f.id
101 AND NOT EXISTS (SELECT 1 FROM src_associations sa WHERE sa.source = s.id)
102 AND NOT EXISTS (SELECT 1 FROM binaries b WHERE b.source = s.id)""");
104 #### XXX: this should ignore cases where the files for the binary b
105 #### have been marked for deletion (so the delay between bins go
106 #### byebye and sources go byebye is 0 instead of StayOfExecution)
110 projectB.query("BEGIN WORK");
115 # Mark the .dsc file for deletion
116 projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s AND last_used IS NULL" % (now_date, dsc_file_id))
117 # Mark all other files references by .dsc too if they're not used by anyone else
118 x = projectB.query("SELECT f.id FROM files f, dsc_files d WHERE d.source = %s AND d.file = f.id" % (source_id));
119 for j in x.getresult():
121 y = projectB.query("SELECT id FROM dsc_files d WHERE d.file = %s" % (file_id));
122 if len(y.getresult()) == 1:
123 projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s AND last_used IS NULL" % (now_date, file_id));
124 projectB.query("COMMIT WORK");
126 # Check for any sources which are marked for deletion but which
127 # are now used again.
129 q = projectB.query("""
130 SELECT f.id FROM source s, files f, dsc_files df
131 WHERE f.last_used IS NOT NULL AND s.id = df.source AND df.file = f.id
132 AND ((EXISTS (SELECT 1 FROM src_associations sa WHERE sa.source = s.id))
133 OR (EXISTS (SELECT 1 FROM binaries b WHERE b.source = s.id)))""");
135 #### XXX: this should also handle deleted binaries specially (ie, not
136 #### reinstate sources because of them
139 # Could be done in SQL; but left this way for hysterical raisins
140 # [and freedom to innovate don'cha know?]
141 projectB.query("BEGIN WORK");
144 projectB.query("UPDATE files SET last_used = NULL WHERE id = %s" % (file_id));
145 projectB.query("COMMIT WORK");
147 ########################################
150 global delete_date, now_date;
152 # FIXME: this is evil; nothing should ever be in this state. if
153 # they are, it's a bug and the files should not be auto-deleted.
157 print "Checking for unused files..."
158 q = projectB.query("""
159 SELECT id FROM files f
160 WHERE NOT EXISTS (SELECT 1 FROM binaries b WHERE b.file = f.id)
161 AND NOT EXISTS (SELECT 1 FROM dsc_files df WHERE df.file = f.id)""");
163 projectB.query("BEGIN WORK");
164 for i in q.getresult():
166 projectB.query("UPDATE files SET last_used = '%s' WHERE id = %s" % (now_date, file_id));
167 projectB.query("COMMIT WORK");
169 def clean_binaries():
170 global delete_date, now_date;
172 # We do this here so that the binaries we remove will have their
173 # source also removed (if possible).
175 # XXX: why doesn't this remove the files here as well? I don't think it
176 # buys anything keeping this separate
177 print "Cleaning binaries from the DB..."
178 if not Options["No-Action"]:
179 before = time.time();
180 sys.stdout.write("[Deleting from binaries table... ");
181 sys.stderr.write("DELETE FROM binaries WHERE EXISTS (SELECT 1 FROM files WHERE binaries.file = files.id AND files.last_used <= '%s')\n" % (delete_date));
182 projectB.query("DELETE FROM binaries WHERE EXISTS (SELECT 1 FROM files WHERE binaries.file = files.id AND files.last_used <= '%s')" % (delete_date));
183 sys.stdout.write("done. (%d seconds)]\n" % (int(time.time()-before)));
185 ########################################
188 global delete_date, now_date;
192 print "Cleaning out packages..."
194 date = time.strftime("%Y-%m-%d");
195 dest = Cnf["Dir::Morgue"] + '/' + Cnf["Rhona::MorgueSubDir"] + '/' + date;
196 if not os.path.exists(dest):
200 if not Options["No-Action"]:
201 before = time.time();
202 sys.stdout.write("[Deleting from source table... ");
203 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));
204 projectB.query("DELETE FROM source WHERE EXISTS (SELECT 1 FROM files WHERE source.file = files.id AND files.last_used <= '%s')" % (delete_date));
205 sys.stdout.write("done. (%d seconds)]\n" % (int(time.time()-before)));
207 # Delete files from the pool
208 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));
209 for i in q.getresult():
210 filename = i[0] + i[1];
211 if not os.path.exists(filename):
212 utils.warn("can not find '%s'." % (filename));
214 if os.path.isfile(filename):
215 if os.path.islink(filename):
217 if Options["No-Action"]:
218 print "Removing symlink %s..." % (filename);
222 size += os.stat(filename)[stat.ST_SIZE];
225 dest_filename = dest + '/' + os.path.basename(filename);
226 # If the destination file exists; try to find another filename to use
227 if os.path.exists(dest_filename):
228 dest_filename = utils.find_next_free(dest_filename);
230 if Options["No-Action"]:
231 print "Cleaning %s -> %s ..." % (filename, dest_filename);
233 utils.move(filename, dest_filename);
235 utils.fubar("%s is neither symlink nor file?!" % (filename));
237 # Delete from the 'files' table
238 if not Options["No-Action"]:
239 before = time.time();
240 sys.stdout.write("[Deleting from files table... ");
241 projectB.query("DELETE FROM files WHERE last_used <= '%s'" % (delete_date));
242 sys.stdout.write("done. (%d seconds)]\n" % (int(time.time()-before)));
244 sys.stderr.write("Cleaned %d files, %s.\n" % (count, utils.size_type(size)));
246 ################################################################################
248 def clean_maintainers():
249 print "Cleaning out unused Maintainer entries..."
251 q = projectB.query("""
252 SELECT m.id FROM maintainer m
253 WHERE NOT EXISTS (SELECT 1 FROM binaries b WHERE b.maintainer = m.id)
254 AND NOT EXISTS (SELECT 1 FROM source s WHERE s.maintainer = m.id)""");
258 projectB.query("BEGIN WORK");
260 maintainer_id = i[0];
261 if not Options["No-Action"]:
262 projectB.query("DELETE FROM maintainer WHERE id = %s" % (maintainer_id));
264 projectB.query("COMMIT WORK");
267 sys.stderr.write("Cleared out %d maintainer entries.\n" % (count));
269 ################################################################################
271 def clean_fingerprints():
272 print "Cleaning out unused fingerprint entries..."
274 q = projectB.query("""
275 SELECT f.id FROM fingerprint f
276 WHERE NOT EXISTS (SELECT 1 FROM binaries b WHERE b.sig_fpr = f.id)
277 AND NOT EXISTS (SELECT 1 FROM source s WHERE s.sig_fpr = f.id)""");
281 projectB.query("BEGIN WORK");
283 fingerprint_id = i[0];
284 if not Options["No-Action"]:
285 projectB.query("DELETE FROM fingerprint WHERE id = %s" % (fingerprint_id));
287 projectB.query("COMMIT WORK");
290 sys.stderr.write("Cleared out %d fingerprint entries.\n" % (count));
292 ################################################################################
294 def clean_queue_build():
297 if not Cnf.ValueList("Dinstall::QueueBuildSuites") or Options["No-Action"]:
300 print "Cleaning out queue build symlinks..."
302 our_delete_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-int(Cnf["Rhona::QueueBuildStayOfExecution"])));
305 q = projectB.query("SELECT filename FROM queue_build WHERE last_used <= '%s'" % (our_delete_date));
306 for i in q.getresult():
308 if not os.path.exists(filename):
309 utils.warn("%s (from queue_build) doesn't exist." % (filename));
311 if not Cnf.FindB("Dinstall::SecurityQueueBuild") and not os.path.islink(filename):
312 utils.fubar("%s (from queue_build) should be a symlink but isn't." % (filename));
315 projectB.query("DELETE FROM queue_build WHERE last_used <= '%s'" % (our_delete_date));
318 sys.stderr.write("Cleaned %d queue_build files.\n" % (count));
320 ################################################################################
323 global Cnf, Options, projectB, delete_date, now_date;
325 Cnf = utils.get_conf()
326 for i in ["Help", "No-Action" ]:
327 if not Cnf.has_key("Rhona::Options::%s" % (i)):
328 Cnf["Rhona::Options::%s" % (i)] = "";
330 Arguments = [('h',"help","Rhona::Options::Help"),
331 ('n',"no-action","Rhona::Options::No-Action")];
333 apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
334 Options = Cnf.SubTree("Rhona::Options")
339 projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
341 now_date = time.strftime("%Y-%m-%d %H:%M");
342 delete_date = time.strftime("%Y-%m-%d %H:%M", time.localtime(time.time()-int(Cnf["Rhona::StayOfExecution"])));
350 clean_fingerprints();
353 ################################################################################
355 if __name__ == '__main__':