Extract zips module

1 files changed, 28 insertions, 0 deletions

diff --git a/mammoth/zips.py b/mammoth/zips.py
new file mode 100644
index 0000000..c6fe358
--- /dev/null
+++ b/mammoth/zips.py
@@ -0,0 +1,28 @@
+import contextlib
+
+from zipfile import ZipFile
+
+
+def open_zip(fileobj, mode):
+    return _Zip(ZipFile(fileobj, mode))
+
+
+class _Zip(object):
+    def __init__(self, zip_file):
+        self._zip_file = zip_file
+    
+    def __enter__(self):
+        return self
+    
+    def __exit__(self, *args):
+        self._zip_file.close()
+
+    def open(self, name):
+        return contextlib.closing(self._zip_file.open(name))
+
+    def exists(self, name):
+        try:
+            self._zip_file.getinfo(name)
+            return True
+        except KeyError:
+            return False