strabulous.py

"""A slimmed down version of something I use at work.

Mostly just some helpers to go with baste and fabric to manage a project.

TODO: yes, we could probably use gulp or some other fancy new fangled project,
but I'm not familiar with any of them so I probalby won't set it up. Feel free to.
"""

from subprocess import getoutput
import datetime
import os.path

from fabric.api import (
    local, settings, hosts, get, run, put, hide
)
from fabric import colors
from fabric.contrib.console import confirm

from baste import (
    project_relative,
    RsyncMedia,
    RsyncDeployment,
    UbuntuPgCreateDbAndUser,
    PgLoadPlain,
)


# -------------------------------------------------------------------------------
def update(all_repos, python_repos, python_package_name, python_version):
    """
    Update all of the dependencies to their latest versions.
    """
    for repo in list(all_repos.values()):
        repo.update()

    for repo in list(python_repos.values()):
        python_dependency(repo.name, python_version)
    python_dependency(python_package_name, python_version)


# -------------------------------------------------------------------------------
def createdb(database_name, database_user, get_password):
    if confirm(colors.red(
        "Show DB Password? You should only run this when others aren't looking over your shoulder. Run the command?")):
        print((colors.blue("Password: ") + colors.yellow(get_password())))
    if confirm(colors.red("This will overwrite local data, are you sure?")):
        UbuntuPgCreateDbAndUser(database_name, database_user)()


# -------------------------------------------------------------------------------
def latest_live_media(live_media_path, local_media_path, password_file_name, live_ssh_username,
                      live_domain):
    local_dir = local_media_path
    remote_dir = live_media_path
    if confirm(colors.red("This will overwrite local data, are you sure?")):
        RsyncMedia(
            '%s@%s' % (live_ssh_username, live_domain),
            remote_directory=remote_dir,
            local_directory=local_dir
        )()


# -------------------------------------------------------------------------------
def latest_live_db(live_backup_script_path, live_latest_sql_file_path, database_name, database_user,
                   get_password):
    if confirm(colors.red("This will overwrite local data, are you sure?")):
        if confirm(colors.red("Create a new backup?")):
            run(live_backup_script_path)
        local_target = project_relative("data/latestdb.sql.bz2")
        if confirm(colors.red("Download New backup?")):
            local_db = get(live_latest_sql_file_path, local_target)[0]
        else:
            local_db = local_target
        with settings(warn_only=True):
            if confirm(colors.red(
                "Show password? You should only run this when others aren't looking over your shoulder. Show database password?")):
                print((colors.blue("Password: ") + colors.yellow(get_password())))
            PgLoadPlain(local_db, database_name, database_user)()