vscode-python/pythonFiles/testing_tools/adapter/util.py at master · rrmistry/vscode-python · GitHub

History

92 lines (78 loc) · 2.12 KB

Raw

# Licensed under the MIT License.

import contextlib

try:

from io import StringIO

except ImportError:

from StringIO import StringIO # 2.7

import sys

@contextlib.contextmanager

def noop_cm():

yield

@contextlib.contextmanager

def hide_stdio():

"""Swallow stdout and stderr."""

ignored = StdioStream()

sys.stdout = ignored

sys.stderr = ignored

try:

yield ignored

finally:

sys.stdout = sys.__stdout__

sys.stderr = sys.__stderr__

if sys.version_info < (3,):

class StdioStream(StringIO):

def write(self, msg):

StringIO.write(self, msg.decode())

else:

StdioStream = StringIO

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

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)

try:

from shlex import quote as _quote_arg

except ImportError:

def _quote_arg(arg):

parts = None

for i, c in enumerate(arg):

if c.isspace():

pass

elif c == '"':

pass

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

vscode-python/pythonFiles/testing_tools/adapter/util.py at master · rrmistry/vscode-python · GitHub

Expand file tree

Breadcrumbs

util.py

Latest commit

History

Breadcrumbs

util.py

File metadata and controls