fabfile.py

import base64
import os
import json
import urllib2

from contextlib import contextmanager

from fabric.api import local, lcd, abort, settings
from fabric.decorators import task

from FlowCytometryTools import __version__

DL_DIR = "doc/source/_static/downloads"

BUILD_DIRS = (
    "dist",
    "doc/build",
    "doc/source/API",
    "build",
    "FlowCytometryTools.egg-info",
)

SDIST_RST_FILES = (
    "README.rst",
)

SDIST_TXT_FILES = [os.path.splitext(x)[0] + ".txt" for x in SDIST_RST_FILES]

BASE_PATH = os.path.abspath(os.path.dirname(__file__))


###############################################################################
# Misc.
###############################################################################
@task
def clean():
    """Clean build files."""
    with lcd(BASE_PATH):
        for build_dir in list(BUILD_DIRS):
            local("rm -rf %s" % build_dir)

@task
def html():
    """Make html files."""
    with lcd(os.path.join(BASE_PATH, 'doc')):
        local("make html")
        local("touch build/html/.nojekyll")

@task
def upload_doc():
    with lcd(BASE_PATH): 
        local('git branch -D gh-pages')
        local('git checkout -b gh-pages')
        local('git add -f doc/build/html')
        local('git commit -m "version {}"'.format(__version__))
        local('git push origin `git subtree split --prefix doc/build/html`:gh-pages --force')

@task
def serve(port="8000"):
    """
    Serve website from localhost.
    @param port  Port to run server on.
    """
    with lcd(BASE_PATH):
        with lcd("doc/build/html"):
            local("python -m SimpleHTTPServer %s" % port)


###############################################################################
# PyPI
###############################################################################
@contextmanager
def _dist_wrapper():
    """Add temporary distribution build files (and then clean up)."""
    try:
        # Copy select *.rst files to *.txt for build.
        for rst_file, txt_file in zip(SDIST_RST_FILES, SDIST_TXT_FILES):
            local("cp %s %s" % (rst_file, txt_file))

        # Perform action.
        yield
    finally:
        # Clean up temp *.txt files.
        for rst_file in SDIST_TXT_FILES:
            local("rm -f %s" % rst_file, capture=False)


@task
def sdist():
    """Package into distribution."""
    with _dist_wrapper():
        local("python setup.py sdist", capture=False)

@task
def pypi_register(server='pypitest'):
    """Register and prep user for PyPi upload.

    .. note:: May need to weak ~/.pypirc file per issue:
        http://stackoverflow.com/questions/1569315
    """
    base_command = 'python setup.py register'

    if server == 'pypitest':
        command = base_command + ' -r https://testpypi.python.org/pypi'
    else:
        command = base_command

    _execute_setup_command(command)

@task
def upload_to_pypi():
    _execute_setup_command('python setup.py sdist upload')

@task
def upload_to_test_pypi():
    _execute_setup_command('python setup.py sdist upload -r https://testpypi.python.org/pypi')

@task 
def install_from_pypi():
    _execute_setup_command('pip install FlowCytometryTools')

@task
def install_from_test_pypi():
    _execute_setup_command('pip install -i https://testpypi.python.org/pypi FlowCytometryTools')

def _execute_setup_command(command):
    with lcd(BASE_PATH):
        with _dist_wrapper():
            local(command, capture=False)