sync

[dak.git] / tea

diff --git a/tea b/tea
index c8ca3ae647a1399bd2438cc2014a57e62df88cce..dc8aade08447ddd5d25896533c5b780fc14e5c94 100755 (executable)
--- a/tea
+++ b/tea
@@ -2,7 +2,7 @@
 
 # Sanity check the database
 # Copyright (C) 2000  James Troup <james@nocrew.org>
-# $Id: tea,v 1.2 2001-01-10 06:00:12 troup Exp $
+# $Id: tea,v 1.3 2001-01-16 21:52:37 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
@@ -25,7 +25,7 @@
 
 ################################################################################
 
-import pg, sys, os
+import pg, sys, os, string, stat
 import utils, db_access
 import apt_pkg;
 
@@ -33,34 +33,66 @@ import apt_pkg;
 
 Cnf = None;
 projectB = None;
+db_files = {};
+waste = 0.0;
+excluded = {};
+
+def process_dir (arg, dirname, filenames):
+    global waste, db_files, excluded;
+    
+    if string.find(dirname, '/disks-') != -1 or string.find(dirname, 'upgrade-') != -1: 
+        return;
+    # hack; can't handle .changes files
+    if string.find(dirname, 'proposed-updates') != -1:
+        return;
+    for name in filenames:
+        filename = os.path.abspath(dirname+'/'+name);
+        filename = string.replace(filename, 'potato-proposed-updates', 'proposed-updates');
+        if os.path.isfile(filename) and not os.path.islink(filename) and not db_files.has_key(filename) and not excluded.has_key(filename):
+            waste = waste + os.stat(filename)[stat.ST_SIZE];
+            print filename
 
 ################################################################################
 
 def main ():
-    global Cnf, projectB;
+    global Cnf, projectB, db_files, waste, excluded;
 
     apt_pkg.init();
     
     Cnf = apt_pkg.newConfiguration();
     apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file());
 
-    Arguments = [('d',"debug","Claire::Options::Debug", "IntVal"),
-                 ('h',"help","Claire::Options::Help"),
-                 ('v',"version","Claire::Options::Version")];
+    Arguments = [('d',"debug","Tea::Options::Debug", "IntVal"),
+                 ('h',"help","Tea::Options::Help"),
+                 ('v',"version","Tea::Options::Version")];
 
     apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
-
     projectB = pg.connect('projectb', 'localhost');
-
     db_access.init(Cnf, projectB);
 
+    print "Building list of Database files...";
+
     q = projectB.query("SELECT l.path, f.filename FROM files f, location l WHERE f.location = l.id")
     ql = q.getresult();
 
+    db_files = {};
     for i in ql:
-       filename = i[0] + i[1];
+       filename = os.path.abspath(i[0] + i[1]);
+        db_files[filename] = "";
         if os.access(filename, os.R_OK) == 0:
-            print filename
+            sys.stderr.write("W: '%s' doesn't exist.\n" % (filename));
+
+    file = utils.open_file(Cnf["Dir::OverrideDir"]+'override.unreferenced','r');
+    for filename in file.readlines():
+        filename = filename[:-1];
+        excluded[filename] = "";
+
+    print "Checking against existent files...";
+
+    os.path.walk(Cnf["Dir::RootDir"]+'dists/', process_dir, None);
+
+    print
+    print "%s wasted..." % (utils.size_type(waste));
 
 #######################################################################################