Fix getting directory member that doesn't exist

ryanlovett · Mar 7, 2015 · f5f8f4b · f5f8f4b
1 parent bfa625e
commit f5f8f4b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/library/release_assignment.py b/library/release_assignment.py
@@ -61,9 +61,13 @@ def main():
     tf = tarfile.open(src, 'r:gz')
     prefix = os.path.commonprefix(tf.getnames()).rstrip("/")
     if prefix != '':
-        member = tf.getmember(prefix)
-        if not member.isdir():
-            prefix = os.path.dirname(prefix)
+        try:
+            member = tf.getmember(prefix)
+        except KeyError:
+            pass
+        else:
+            if not member.isdir():
+                prefix = os.path.dirname(prefix)
     tf.close()
     if prefix == '':
         module.fail_json(msg="Archive has no common prefix")