comparison mercurial/pycompat.py @ 31424:4acc49335a6e

py3: optimize py3 compat.bytechr using Struct.pack

With Python 3.4.3, timeit says 0.437 usec -> 0.0685 usec. With Python
3.6, timeit says 0.157 usec -> 0.0907 usec. So it's faster on both
versions, but the speedup varies a lot.

Thanks to Gregory Szorc for the suggestion.

31423:568d80b24b3a

if ispy3:
    import builtins
    import functools
    import io

    fsencode = os.fsencode
    fsdecode = os.fsdecode
    # A bytes version of os.name.
    osname = os.name.encode('ascii')

    # TODO: On Windows, the native argv is wchar_t, so we'll need a different
    # workaround to simulate the Python 2 (i.e. ANSI Win32 API) behavior.
    if getattr(sys, 'argv', None) is not None:
        sysargv = list(map(os.fsencode, sys.argv))

    def bytechr(i):
        return bytes([i])

    def iterbytestr(s):
        """Iterate bytes as if it were a str object of Python 2"""
        return iter(s[i:i + 1] for i in range(len(s)))

31424:4acc49335a6e

if ispy3:
    import builtins
    import functools
    import io
    import struct

    fsencode = os.fsencode
    fsdecode = os.fsdecode
    # A bytes version of os.name.
    osname = os.name.encode('ascii')

    # TODO: On Windows, the native argv is wchar_t, so we'll need a different
    # workaround to simulate the Python 2 (i.e. ANSI Win32 API) behavior.
    if getattr(sys, 'argv', None) is not None:
        sysargv = list(map(os.fsencode, sys.argv))

    bytechr = struct.Struct('>B').pack

    def iterbytestr(s):
        """Iterate bytes as if it were a str object of Python 2"""
        return iter(s[i:i + 1] for i in range(len(s)))