iframe-proxy | Sunbelt Computer Software

287 lines (237 loc) · 6.65 KB
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
import contextlib
    from io import StringIO
except ImportError:
    from StringIO import StringIO  # 2.7
import os.path
import tempfile
@contextlib.contextmanager
def noop_cm():
def group_attr_names(attrnames):
    grouped = {
        "dunder": [],
        "private": [],
        "constants": [],
        "classes": [],
        "vars": [],
        "other": [],
    for name in attrnames:
        if name.startswith("__") and name.endswith("__"):
            group = "dunder"
        elif name.startswith("_"):
            group = "private"
        elif name.isupper():
            group = "constants"
        elif name.islower():
            group = "vars"
        elif name == name.capitalize():
            group = "classes"
        else:
            group = "other"
        grouped[group].append(name)
    return grouped
if sys.version_info < (3,):
    _str_to_lower = lambda val: val.decode().lower()
    _str_to_lower = str.lower
#############################
# file paths
_os_path = os.path
# Uncomment to test Windows behavior on non-windows OS:
# import ntpath as _os_path
PATH_SEP = _os_path.sep
NORMCASE = _os_path.normcase
DIRNAME = _os_path.dirname
BASENAME = _os_path.basename
IS_ABS_PATH = _os_path.isabs
PATH_JOIN = _os_path.join
def fix_path(
    _pathsep=PATH_SEP,
    """Return a platform-appropriate path for the given path."""
    if not path:
        return "."
    return path.replace("/", _pathsep)
def fix_relpath(
    _fix_path=fix_path,
    _path_isabs=IS_ABS_PATH,
    _pathsep=PATH_SEP,
    """Return a ./-prefixed, platform-appropriate path for the given path."""
    path = _fix_path(path)
    if path in (".", ".."):
        return path
    if not _path_isabs(path):
        if not path.startswith("." + _pathsep):
            path = "." + _pathsep + path
    return path
def _resolve_relpath(
    rootdir=None,
    _path_isabs=IS_ABS_PATH,
    _normcase=NORMCASE,
    _pathsep=PATH_SEP,
    # "path" is expected to use "/" for its path separator, regardless
    # of the provided "_pathsep".
    if path.startswith("./"):
        return path[2:]
    if not _path_isabs(path):
        return path
    # Deal with root-dir-as-fileid.
    _, sep, relpath = path.partition("/")
    if sep and not relpath.replace("/", ""):
        return ""
    if rootdir is None:
        return None
    rootdir = _normcase(rootdir)
    if not rootdir.endswith(_pathsep):
        rootdir += _pathsep
    if not _normcase(path).startswith(rootdir):
        return None
    return path[len(rootdir) :]
def fix_fileid(
    fileid,
    rootdir=None,
    normalize=False,
    strictpathsep=None,
    _pathsep=PATH_SEP,
    **kwargs
    """Return a pathsep-separated file ID ("./"-prefixed) for the given value.
    The file ID may be absolute.  If so and "rootdir" is
    provided then make the file ID relative.  If absolute but "rootdir"
    is not provided then leave it absolute.
    if not fileid or fileid == ".":
        return fileid
    # We default to "/" (forward slash) as the final path sep, since
    # that gives us a consistent, cross-platform result.  (Windows does
    # actually support "/" as a path separator.)  Most notably, node IDs
    # from pytest use "/" as the path separator by default.
    _fileid = fileid.replace(_pathsep, "/")
    relpath = _resolve_relpath(
        _fileid,
        rootdir,
        _pathsep=_pathsep,
        # ...
        **kwargs
    if relpath:  # Note that we treat "" here as an absolute path.
        _fileid = "./" + relpath
    if normalize:
        if strictpathsep:
            raise ValueError("cannot normalize *and* keep strict path separator")
        _fileid = _str_to_lower(_fileid)
    elif strictpathsep:
        # We do not use _normcase since we want to preserve capitalization.
        _fileid = _fileid.replace("/", _pathsep)
    return _fileid
#############################
@contextlib.contextmanager
def _replace_fd(file, target):
    Temporarily replace the file descriptor for `file`,
    for which sys.stdout or sys.stderr is passed.
        fd = file.fileno()
    except (AttributeError, io.UnsupportedOperation):
        # `file` does not have fileno() so it's been replaced from the
        # default sys.stdout, etc. Return with noop.
        yield
        return
    target_fd = target.fileno()
    # Keep the original FD to be restored in the finally clause.
    dup_fd = os.dup(fd)
        # Point the FD at the target.
        os.dup2(target_fd, fd)
        try:
            yield
        finally:
            # Point the FD back at the original.
            os.dup2(dup_fd, fd)
    finally:
        os.close(dup_fd)
@contextlib.contextmanager
def _replace_stdout(target):
    orig = sys.stdout
    sys.stdout = target
        yield orig
    finally:
        sys.stdout = orig
@contextlib.contextmanager
def _replace_stderr(target):
    orig = sys.stderr
    sys.stderr = target
        yield orig
    finally:
        sys.stderr = orig
if sys.version_info < (3,):
    _coerce_unicode = lambda s: unicode(s)
    _coerce_unicode = lambda s: s
@contextlib.contextmanager
def _temp_io():
    sio = StringIO()
    with tempfile.TemporaryFile("r+") as tmp:
        try:
            yield sio, tmp
        finally:
            tmp.seek(0)
            buff = tmp.read()
            sio.write(_coerce_unicode(buff))
@contextlib.contextmanager
def hide_stdio():
    """Swallow stdout and stderr."""
    with _temp_io() as (sio, fileobj):
        with _replace_fd(sys.stdout, fileobj):
            with _replace_stdout(fileobj):
                with _replace_fd(sys.stderr, fileobj):
                    with _replace_stderr(fileobj):
                        yield sio
#############################
def shlex_unsplit(argv):
    """Return the shell-safe string for the given arguments.
    This effectively the equivalent of reversing shlex.split().
    argv = [_quote_arg(a) for a in argv]
    return " ".join(argv)
    from shlex import quote as _quote_arg
except ImportError:
    def _quote_arg(arg):
        parts = None
        for i, c in enumerate(arg):
            if c.isspace():
            elif c == '"':
            elif c == "'":
                c = "'\"'\"'"
            else:
                continue
            if parts is None:
                parts = list(arg)
            parts[i] = c
        if parts is not None:
            arg = "'" + "".join(parts) + "'"
        return arg
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

util.py

Latest commit

History

util.py

File metadata and controls
