build.py

#!/usr/bin/env python

import sys
import os
import sh

from argparse import ArgumentParser


def build_and_publish(path, channel):
    try:
        token = os.environ['ANACONDA_TOKEN']
    except KeyError:
        sys.exit("Must set $ANACONDA_TOKEN")
    anaconda = sh.Command('anaconda').bake(t=token)
    conda = sh.Command('conda')

    binfile = conda.build("--output", path).strip()
    print "Building..."
    conda.build(path)
    print "Upload to Anaconda.org..."
    anaconda.upload(binfile, force=True, channel=channel)


def conda_paths(project_name):
    conda_recipes_dir = os.path.join(project_name, 'conda')

    if not os.path.isdir(conda_recipes_dir):
        sys.exit('no such dir: {}'.format(conda_recipes_dir))

    for name in sorted(os.listdir(conda_recipes_dir)):
        yield os.path.join(conda_recipes_dir, name)


def parse_args():
    parser = ArgumentParser()
    parser.add_argument('-p', '--project', required=True)
    parser.add_argument('-c', '--channel', required=False, default='main')
    parser.add_argument('-s', '--site', required=False, default=None)
    args = parser.parse_args()
    return args


def main():
    args = parse_args()
    for conda_path in conda_paths(args.project):
        build_and_publish(conda_path, channel=args.channel)


if __name__ == '__main__':
    main()