mysql.py

"""
Scripts to manage MySQL users and databases.
"""

from .utils import confirm, DocOptArgs, entrypoint
from ..plumbing.common import Owner
from ..tasks import mysql


@entrypoint
def create(opts: DocOptArgs, owner: Owner):
    """
    Create a MySQL database for a member or society.

    A corresponding MySQL account will also be created if needed.

    Usage: {script} OWNER [SUFFIX]
    """
    name = mysql._user_name(owner)
    suffix = opts["SUFFIX"]
    print("User: {}".format(name))
    print("Databases: {}{}".format(name, ", {}/{}".format(name, suffix) if suffix else ""))
    confirm("Create these?")
    with mysql.context() as cursor:
        result = mysql.create_account(cursor, owner)
        passwd, database = result.value
        if passwd:
            print("Created account")
        if database:
            print("Created default database {!r}".format(database))
        if suffix:
            result = mysql.create_database(cursor, owner, suffix)
            if result:
                print("Created database {!r}".format(result.value))


@entrypoint
def passwd(owner: Owner):
    """
    Reset a MySQL user's password.

    Usage: {script} OWNER
    """
    name = mysql._user_name(owner)
    confirm("Reset {}'s password?".format(name))
    with mysql.context() as cursor:
        mysql.reset_password(cursor, owner)
        print("Password changed")


@entrypoint
def drop(owner: Owner):
    """
    Drop a MySQL user and all their databases.

    Usage: {script} OWNER
    """
    name = mysql._user_name(owner)
    with mysql.context() as cursor:
        dbs = mysql.get_owned_databases(cursor, owner)
    print("User: {}".format(name))
    if dbs:
        print("Databases: {}".format(", ".join(dbs)))
    confirm("Delete these?")
    with mysql.context() as cursor:
        if mysql.drop_all_databases(cursor, owner):
            print("Dropped databases")
        if mysql.drop_account(cursor, owner):
            print("Dropped user")