Custom task for fabric

fabfile.py

class CustomTask(Task):

    def __init__(self, func, myarg, *args, **kwargs):
        super(CustomTask, self).__init__(*args, **kwargs)
        self.func = func
        self.myarg = myarg
        if env.hosts == []:
            logger.debug("env.hosts is empty")
            # if hasattr(env, 'env_name'):
            logger.debug("Found env_name in env")
            set_dynamic_inventory()

    def run(self, *args, **kwargs):
        return self.func(*args, **kwargs)


@task(task_class=CustomTask, myarg='value')