deploy.py

# -*-*- encoding: utf-8 -*-*-
#
# gateway4labs is free software: you can redistribute it and/or modify
# it under the terms of the BSD 2-Clause License
# gateway4labs is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

import os
import sys

import getpass
import traceback
from optparse import OptionParser

if not os.path.exists('config.py'):
    print >> sys.stderr, "Missing config.py. Copy config.py.dist into config.py"
    sys.exit(-1)

try:
    from config import SQLALCHEMY_DATABASE_URI
except ImportError:
    try:
        from config import SQLALCHEMY_ENGINE_STR as SQLALCHEMY_DATABASE_URI
    except ImportError:
        raise Exception("SQLALCHEMY_DATABASE_URI not found in the configuration file" )

try:
    from config import USE_PYMYSQL
except ImportError:
    USE_PYMYSQL = True

import config
heroku = os.environ.get('HEROKU', None)

if heroku is None:
    if config.ENGINE == 'mysql':
        if USE_PYMYSQL:
            import pymysql as dbi
        else:
            import MySQLdb as dbi
    elif config.ENGINE == 'sqlite':
        import sqlite3 as dbi
    else:
        print >> sys.stderr, "Unsupported engine %s. You will have to create the database and the users by your own." % config.ENGINE

from labmanager.db import init_db
from labmanager.sample_data import add_sample_users


def create_user(db_user, db_password):
    if config.ENGINE == 'mysql':
        sentences = (
            "DROP USER '%s'@'localhost'" % config.USERNAME,
            "CREATE USER '%s'@'localhost' IDENTIFIED BY '%s'" % (config.USERNAME, config.PASSWORD),
            "GRANT ALL PRIVILEGES ON %s.* TO '%s'@'localhost' IDENTIFIED BY '%s'"  % (config.DBNAME, config.USERNAME, config.PASSWORD),
        )

        for num, sentence in enumerate(sentences):
            try:
                connection = dbi.connect(user=db_user, passwd=db_password)
                cursor = connection.cursor()
                cursor.execute(sentence)
                connection.commit()
                connection.close()
            except:
                if num != 0: # If user does not exist
                    raise

def create_db(db_user, db_password):
    if config.ENGINE == 'mysql':
        try:
            connection = dbi.connect(user=db_user, passwd=db_password, db = config.DBNAME, host = config.HOST)
        except:
            pass # DB does not exist
        else:
            cursor = connection.cursor()
            cursor.execute("DROP DATABASE IF EXISTS %s" % config.DBNAME)
            connection.commit()
            connection.close()

        connection = dbi.connect(user=db_user, passwd=db_password, host = config.HOST)
        cursor = connection.cursor()
        cursor.execute("CREATE DATABASE %s" % config.DBNAME)
        connection.commit()
        connection.close()

if __name__ == '__main__':
    parser = OptionParser()
    parser.add_option("-c", "--create-user",
                    action="store_true", dest="create_user", default=False,
                    help="Creates the user and password in the database first")
    parser.add_option("-d", "--create-db",
                    action="store_true", dest="create_db", default=False,
                    help="Creates the database")
    parser.add_option("-u", "--add-sample-users",
                    action="store_true", dest="add_sample_users", default=False,
                    help="Adds sample users")

    parser.add_option("--db-user",
                    dest="db_user", default="root",
                    help="Database user")
    parser.add_option("--db-password",
                    dest="db_password", default=None,
                    help="Database password")

    (options, args) = parser.parse_args()

    if config.ENGINE == 'mysql':
        db_user     = options.db_user
        db_password = options.db_password
        if db_password is None:
            db_password = getpass.getpass( "MySQL administrator password: " )
    else:
        db_user = db_password = ""

    if options.create_user:
        create_user(db_user, db_password)

    if options.create_db:
        create_db(db_user, db_password)

    if options.add_sample_users:
        add_sample_users()
    else:
        init_db(drop=True)