quick-start.py

#!/usr/bin/env python
# encoding: utf-8

from time import sleep
from os import getcwd
from subprocess import call, Popen, PIPE


def update_image_tags(version=None):
    """
    A simple function to configure the OpenEats version.

    :param version: The version of OpenEats the users wants to run.
                    This is a git Tag.
    :return: A file called `docker-prod.version.yml`.
             With the version of each image to pull.
    """
    version = version if version is not None else 'latest'
    version = '''version: '2.3'
services:
  api:
    image: openeats/openeats-api:%s
  web:
    image: openeats/openeats-web:%s
  nginx:
    image: openeats/openeats-nginx:%s
''' % (version, version, version)
    with open('docker-prod.version.yml', 'w') as f:
        f.write(version)


def download_images(version=None):
    """ Download the required images """
    version = version if version is not None else 'latest'
    print("==================")
    print("Downloading Images")
    print("==================")
    call(['docker', 'pull', 'openeats/openeats-api:' + version])
    call(['docker', 'pull', 'openeats/openeats-web:' + version])
    call(['docker', 'pull', 'openeats/openeats-nginx:' + version])


def start_containers():
    """
    Takes a back up of the Recipe images and DB.
    Restarts OpenEats with a new (or the same) version.
    """
    print("==================")
    print("Starting OpenEats")
    print("==================")

    # Check if the DB is up and running locally.
    # If it is then take a backup.
    # If the user is using a remote DB, do nothing.
    # If no DB is found, Start the docker DB and wait 45s to start.
    p = Popen(
        ['docker', 'ps', '-q', '-f', 'name=openeats_db_1'],
        stdin=PIPE,
        stdout=PIPE,
        stderr=PIPE
    )
    output, err = p.communicate(b"input data that is passed to subprocess' stdin")
    if output and not err:
        print("Taking a database backup (saving as openeats.sql)...")
        call(
            'docker exec openeats_db_1 sh -c ' +
            '\'exec mysqldump openeats -uroot -p"$MYSQL_ROOT_PASSWORD"\'' +
            ' > openeats.sql',
            shell=True
        )
    elif 'MYSQL_HOST' in open('env_prod.list').read():
        # TODO: add process to backup remote DB
        print("Using remote DB...")
    else:
        print("Creating the DB. This may take a minute...")
        call(['docker-compose', '-f', 'docker-prod.yml', 'up', '-d', 'db'])
        sleep(45)

    # Check if the API is up.
    # If it is then take a backup of the Recipe images folder.
    # The backup folder is called `site-media`.
    p = Popen(
        ['docker', 'ps', '-q', '-f', 'name=openeats_api_1'],
        stdin=PIPE,
        stdout=PIPE,
        stderr=PIPE
    )
    output, err = p.communicate(b"input data that is passed to subprocess' stdin")
    if output and not err:
        print("Taking a image backup save to 'site-media'...")
        call(
            'docker cp openeats_api_1:/code/site-media/ ' + getcwd(),
            shell=True
        )

    # Stop each container that needs to be updated.
    # Don't stop the DB! There is no reason to.
    call([
        'docker-compose',
        '-f', 'docker-prod.yml',
        '-f', 'docker-prod.version.yml',
        '-f', 'docker-prod.override.yml',
        'stop', 'nginx'
    ])
    call([
        'docker-compose',
        '-f', 'docker-prod.yml',
        '-f', 'docker-prod.version.yml',
        '-f', 'docker-prod.override.yml',
        'stop', 'api'
    ])
    call([
        'docker-compose',
        '-f', 'docker-prod.yml',
        '-f', 'docker-prod.version.yml',
        '-f', 'docker-prod.override.yml',
        'stop', 'web'
    ])

    # Start all the containers
    call([
        'docker-compose',
        '-f', 'docker-prod.yml',
        '-f', 'docker-prod.version.yml',
        '-f', 'docker-prod.override.yml',
        'up', '-d'
    ])

    print("App started. Please wait ~30 seconds for the containers to come online.")


if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(
        description='OpenEats quick setup script. '
                    'This script will restart your OpenEats server and '
                    'take a database and recipe image backup.'
    )
    parser.add_argument(
        '-t',
        '--tag',
        type=str,
        help='The git tag of OpenEats you want to run. '
             'If not included, then the master branch will be used.'
    )
    args = parser.parse_args()

    update_image_tags(args.tag)
    download_images(args.tag)
    start_containers()