def __call__(self, func):
name = self.name
if name is None:
name = func.func_name
if self.camel_case:
name = camel_case(name)
logger.debug('register function {} as {}'.format(func.func_name, name))
REGISTRY[name] = (func.func_name, self.postprocess)
return func
def __call__(self, func):
name = self.name
if name is None:
name = func.func_name
if self.camel_case:
name = camel_case(name)
logger.debug('register function {} as {}'.format(func.func_name, name))
REGISTRY[name] = (func.func_name, self.postprocess)
return func
def wrapper(obj, *args, **kw):
cmd = self.cmd
if cmd is None:
cmd = func.func_name
if self.camel_case:
cmd = camel_case(cmd)
r = obj.ask(cmd)
if self.returntype:
try:
r = self.returntype(r)
except BaseException:
pass
return r
def wrapper(obj, *args, **kw):
cmd = self.cmd
if cmd is None:
cmd = func.func_name
if self.camel_case:
cmd = camel_case(cmd)
r = obj.ask(cmd)
if self.returntype:
try:
r = self.returntype(r)
except BaseException:
pass
return r
def wrapper(obj, *args, **kw):
cmd = self.cmd
if cmd is None:
cmd = func.__name__
if self.camel_case:
cmd = camel_case(cmd)
r = obj.ask(cmd, verbose=True)
if self.returntype:
try:
r = self.returntype(r)
except BaseException as e:
obj.critical('excpection running command {}. {}'.format(cmd, e))
return r
def test_already_camelcase(self):
n = self.expected
cn = camel_case(n)
self.assertEqual(n, cn)
def test_all_lowercase(self):
n = 'abqvolc'
cn = camel_case(n)
self.assertEqual('Abqvolc', cn)
def test_slash(self):
n = 'abq/volc'
cn = camel_case(n)
self.assertEqual(self.expected, cn)
def test_space(self):
n = 'abq volc'
cn = camel_case(n)
self.assertEqual(self.expected, cn)
def test_underscore(self):
n = 'abq_volc'
cn = camel_case(n)
self.assertEqual(self.expected, cn)
