comparison mercurial/archival.py @ 45942:89a2afe31e82

formating: upgrade to black 20.8b1

This required a couple of small tweaks to un-confuse black, but now it
works. Big formatting changes come from:

 * Dramatically improved collection-splitting logic upstream
 * Black having a strong (correct IMO) opinion that """ is better than '''

Differential Revision: https://phab.mercurial-scm.org/D9430

45941:346af7687c6f

_UNX_IFREG = 0x8000
_UNX_IFLNK = 0xA000


def tidyprefix(dest, kind, prefix):
    '''choose prefix to use for names in archive.  make sure prefix is
    safe for consumers.'''

    if prefix:
        prefix = util.normpath(prefix)
    else:
        if not isinstance(dest, bytes):

    return out.getvalue()


class tarit(object):
    '''write archive to tar file or stream.  can write uncompressed,
    or compress with gzip or bzip2.'''

    if pycompat.ispy3:
        GzipFileWithTime = gzip.GzipFile  # camelcase-required
    else:

                    zlib.Z_BEST_COMPRESSION,
                    fileobj,
                    mtime=mtime,
                )
                self.fileobj = gzfileobj
                return tarfile.TarFile.taropen(  # pytype: disable=attribute-error
                    name, pycompat.sysstr(mode), gzfileobj
                )
            else:
                try:
                    return tarfile.open(
                        name, pycompat.sysstr(mode + kind), fileobj

        if self.fileobj:
            self.fileobj.close()


class zipit(object):
    '''write archive to zip file or stream.  can write uncompressed,
    or compressed with deflate.'''

    def __init__(self, dest, mtime, compress=True):
        if isinstance(dest, bytes):
            dest = pycompat.fsdecode(dest)
        self.z = zipfile.ZipFile(

    match=None,
    prefix=b'',
    mtime=None,
    subrepos=False,
):
    '''create archive of repo as it was at node.

    dest can be name of directory, name of archive file, or file
    object to write archive to.

    kind is type of archive to create.

    prefix is name of path to put before every archive member.

    mtime is the modified time, in seconds, or None to use the changeset time.

    subrepos tells whether to include subrepos.
    '''

    if kind == b'txz' and not pycompat.ispy3:
        raise error.Abort(_(b'xz compression is only available in Python 3'))

    if kind == b'files':

45942:89a2afe31e82

_UNX_IFREG = 0x8000
_UNX_IFLNK = 0xA000


def tidyprefix(dest, kind, prefix):
    """choose prefix to use for names in archive.  make sure prefix is
    safe for consumers."""

    if prefix:
        prefix = util.normpath(prefix)
    else:
        if not isinstance(dest, bytes):

    return out.getvalue()


class tarit(object):
    """write archive to tar file or stream.  can write uncompressed,
    or compress with gzip or bzip2."""

    if pycompat.ispy3:
        GzipFileWithTime = gzip.GzipFile  # camelcase-required
    else:

                    zlib.Z_BEST_COMPRESSION,
                    fileobj,
                    mtime=mtime,
                )
                self.fileobj = gzfileobj
                return (
                    tarfile.TarFile.taropen(  # pytype: disable=attribute-error
                        name, pycompat.sysstr(mode), gzfileobj
                    )
                )
            else:
                try:
                    return tarfile.open(
                        name, pycompat.sysstr(mode + kind), fileobj

        if self.fileobj:
            self.fileobj.close()


class zipit(object):
    """write archive to zip file or stream.  can write uncompressed,
    or compressed with deflate."""

    def __init__(self, dest, mtime, compress=True):
        if isinstance(dest, bytes):
            dest = pycompat.fsdecode(dest)
        self.z = zipfile.ZipFile(

    match=None,
    prefix=b'',
    mtime=None,
    subrepos=False,
):
    """create archive of repo as it was at node.

    dest can be name of directory, name of archive file, or file
    object to write archive to.

    kind is type of archive to create.

    prefix is name of path to put before every archive member.

    mtime is the modified time, in seconds, or None to use the changeset time.

    subrepos tells whether to include subrepos.
    """

    if kind == b'txz' and not pycompat.ispy3:
        raise error.Abort(_(b'xz compression is only available in Python 3'))

    if kind == b'files':