mysql_password.py

import os

from contextlib import contextmanager
from fabric.api import env, hide, local


@contextmanager
def mysql_password(database):
    contents = """
    [client]
    user={}
    password={}
    """.format(database['USER'], database['PASSWORD'])

    credentials_file = os.path.join(
        os.path.dirname(os.path.realpath(__file__)),
        '.credentials.tmp')

    local('touch {}'.format(credentials_file))
    local('chmod 600 {}'.format(credentials_file))

    with hide('running'):
        local('echo "{}" > {}'.format(contents, credentials_file))

    env['credentials_file'] = credentials_file

    try:
        yield
    finally:
        local('rm {}'.format(credentials_file))

        del env['credentials_file']