1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
from __future__ import print_function
import subprocess
import contextlib
import os
from os import path
import argparse
from importlib import import_module
from django.conf import settings
from django.core import management
from django.template import Template, Context, loader
from project_detail import NAME, PATH
CUR_DIR = os.getcwd()
SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
PARENT_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir))
TEMPLATE_DIR = path.join(PARENT_DIR, 'templates')
def main():
#Parse command-line to obtain the arguments and/or options
# create top-level parser object
parser = argparse.ArgumentParser(prog="vimarsh")
subparser = parser.add_subparsers(dest="subcommand")
# create parser for the "create_demo" subcommand
create_demo_parser = subparser.add_parser("create_demo",
help="Create a new demo Django project")
create_demo_parser.add_argument("project_name", type=str,
help="name of demo Django project")
create_demo_parser.add_argument("-p", "--path", type=str,
help="path of demo Django project")
# create parser for the "run_demo" subcommand
run_demo_parser = subparser.add_parser("run_demo",
help="Initialise django server and run the demo project")
# create parser for the "run_code_server" subcommand
code_server_parser = subparser.add_parser("run_code_server",
help="Initialise Vimarsh code server")
code_server_parser.add_argument("-P", "--ports", type=int, nargs='+',
help="code server ports")
args = parser.parse_args()
if args.subcommand == "create_demo":
if args.path:
create_demo(args.project_name, args.path)
else:
create_demo(args.project_name)
elif args.subcommand == "run_demo":
try:
run_demo(NAME, PATH)
except Exception as e:
if not NAME or not PATH:
print("Error: Unable to find Project Name or Path variables\n")
else:
print("Error: {0}\n".format(e))
subparser.print_help()
elif args.subcommand == "run_code_server":
if args.ports:
run_server(args.ports)
else:
run_server()
def create_demo(project_name='vimarsh_demo', project_dir=CUR_DIR):
try:
management.call_command('startproject', project_name, project_dir)
print("Demo Django project '{0}' created at '{1}'".format(project_name,
project_dir))
except Exception, e:
print("Error: {0}\nExiting Vimarsh Installer".format(e))
if project_dir is None:
top_dir = path.join(os.getcwd(), project_name)
else:
top_dir = project_dir
project_path = path.join(top_dir, project_name)
# Store project details
_set_project_details(project_name, top_dir)
with _chdir(project_path):
root_urlconf = "{0}.{1}".format(project_name, 'demo_urls')
settings_template_path = path.join(TEMPLATE_DIR, 'demo_settings.py')
settings_target_path = path.join(project_path, 'demo_settings.py')
settings_context = Context({'project_name': project_name, # 'custom_apps': custom_apps,
'root_urlconf': root_urlconf})
urls_template_path = path.join(TEMPLATE_DIR, 'demo_urls.py')
urls_target_path = path.join(project_path, 'demo_urls.py')
command = ("python ../manage.py syncdb "
"--noinput --settings={0}.demo_settings").format(project_name)
# Create demo_settings file
_render_demo_files(settings_template_path, settings_target_path, settings_context)
# Create demo_urls file
_render_demo_files(urls_template_path, urls_target_path)
# Run syncdb
subprocess.call(command, shell=True)
def run_demo(project_name, top_dir):
with _chdir(top_dir):
project_path = path.join(top_dir, 'manage.py')
command = ("python manage.py runserver "
"--settings={0}.demo_settings").format(project_name)
subprocess.call(command, shell=True)
def run_server():
try:
from testapp.exam import code_server
code_server.main()
except Exception as e:
print("Error: {0}\nExiting Vimarsh code server".format(e))
def _set_project_details(project_name, top_dir):
file_path = path.join(SCRIPT_DIR, 'project_detail.py')
detail = "NAME ='{0}'\nPATH ='{1}'".format(project_name, top_dir)
with open(file_path, 'w') as data_store:
data_store.write(detail)
def _render_demo_files(template_path, output_path, context=None):
with open(template_path, 'r') as template_file:
content = template_file.read()
if context:
content = content.decode('utf-8')
template = Template(content)
content = template.render(context)
content = content.encode('utf-8')
with open(output_path, 'w') as new_file:
new_file.write(content)
@contextlib.contextmanager
def _chdir(path):
starting_directory = os.getcwd()
try:
os.chdir(path)
yield
finally:
os.chdir(starting_directory)
if __name__ == '__main__':
main()
|
