dl_cleanup: Add regex for foo_1_2_3

[openwrt/staging/yousong.git] / scripts / dl_cleanup.py

diff --git a/scripts/dl_cleanup.py b/scripts/dl_cleanup.py
index e83c82ac2c629ad1c2e9ab5e127cec52ec663289..7adc475d4da5d2812ba913843468f6b88402b2db 100755 (executable)
--- a/scripts/dl_cleanup.py
+++ b/scripts/dl_cleanup.py
@@ -9,9 +9,13 @@
 import sys
 import os
 import re
+import getopt
 
 DEBUG = 0
 
+# Commandline options
+opt_dryrun = False
+
 
 def parseVer_1234(match):
        progname = match.group(1)
@@ -23,7 +27,10 @@ def parseVer_1234(match):
 
 def parseVer_123(match):
        progname = match.group(1)
-       patchlevel = match.group(5)
+       try:
+               patchlevel = match.group(5)
+       except (IndexError), e:
+               patchlevel = None
        if patchlevel:
                patchlevel = ord(patchlevel[0])
        else:
@@ -36,7 +43,10 @@ def parseVer_123(match):
 
 def parseVer_12(match):
        progname = match.group(1)
-       patchlevel = match.group(4)
+       try:
+               patchlevel = match.group(4)
+       except (IndexError), e:
+               patchlevel = None
        if patchlevel:
                patchlevel = ord(patchlevel[0])
        else:
@@ -72,18 +82,20 @@ versionRegex = (
        (re.compile(r"(.+)[-_](\d+)\.(\d+)\.(\d+)\.(\d+)"), parseVer_1234),     # xxx-1.2.3.4
        (re.compile(r"(.+)[-_](\d\d\d\d)-?(\d\d)-?(\d\d)"), parseVer_ymd),      # xxx-YYYY-MM-DD
        (re.compile(r"(.+)[-_](\d+)\.(\d+)\.(\d+)(\w?)"), parseVer_123),        # xxx-1.2.3a
+       (re.compile(r"(.+)[-_](\d+)_(\d+)_(\d+)"), parseVer_123),               # xxx-1_2_3
        (re.compile(r"(.+)[-_](\d+)\.(\d+)(\w?)"), parseVer_12),                # xxx-1.2a
        (re.compile(r"(.+)[-_]r?(\d+)"), parseVer_r),                           # xxx-r1111
 )
 
-blacklist = (
-       re.compile(r"wl_apsta.*"),
-       re.compile(r"boost.*"),
-       re.compile(r".*\.fw"),
-       re.compile(r".*\.arm"),
-       re.compile(r".*\.bin"),
-       re.compile(r"RT\d+_Firmware.*"),
-)
+blacklist = [
+       ("linux",               re.compile(r"linux-.*")),
+       ("gcc",                 re.compile(r"gcc-.*")),
+       ("wl_apsta",            re.compile(r"wl_apsta.*")),
+       (".fw",                 re.compile(r".*\.fw")),
+       (".arm",                re.compile(r".*\.arm")),
+       (".bin",                re.compile(r".*\.bin")),
+       ("rt-firmware",         re.compile(r"RT\d+_Firmware.*")),
+]
 
 class EntryParseError(Exception): pass
 
@@ -114,7 +126,7 @@ class Entry:
        def deleteFile(self):
                path = (self.directory + "/" + self.filename).replace("//", "/")
                print "Deleting", path
-               if not DEBUG:
+               if not opt_dryrun:
                        os.unlink(path)
 
        def __eq__(self, y):
@@ -125,22 +137,54 @@ class Entry:
 
 def usage():
        print "OpenWRT download directory cleanup utility"
-       print "Usage: " + sys.argv[0] + " path/to/dl"
+       print "Usage: " + sys.argv[0] + " [OPTIONS] <path/to/dl>"
+       print ""
+       print " -d|--dry-run            Do a dry-run. Don't delete any files"
+       print " -B|--show-blacklist     Show the blacklist and exit"
+       print " -w|--whitelist ITEM     Remove ITEM from blacklist"
 
 def main(argv):
-       if len(argv) != 2:
+       global opt_dryrun
+
+       try:
+               (opts, args) = getopt.getopt(argv[1:],
+                       "hdBw:",
+                       [ "help", "dry-run", "show-blacklist", "whitelist=", ])
+               if len(args) != 1:
+                       raise getopt.GetoptError()
+       except getopt.GetoptError:
                usage()
                return 1
-       directory = argv[1]
+       directory = args[0]
+       for (o, v) in opts:
+               if o in ("-h", "--help"):
+                       usage()
+                       return 0
+               if o in ("-d", "--dry-run"):
+                       opt_dryrun = True
+               if o in ("-w", "--whitelist"):
+                       for i in range(0, len(blacklist)):
+                               (name, regex) = blacklist[i]
+                               if name == v:
+                                       del blacklist[i]
+                                       break
+                       else:
+                               print "Whitelist error: Item", v,\
+                                     "is not in blacklist"
+                               return 1
+               if o in ("-B", "--show-blacklist"):
+                       for (name, regex) in blacklist:
+                               print name
+                       return 0
 
        # Create a directory listing and parse the file names.
        entries = []
        for filename in os.listdir(directory):
                if filename == "." or filename == "..":
                        continue
-               for black in blacklist:
-                       if black.match(filename):
-                               if DEBUG:
+               for (name, regex) in blacklist:
+                       if regex.match(filename):
+                               if opt_dryrun:
                                        print filename, "is blacklisted"
                                break
                else:
@@ -167,7 +211,7 @@ def main(argv):
                        for version in versions:
                                if version != lastVersion:
                                        version.deleteFile()
-                       if DEBUG:
+                       if opt_dryrun:
                                print "Keeping", lastVersion.filename
 
        return 0