Changeset 64 for trunk

Timestamp:

13/09/07 03:59:17 (4 years ago)

Author:

nigel.sim

Message:

• added cli option parsing
• added directory exclusion

File:

• trunk/spl-editor/main.py (modified) (3 diffs)

trunk/spl-editor/main.py

@@ -3 +3 @@
 import os, os.path
 import sys
+import getopt
 
 byArtist = {}
@@ -10 +11 @@
 #playlists = '/home/nigel/misc/Music/playlists'
 def showUsage():
-        print "main.py <Root directory> <Playlist directory>"
+        print "main.py [-e <excluded directory> ...] <Root directory> <Playlist directory>"
 
 
 if __name__ == '__main__':
-        if len(sys.argv) != 3:
+        optlist, args = getopt.getopt(sys.argv[1:],'e:')
+        if len(args) != 2:
                 showUsage()
                 sys.exit(1)
         else:
-                start = sys.argv[1]
-                playlists = sys.argv[2]
+                start = args[0]
+                playlists = args[1]
+                excludeList = []
+                for tag, exdir in optlist:
+                        if tag == '-e':
+                                excludeList.append(exdir)
+        # Iterate through the dirs below the start dir
         for root, dirs, files in os.walk(start):
+                # Check the dir isn't excluded
+                skip = False
+                for ex in excludeList:
+                        if root.startswith(ex):
+                                print "skipping", root
+                                skip = True
+                                continue
+                if skip:
+                        continue
+
                 for f in files:
                         md = kaa.metadata.parse(os.path.join(root, f))
@@ -36 +53 @@
 
         for a in byArtist:
-                files = [md.fileName.replace(start,'') for md in byArtist[a] ]
+                files = [md.fileName.replace(start,'').replace('/','\\') for md in byArtist[a] ]
                 pl = spl.SPL()
                 pl.list = files