#!/usr/bin/env python

import os
import sys
import otree
from django.core.management import execute_from_command_line


STARTPROJECT = 'startproject'
STARTAPP = 'startapp'


if __name__ == "__main__":

    command = sys.argv[1]
    project_or_app_name = sys.argv[2]

    if command == STARTPROJECT:
        if project_or_app_name.lower() == 'otree':
            print 'Not an allowed project name.'
            sys.exit(1)
        project_template = os.path.join(os.path.dirname(otree.__file__), 'project_template.zip')
        argv = ['', STARTPROJECT, '--template={}'.format(project_template), project_or_app_name]
    elif command == STARTAPP:
        app_template = os.path.join(os.path.dirname(otree.__file__), 'app_template.zip')
        argv = ['', STARTAPP, '--template={}'.format(app_template), project_or_app_name]
    else:
        print 'Available commands are: {}'.format(', '.join([STARTAPP, STARTPROJECT]))
        sys.exit(1)

    argv += sys.argv[3:]  # Append any additional flags that were parsed.
    execute_from_command_line(argv)
    