%PDF-
%PDF-
Mini Shell
Mini Shell
# -*- coding: utf-8 -*-
from __future__ import absolute_import, unicode_literals, print_function
import re
import logging
log = logging.getLogger(__name__)
try:
from testinfra import modules
TESTINFRA_PRESENT = True
except ImportError:
TESTINFRA_PRESENT = False
__all__ = []
__virtualname__ = 'testinfra'
def __virtual__():
if TESTINFRA_PRESENT:
return __virtualname__
return False, 'The Testinfra package is not available'
def _wrap_module_function(func_name):
def _module_function_wrapper(name, **methods):
func = __salt__[func_name]
result, passes, fails = func(name, **methods)
comment = '\n'.join(passes + fails)
if __opts__['test'] and not result:
result = None
return {'name': name,
'comment': comment,
'result': result,
'changes': {}}
return _module_function_wrapper
def _to_snake_case(pascal_case):
"""Convert a PascalCase string to its snake_case equivalent.
:param pascal_case: PascalCased string to be converted
:returns: snake_case string
:rtype: str
"""
snake_case = re.sub('(^|[a-z])([A-Z])',
lambda match: '{0}_{1}'.format(match.group(1).lower(),
match.group(2).lower()),
pascal_case)
return snake_case.lower().strip('_')
def _generate_functions():
try:
modules_ = [_to_snake_case(module_) for module_ in modules.__all__]
except AttributeError:
modules_ = [module_ for module_ in modules.modules]
for module_name in modules_:
func_name = 'testinfra.{0}'.format(module_name)
__all__.append(module_name)
log.debug('Generating state for module %s as function %s',
module_name, func_name)
globals()[module_name] = _wrap_module_function(func_name)
if TESTINFRA_PRESENT:
_generate_functions()
Zerion Mini Shell 1.0