wooyek/cookiecutter-django-app

#!/usr/bin/env python
import os
import platform
import shutil
import sys
from subprocess import call

import six

PROJECT_DIRECTORY = os.path.realpath(os.path.curdir)

PROJECT_SLUG = "{{ cookiecutter.project_slug }}"
CREATE_VIRTUAL_ENVIRONMENT = '{{ cookiecutter.create_virtual_environment }}'.lower() == 'y'
CREATE_EXAMPLE_PROJECT = '{{ cookiecutter.create_example_project }}'.lower() == 'y'
CREATE_AUTHOR_FILE = '{{ cookiecutter.create_author_file }}' != 'y'
REPO_URL = "{{ cookiecutter.repo_url }}"
INCLUDE_SPHINX_DOC = '{{ cookiecutter.include_sphinx_doc }}' != 'y'
GIT_INIT = '{{ cookiecutter.git_init }}'.lower() == 'y'
RUN_TESTS = '{{ cookiecutter.run_tests_on_init }}' == 'y'
INCLUDE_SPHINX = '{{ cookiecutter.include_sphinx_doc }}' == 'y'

VEX_AVAILABLE = shutil.which('vex')
PEW_AVAILABLE = shutil.which('pew')
PIPENV_AVAILABLE = shutil.which('pipenv')


def remove_file(filepath):
    os.remove(os.path.join(PROJECT_DIRECTORY, filepath))


def bootstrap_venv():
    # On Windows  current python3 may just be python or not available from PATH
    python = sys.executable or 'python3'
    if PEW_AVAILABLE:
        print("-------> Python Env Wrapper: Bootstrapping virtual environment")
        # delegate all the work to the pew tool
        call((
            "pew", 'new',
            "--python", python,
            '-a', PROJECT_DIRECTORY,
            '-i', 'setuptools',
            '-r', 'requirements/development.txt',
            '--dont-activate',
            '{{ cookiecutter.project_slug }}'
        ))
        call(["pew", "in", PROJECT_SLUG, "pip", "-e", "."])

    else:
        if six.PY2:
            print("-------> VirtualEnv: Bootstrapping virtual environment")
            call(["virtualenv", "--clear", ".pyvenv"])
        else:
            print("-------> PyVenv: Bootstrapping virtual environment")
            call([python, "-m", "venv", "--clear", ".pyvenv"])

        venv_py = ".pyvenv/Scripts/python" if platform.system() == "Windows" else ".pyvenv/bin/python"
        call([venv_py, "-m", "pip", "install", "-U", "pip", "setuptools"])
        call([venv_py, "-m", "pip", "install", "-r", "requirements/development.txt"])
        call([venv_py, "-m", "pip", "install", "-e", "."])


def git_init():
    print("-------> Initializing git repo")
    call(["git", "init"])
    call(["git", "add", "--all"])
    call(["git", "commit", "-am", "init"])
    call(["git", "flow", "init", "-d"])
    call(["git", "remote", "add", "origin", REPO_URL])


if __name__ == '__main__':
    if CREATE_AUTHOR_FILE:
        remove_file('AUTHORS.rst')
        remove_file('docs/authors.rst')

    if CREATE_VIRTUAL_ENVIRONMENT:
        bootstrap_venv()
        if INCLUDE_SPHINX:
            if VEX_AVAILABLE:
                print("-------> Building documentation through vex + make")
                call(["vex", PROJECT_SLUG, "make", "docs"])
            else:
                # TODO: Activate pyvenv before running make
                call(["make", "docs"])

    if INCLUDE_SPHINX_DOC:
        shutil.rmtree(os.path.join(PROJECT_DIRECTORY, 'docs'))

    if not CREATE_EXAMPLE_PROJECT:
        location = os.path.join(PROJECT_DIRECTORY, 'example_project')
        shutil.rmtree(location)

    if GIT_INIT:
        git_init()

    if RUN_TESTS:
        print("-------> Running tests")
        # call(["detox", "--skip-missing-interpreters"])
        # call(["detox", "--skip-missing-interpreters", "-e", "clean,py35-django-111,check,report,docs,spell"])
        # call(["make", "test"])
        call(["make", "tox"])

    print(""" 
 ╭────────────────────────────────────────────────────────────────────────────
 │ Hi {{ cookiecutter.repo_username }}, 
 │  
 │ as you probably know, creating and maintaining any open source is a massive 
 │ amount of free work. So please spread the word, star any project you use 
 │ and say thank's to it's authors on twitter.
 │ 
 │ Thx, @wooyek
 │ https://github.com/wooyek
 ╰────────────────────────────────────────────────────────────────────────────
""")
    if CREATE_VIRTUAL_ENVIRONMENT and VEX_AVAILABLE:
        call(["vex", PROJECT_SLUG])