Allow directory names to be passed to directory.py.

1 files changed, 6 insertions, 1 deletions

diff --git a/directory.py b/directory.py
index dafebf0..11ac4f0 100644
--- a/directory.py
+++ b/directory.py
@@ -142,4 +142,9 @@ def scan(root = None):
 		session.close()
 
 if __name__ == '__main__':
-	scan()
+	import sys
+	if len(sys.argv) > 1:
+		for root in sys.argv[1:]:
+			scan(root)
+	else:
+		scan()