96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
import argparse
|
|
import os
|
|
import pkg_resources
|
|
import sys
|
|
|
|
DJANGO_MAIN_MODULE = 'main'
|
|
VERSION = '0.1'
|
|
|
|
|
|
class AdminCommand(object):
|
|
@staticmethod
|
|
def _setup_argparser():
|
|
kwargs = {
|
|
'description': 'Tool to manage your test django installation.',
|
|
}
|
|
parser = argparse.ArgumentParser(**kwargs)
|
|
|
|
parser.add_argument('-V', '--version', action='version',
|
|
version='%(prog)s ' + VERSION)
|
|
|
|
subparsers = parser.add_subparsers(dest='subcmd', metavar='CMD',
|
|
title='subcommands',
|
|
description="Use '%(prog)s CMD -h' to show help for a subcommand")
|
|
subparser = subparsers.add_parser('setup', help='Setup the installation.')
|
|
subparser.add_argument('path', metavar='PATH',
|
|
help='A directory, where the project files will be installed.')
|
|
|
|
return parser
|
|
|
|
def _parse_args(self, argv=None):
|
|
if argv is None:
|
|
argv = sys.argv[1:]
|
|
if not argv:
|
|
argv = ['--help']
|
|
|
|
return self._argparser.parse_args(argv)
|
|
|
|
@staticmethod
|
|
def _subcmd_setup(cmd_args):
|
|
django_main_module = DJANGO_MAIN_MODULE
|
|
django_base_dir = cmd_args.path
|
|
|
|
sys.stdout.write('Setup installation in \'{path}\'...\n'.format(path=django_base_dir))
|
|
|
|
if os.path.exists(django_base_dir):
|
|
if not os.path.isdir(django_base_dir):
|
|
sys.stderr.write('{path}: Not a directory.\n'.format(path=django_base_dir))
|
|
return os.EX_USAGE
|
|
else:
|
|
os.makedirs(django_base_dir)
|
|
|
|
sys.stdout.write('Creating django project...\n')
|
|
django_cmd = 'django-admin startproject {name} "{path}"'.format(name=django_main_module,
|
|
path=django_base_dir)
|
|
exitval = os.system(django_cmd)
|
|
if exitval != os.EX_OK:
|
|
return exitval
|
|
|
|
sys.stdout.write('Configure django project...\n')
|
|
|
|
input_file = os.path.join('django_project_config', 'additional_settings.py')
|
|
output_file = os.path.join(django_base_dir, django_main_module, 'settings.py')
|
|
with open(output_file, 'ab') as f:
|
|
f.write(pkg_resources.resource_string(__package__, input_file))
|
|
|
|
sys.stdout.write('Creating directories...\n')
|
|
dirs = [
|
|
os.path.join(django_base_dir, 'var', 'log'),
|
|
os.path.join(django_base_dir, 'var', 'www', 'static'),
|
|
]
|
|
|
|
for d in dirs:
|
|
sys.stdout.write(' - %s\n' % d)
|
|
os.makedirs(d)
|
|
|
|
return os.EX_OK
|
|
|
|
def __init__(self):
|
|
self._argparser = self._setup_argparser()
|
|
|
|
def __call__(self, argv=None):
|
|
cmd_args = self._parse_args(argv)
|
|
subcmd = cmd_args.subcmd
|
|
method_name = '_subcmd_{}'.format(subcmd)
|
|
method = getattr(self, method_name)
|
|
exitval = method(cmd_args)
|
|
return exitval
|
|
|
|
|
|
def main():
|
|
cmd = AdminCommand()
|
|
exitval = cmd()
|
|
sys.exit(exitval)
|