fabricのenvをyamlで定義する

fabfile.py

# -*- coding: utf-8 -*-

from fabric.api import *
from yaml_loader import load_env_from_yaml

load_env_from_yaml('env.yaml', globals())

yaml_loader.py

# -*- coding: utf-8 -*-

import os, time
import yaml
from fabric.api import env,task

def load_env_from_yaml(yaml_file, namespace):
    u'''yamlファイルから実行環境タスクを作成'''
    if os.path.exists(yaml_file):
        env_data = yaml.safe_load(open(yaml_file).read())
        for env_name, env_dict in env_data.items():
            __create_env_task(env_name, env_dict, namespace)
    else:
        print('No {0} found. skipped.'.format(yaml_file))

class EnvTask(object):

    def __init__(self, name):
        self.name = name

    def set_env(self, env_dict):
        env.update(env_dict)

def __create_env_task(env_name, env_dict, namespace):
    u'''実行環境タスクの定義'''

    e = EnvTask(env_name)
    f = e.set_env
    f.__func__.__doc__ = 'Set environment for {0}'.format(env_name)
    task_name = 'env_{0}'.format(env_name)
    wrapper = task(name=task_name)
    rand = '%d' % (time.time() * 100000)
    namespace['task_%s_%s' % (task_name, rand)] = wrapper(f)