django_shortcuts.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import sys
from subprocess import call

ALIASES = {
    # Django
    'c': 'collectstatic --noinput --link',
    'cc': 'collectstatic --noinput',
    #'r'  : 'runserver',
    'sd': 'syncdb',
    'sp': 'startproject',
    'sa': 'startapp',
    't': 'test',

    # Shell
    'd': 'dbshell',
    's': 'shell',

    # Auth
    'csu': 'createsuperuser',
    'cpw': 'changepassword',

    # Migrations
    'm': 'migrate',
    'mm': 'makemigrations',

    # Haystack
    'ix': 'update_index',
    'rix': 'rebuild_index',

    # Django Extensions
    'sk': 'generate_secret_key',
    'rdb': 'reset_db',
    'r': 'runserver_plus',
    'shp': 'shell_plus',
    'url': 'show_urls',
    'gm': 'graph_models',
    'rs': 'runscript',

    # Django Bower
    'bower': 'bower_install -- --allow-root',
}


def print_aliases():
    aliases_list = sorted(ALIASES.items())  # Out of the box sorting works in that case.
    print 'Available aliases:'

    padding = max([len(alias) for alias, _ in aliases_list])

    for alias in aliases_list:
        print ('\t\033[32m{: <%d}\033[0m   →   {}' % padding).format(*alias)


def run(command=None, *arguments):
    """
    Run the given command.

    Parameters:
    :param command: A string describing a command.
    :param arguments: A list of strings describing arguments to the command.
    """

    if command is None:
        print_aliases()
        sys.exit('django-shortcuts: No argument was supplied, please specify one.')

    if command in ALIASES:
        command = ALIASES[command]

    if command == 'startproject':
        return call('django-admin.py startproject %s' % ' '.join(arguments), shell=True)

    script_path = os.getcwd()
    while not os.path.exists(os.path.join(script_path, 'manage.py')):
        base_dir = os.path.dirname(script_path)
        if base_dir != script_path:
            script_path = base_dir
        else:
            sys.exit('django-shortcuts: No \'manage.py\' script found in this directory or its parents.')

    return call('%(python)s %(script_path)s %(command)s %(arguments)s' % {
        'python': sys.executable,
        'script_path': os.path.join(script_path, 'manage.py'),
        'command': command or '',
        'arguments': ' '.join(arguments)
    }, shell=True)


def main():
    """Entry-point function."""
    sys.exit(run(*sys.argv[1:]))

if __name__ == '__main__':
    main()