1 files changed, 110 insertions, 0 deletions
diff --git a/prunedrpms/PruneDrpms.py b/prunedrpms/PruneDrpms.py
new file mode 100755
index 0000000..629970a
--- /dev/null
+++ b/prunedrpms/PruneDrpms.py
@@ -0,0 +1,110 @@
+#!/usr/bin/python -t
+# -*- mode: Python; indent-tabs-mode: nil; -*-
+#
+# 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
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+import errno, os, sys
+import fnmatch, re
+import rpmUtils.transaction, rpmUtils.miscutils
+import string
+
+DEBUG = False
+
+def getFileList(path, ext, filelist=[]):
+    """Return all files in path matching ext, store them in filelist,
+    recurse dirs. Returns a list object"""
+
+    extlen = len(ext)
+    totalpath = os.path.normpath(path)
+    try:
+        dir_list = os.listdir(totalpath)
+    except OSError, e:
+        errorprint(_('Error accessing directory %s, %s') % (totalpath, e))
+        sys.exit(1)
+
+    for d in dir_list:
+        if os.path.isdir(totalpath + '/' + d):
+            filelist = getFileList(os.path.join(totalpath, d), ext, filelist)
+        else:
+            if string.lower(d[-extlen:]) == '%s' % (ext):
+                filelist.append(os.path.join(totalpath, d))
+
+    return filelist
+
+def pruneRepo(srcdir):
+    ts = rpmUtils.transaction.initReadOnlyTransaction()
+    changed = False
+    
+    # Create list of src.rpm files.
+    # We don't use "glob", so sub-directories are supported.
+    print 'Expiring :', srcdir
+
+    rpmfiles = getFileList(srcdir, ".rpm")
+    foundrpms = {}
+    for f in rpmfiles:
+        try:
+           hdr = rpmUtils.miscutils.hdrFromPackage(ts,f)
+        except:
+           print "Unable to open %s" % f
+        else:
+            n = hdr['name']
+            v = hdr['version']
+            r = hdr['release']
+            e = hdr['epoch']
+            a = hdr['arch']
+            if e is None:
+                e = 0
+            foundrpms.setdefault((n,e,v,r,a), True)
+
+    drpmfiles = getFileList(srcdir, ".drpm")
+
+    rm_files = []
+    for f in drpmfiles:
+        try:
+           hdr = rpmUtils.miscutils.hdrFromPackage(ts,f)
+        except:
+           print "Unable to open %s" % f
+        else:
+            n = hdr['name']
+            v = hdr['version']
+            r = hdr['release']
+            e = hdr['epoch']
+            a = hdr['arch']
+            if e is None:
+                e = 0
+            if not foundrpms.has_key((n,e,v,r,a)):
+                rm_files.append(f)
+
+    for f in rm_files:
+        print '  Removing', os.path.basename(f)
+        if not DEBUG:
+            os.remove(f)
+
+
+def main(srcdir):
+    assert rpmUtils.miscutils.compareEVR((1,2,3),(1,2,0)) > 0
+    assert rpmUtils.miscutils.compareEVR((0,1,2),(0,1,2)) == 0
+    assert rpmUtils.miscutils.compareEVR((1,2,3),(4,0,99)) < 0
+
+    return pruneRepo(srcdir)
+
+
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print 'Usage: %s <folder>\n' % os.path.basename(sys.argv[0])
+        sys.exit(errno.EINVAL)
+
+    main(sys.argv[1])
+    sys.exit(0)