toolchain: supports .zip root directory detection

2015-02-09 11:58:29 +01:00 · 2015-02-09 11:58:29 +01:00 · 0415b214f7
parent 3e2ff0f300
commit 0415b214f7
1 changed files with 4 additions and 1 deletions
--- a/toolchain.py
+++ b/toolchain.py
@ -319,8 +319,11 @@ class Recipe(object):
            archive = tarfile.open(filename)
            root = archive.next().path.split("/")
            return root[0]
+        elif filename.endswith(".zip"):
+            with zipfile.ZipFile(filename) as zf:
+                return dirname(zf.namelist()[0])
        else:
-            print("Error: cannot detect root direction")
+            print("Error: cannot detect root directory")
            print("Unrecognized extension for {}".format(filename))
            raise Exception()