go-taskのtaskファイルからforemanのProcfileを生成する

foreman_gotask_service.py

import argparse
import sys

import yaml


def main(argv=sys.argv[1:]):
    parser = argparse.ArgumentParser()
    parser.add_argument('tag', default='default', nargs='?')
    parser.add_argument('-t', '--taskfile', default='Taskfile.yml')
    parser.add_argument('-o', '--output', default='Procfile')
    args = parser.parse_args(argv)

    data = yaml.load(open(args.taskfile, 'r'))
    tasks = {
        task_name: {
            "tags": data['tasks'][task_name]['service']['tags'].split(),
            "watch": bool(data['tasks'][task_name]['service'].get('watch')),
        }
        for task_name in data['tasks'] if data['tasks'][task_name].get('service')
    }

    lines = [
        f'{task_name}: task {task_name} {"--watch" if task.get("watch") else ""}\n'
        for task_name, task in tasks.items() if args.tag in task['tags']
    ]

    with open(args.output, 'w') as fp:
        fp.writelines(lines)


if __name__ == '__main__':
    main()