From e65ed8623f744414a5989d29991fe87f200a7347 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonas=20Borgstr=C3=B6m?= <jonas@borgstrom.se>
Date: Tue, 23 Feb 2010 22:12:22 +0100
Subject: [PATCH] Initial optparse implementation.

---
 dedupstore/archiver.py | 49 +++++++++++++++++++++++++++++++-----------
 1 file changed, 36 insertions(+), 13 deletions(-)

diff --git a/dedupstore/archiver.py b/dedupstore/archiver.py
index cdaa697eb..e1013da3d 100644
--- a/dedupstore/archiver.py
+++ b/dedupstore/archiver.py
@@ -3,6 +3,8 @@
 import hashlib
 import zlib
 import cPickle
+from optparse import OptionParser
+
 from store import Store
 
 CHUNKSIZE = 256 * 1024
@@ -103,17 +105,18 @@ def __init__(self):
         self.store = Store('/tmp/store')
         self.cache = Cache('/tmp/cache', self.store)
 
-    def create_archive(self, archive_name, path):
+    def create_archive(self, archive_name, paths):
         if archive_name in self.cache.archives:
             raise Exception('Archive "%s" already exists' % archive_name)
         items = []
-        for root, dirs, files in os.walk(path):
-            for d in dirs:
-                name = os.path.join(root, d)
-                items.append(self.process_dir(name, self.cache))
-            for f in files:
-                name = os.path.join(root, f)
-                items.append(self.process_file(name, self.cache))
+        for path in paths:
+            for root, dirs, files in os.walk(path):
+                for d in dirs:
+                    name = os.path.join(root, d)
+                    items.append(self.process_dir(name, self.cache))
+                for f in files:
+                    name = os.path.join(root, f)
+                    items.append(self.process_file(name, self.cache))
         archive = {'name': name, 'items': items}
         hash = self.cache.add_chunk(zlib.compress(cPickle.dumps(archive)))
         self.cache.archives[archive_name] = hash
@@ -150,14 +153,34 @@ def process_file(self, path, cache):
             chunks.append(cache.add_chunk(zlib.compress(data)))
         print 'File: %s (%d chunks)' % (path, len(chunks))
         return {'type': 'FILE', 'path': path, 'size': size, 'chunks': chunks}
-
+    
+    def run(self):
+        parser = OptionParser()
+        parser.add_option("-C", "--cache", dest="cache",
+                          help="cache directory to use", metavar="CACHE")
+        parser.add_option("-s", "--store", dest="store",
+                          help="path to dedupe store", metavar="STORE")
+        parser.add_option("-c", "--create", dest="create_archive",
+                          help="create ARCHIVE", metavar="ARCHIVE")
+        parser.add_option("-d", "--delete", dest="delete_archive",
+                          help="delete ARCHIVE", metavar="ARCHIVE")
+        parser.add_option("-l", "--list-archives", dest="list_archives",
+                        help="list archives")
+        parser.add_option("-V", "--verify", dest="verify_archive",
+                        help="verify archive consistency")
+        parser.add_option("-e", "--extract", dest="extract_archive",
+                        help="extract ARCHIVE")
+        parser.add_option("-L", "--list-archive", dest="list_archive",
+                        help="verify archive consistency", metavar="ARCHIVE")
+        (options, args) = parser.parse_args()
+        if options.delete_archive:
+            self.delete_archive(options.delete_archive)
+        else:
+            self.create_archive(options.create_archive, args)
 
 def main():
     archiver = Archiver()
-    if sys.argv[1] == 'delete':
-        archiver.delete_archive(sys.argv[2])
-    else:
-        archiver.create_archive(sys.argv[1], sys.argv[2])
+    archiver.run()
 
 if __name__ == '__main__':
     main()
\ No newline at end of file