# Copyright (c) Microsoft Corporation. All rights reserved.

# 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