Add zip support to lorry

Change-Id: I29368ddcc64c82bafcbb834c81db30018a144f64

1 files changed, 14 insertions, 10 deletions

diff --git a/lorry b/lorry
index 1ff6f69..da1e556 100755
--- a/lorry
+++ b/lorry
@@ -25,6 +25,7 @@ import sys
 from datetime import datetime
 import shutil
 import traceback
+import functools
 
 __version__ = '0.0'
 
@@ -190,7 +191,8 @@ class Lorry(cliapp.Application):
             'git': self.mirror_git,
             'hg':  self.gitify_hg,
             'svn': self.gitify_svn,
-            'tarball': self.gitify_tarball,
+            'tarball': functools.partial(self.gitify_archive, 'tar'),
+            'zip': functools.partial(self.gitify_archive, 'zip')
         }
         vcstype = spec['type']
         if vcstype not in table:
@@ -477,32 +479,34 @@ class Lorry(cliapp.Application):
         self.run_program(['hg-fast-export', '--quiet', '--force', '-r', '../hg'], 
                          cwd=gitdir)
 
-    def gitify_tarball(self, project_name, dirname, gitdir, spec):
+    def gitify_archive(self, archive_type, project_name, dirname, gitdir, spec):
+        assert archive_type in ['zip', 'tar']
+
         url = spec['url']
         url_path = urllib2.urlparse.urlparse(url)[2]
         basename = os.path.basename(url_path)
-        tardest = os.path.join(dirname, basename)
+        archive_dest = os.path.join(dirname, basename)
         self.progress('.. checking if we need to fetch %s' % basename)
-        if file_missing_or_empty(tardest):
+        if file_missing_or_empty(archive_dest):
             self.progress('.. attempting to fetch.')
             try:
-                with open(tardest, 'w') as tarfile:
+                with open(archive_dest, 'w') as archive_file:
                     urlfile = urllib2.urlopen(spec['url'])
-                    tarfile.write(urlfile.read())
+                    archive_file.write(urlfile.read())
                     urlfile.close()
             except Exception as e:
-                if os.path.exists(tardest):
-                    os.unlink(tardest)
+                if os.path.exists(archive_dest):
+                    os.unlink(archive_dest)
                 raise
         else:
             self.progress('.. no need to run, nothing to do')
             return
         if not os.path.exists(gitdir):
             self.run_program(['git', 'init', '--bare', gitdir])
-        cmdline = ["%s.tar-importer" % lorry_path, tardest]
+
+        cmdline = ["%s.%s-importer" % (lorry_path, archive_type), archive_dest]
         self.run_program(cmdline, cwd=gitdir)
         self.needs_aggressive = True
-            
 
     def push_to_mirror_server(self, name, gitdir,
                               pushrefspecs=['refs/heads/*:refs/heads/*',