text: import six
  from decorator import decorator
  from ..types import Command
  
  
  @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
  