Gecko [ v99win.today ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
in_process	[ DIR ]	drwxr-xr-x
__init__.py	130 B	-rw-r--r--
build.py	3.38 KB	-rw-r--r--
check.py	5.94 KB	-rw-r--r--
colorlog.py	4 KB	-rw-r--r--
compat.py	1.22 KB	-rw-r--r--
dirtools.py	1.1 KB	-rw-r--r--
envbuild.py	5.96 KB	-rw-r--r--
meta.py	2.41 KB	-rw-r--r--
wrappers.py	13.11 KB	-rw-r--r--

Code Editor : dirtools.py

import os
import io
import contextlib
import tempfile
import shutil
import errno
import zipfile

@contextlib.contextmanager
def tempdir():
    """Create a temporary directory in a context manager."""
    td = tempfile.mkdtemp()
    try:
        yield td
    finally:
        shutil.rmtree(td)

def mkdir_p(*args, **kwargs):
    """Like `mkdir`, but does not raise an exception if the
    directory already exists.
    """
    try:
        return os.mkdir(*args, **kwargs)
    except OSError as exc:
        if exc.errno != errno.EEXIST:
            raise

def dir_to_zipfile(root):
    """Construct an in-memory zip file for a directory."""
    buffer = io.BytesIO()
    zip_file = zipfile.ZipFile(buffer, 'w')
    for root, dirs, files in os.walk(root):
        for path in dirs:
            fs_path = os.path.join(root, path)
            rel_path = os.path.relpath(fs_path, root)
            zip_file.writestr(rel_path + '/', '')
        for path in files:
            fs_path = os.path.join(root, path)
            rel_path = os.path.relpath(fs_path, root)
            zip_file.write(fs_path, rel_path)
    return zip_file

${this.title}

Code Editor : dirtools.py