zipfile polyfill for async extract and extractall.

async_zipfile.py

import os
import asyncio
from zipfile import ZipFile

__all__ = ("ZipFile", )


@asyncio.coroutine
def async_extract(self, member, path=None, pwd=None, loop=None):
    loop = loop or asyncio.get_event_loop()
    return loop.run_in_executor(None, self.extract, member, path, pwd)


@asyncio.coroutine
def async_extractall(self, path=None, members=None, pwd=None, loop=None):
    members = members or self.namelist()
    loop = loop or asyncio.get_event_loop()
    path = path or os.getcwd()
    fs = [self.async_extract(zipinfo, path, pwd) for zipinfo in members]
    done = yield from asyncio.gather(*fs, loop=loop, return_exceptions=True)
    # to resolve concurrent mkdirs
    retries = set(members)
    for p in done:
        if not isinstance(p, OSError):
            member = p.replace(path, '').replace(os.path.sep, '/').lstrip('/')
            retries.discard(member)
    coro = asyncio.gather(*[self.async_extract(m, path, pwd) for m in retries],
                          loop=loop)
    yield from coro

ZipFile.async_extract = async_extract
ZipFile.async_extractall = async_extractall