mcarton
19 lines
500 B
Python
19 lines
500 B
Python
import six
|
|
from decorator import decorator
|
|
|
|
|
|
@decorator
|
|
def sudo_support(fn, command):
|
|
"""Removes sudo before calling fn and adds it after."""
|
|
if not command.script.startswith('sudo '):
|
|
return fn(command)
|
|
|
|
result = fn(command.update(script=command.script[5:]))
|
|
|
|
if result and isinstance(result, six.string_types):
|
|
return u'sudo {}'.format(result)
|
|
elif isinstance(result, list):
|
|
return [u'sudo {}'.format(x) for x in result]
|
|
else:
|
|
return result
|