mercurial/util.py

51720:e618a1756b08

import shutil
import stat
import sys
import time
import traceback
import warnings

from typing import (
    Any,
    Iterable,
    Iterator,
    List,
    Optional,
    Tuple,

    encoding,
    error,
    i18n,
    policy,
    pycompat,
    urllibcompat,
)
from .utils import (
    compression,
    hashutil,

    (1, 1 << 10, _(b'%.2f KB')),
    (1, 1, _(b'%.0f bytes')),
)


class transformingwriter:
    """Writable file wrapper to transform data by function"""

    def __init__(self, fp, encode):
        self._fp = fp
        self._encode = encode

    def close(self):
        self._fp.close()

    def flush(self):
        self._fp.flush()

    def write(self, data):
        return self._fp.write(self._encode(data))


# Matches a single EOL which can either be a CRLF where repeated CR
# are removed or a LF. We do not care about old Macintosh files, so a

def tocrlf(s: bytes) -> bytes:
    return _eolre.sub(b'\r\n', s)


def _crlfwriter(fp):
    return transformingwriter(fp, tocrlf)


if pycompat.oslinesep == b'\r\n':
    tonativeeol = tocrlf

    nativeeolwriter = _crlfwriter
else:
    tonativeeol = pycompat.identity
    fromnativeeol = pycompat.identity
    nativeeolwriter = pycompat.identity


# TODO delete since workaround variant for Python 2 no longer needed.
def iterfile(fp):
    return fp

51721:ed28085827ec

import shutil
import stat
import sys
import time
import traceback
import typing
import warnings

from typing import (
    Any,
    BinaryIO,
    Callable,
    Iterable,
    Iterator,
    List,
    Optional,
    Tuple,

    encoding,
    error,
    i18n,
    policy,
    pycompat,
    typelib,
    urllibcompat,
)
from .utils import (
    compression,
    hashutil,

    (1, 1 << 10, _(b'%.2f KB')),
    (1, 1, _(b'%.0f bytes')),
)


class transformingwriter(typelib.BinaryIO_Proxy):
    """Writable file wrapper to transform data by function"""

    def __init__(self, fp: BinaryIO, encode: Callable[[bytes], bytes]) -> None:
        self._fp = fp
        self._encode = encode

    def close(self) -> None:
        self._fp.close()

    def flush(self) -> None:
        self._fp.flush()

    def write(self, data: bytes) -> int:
        return self._fp.write(self._encode(data))


# Matches a single EOL which can either be a CRLF where repeated CR
# are removed or a LF. We do not care about old Macintosh files, so a

def tocrlf(s: bytes) -> bytes:
    return _eolre.sub(b'\r\n', s)


def _crlfwriter(fp: typelib.BinaryIO_Proxy) -> typelib.BinaryIO_Proxy:
    return transformingwriter(fp, tocrlf)


if pycompat.oslinesep == b'\r\n':
    tonativeeol = tocrlf

    nativeeolwriter = _crlfwriter
else:
    tonativeeol = pycompat.identity
    fromnativeeol = pycompat.identity
    nativeeolwriter = pycompat.identity

if typing.TYPE_CHECKING:
    # Replace the various overloads that come along with aliasing other methods
    # with the narrow definition that we care about in the type checking phase
    # only.  This ensures that both Windows and POSIX see only the definition
    # that is actually available.

    def tonativeeol(s: bytes) -> bytes:
        raise NotImplementedError

    def fromnativeeol(s: bytes) -> bytes:
        raise NotImplementedError

    def nativeeolwriter(fp: typelib.BinaryIO_Proxy) -> typelib.BinaryIO_Proxy:
        raise NotImplementedError


# TODO delete since workaround variant for Python 2 no longer needed.
def iterfile(fp):
    return fp