-
Notifications
You must be signed in to change notification settings - Fork 0
/
django-dist.py
264 lines (225 loc) · 9.8 KB
/
django-dist.py
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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
# -*- coding: utf-8 -*-
from __future__ import unicode_literals, print_function
import os
import sys
try:
from jinja2 import Template
from termcolor import cprint
except ImportError:
print('jinja2, termcolor 라이브러리가 필요합니다.', file=sys.stderr)
sys.exit(1)
class Installer(object):
def __init__(self, django_settings_module):
self.project_root = os.path.abspath(os.path.dirname(__file__))
try:
module = __import__(django_settings_module)
for name in django_settings_module.split('.')[1:]:
module = getattr(module, name)
settings = module
except ImportError:
raise RuntimeError('원하는 버전의 Django 팩키지를 먼저 설치해주세요.')
self.req_path = os.path.join(self.project_root, 'requirements.txt')
self.kwargs = {
'project_name': django_settings_module.split('.', 1)[0],
'project_root': self.project_root,
'static_url_prefix': getattr(settings, 'STATIC_URL', '').rstrip('/'),
'static_root': getattr(settings, 'STATIC_ROOT', ''),
'media_url_prefix': getattr(settings, 'MEDIA_URL', '').rstrip('/'),
'media_root': getattr(settings, 'MEDIA_ROOT', ''),
'django_settings_module': django_settings_module,
'db_engine': settings.DATABASES['default']['ENGINE'],
'db_host': settings.DATABASES['default'].get('HOST', ''),
'db_name': settings.DATABASES['default']['NAME'],
'db_user': settings.DATABASES['default'].get('USER', ''),
'db_password': settings.DATABASES['default'].get('PASSWORD', ''),
'is_postgresql': False,
}
if not self.kwargs['static_url_prefix']:
raise RuntimeError('{}.STATIC_URL 을 설정해주세요. '
'ex) "/static/"'.format(django_settings_module))
if not self.kwargs['static_root']:
raise RuntimeError('{}.STATIC_ROOT 를 설정해주세요. '
'ex) os.path.join(BASE_DIR, "..", "staticfiles")'.format(django_settings_module))
if not self.kwargs['media_url_prefix']:
raise RuntimeError('{}.MEDIA_URL 을 설정해주세요. '
'ex) "/media/"'.format(django_settings_module))
if not self.kwargs['media_root']:
raise RuntimeError('{}.MEDIA_ROOT 를 설정해주세요. '
'ex) os.path.join(BASE_DIR, "..", "media")'.format(django_settings_module))
if os.environ['USER'] != 'root':
raise RuntimeError('root 권한으로 실행해주세요.')
if not os.path.exists('/etc/apt/sources.list.d/'):
raise RuntimeError('ubuntu os 만을 지원합니다.')
if not os.path.exists(os.path.join(self.project_root, 'manage.py')):
raise RuntimeError('django project root 로 옮겨주세요.')
if not os.path.exists(self.req_path):
raise RuntimeError('requirements.txt 를 생성해주세요.')
for name in ('postgresql', 'psycopg2', 'postgis'):
if name in self.kwargs['db_engine']:
self.kwargs['is_postgresql'] = True
break
else:
raise RuntimeError('PostgreSQL 데이터베이스만 지원합니다.')
def run(self):
self.init_apt()
self.install_system_packages()
self.init_db()
self.install_python_packages()
self.make_logs_directory()
self.init_django_project()
self.init_nginx()
self.init_uwsgi()
self.service_start()
def init_apt(self):
'Ubuntu 14.04 에서는 PostgreSQL 9.3 이 설치되므로, 상위버전을 설치하기 위한 추가 세팅'
APT_PGDG_PATH = '/etc/apt/sources.list.d/pgdg.list'
if not os.path.exists(APT_PGDG_PATH):
with open(APT_PGDG_PATH, 'a', encoding='utf8') as f:
f.write('\n')
f.write('deb http://apt.postgresql.org/pub/repos/apt/ precise-pgdg main')
self.command_run('wget --quiet -O - https://www.postgresql.org/media/keys/ACCC4CF8.asc | sudo apt-key add -')
def install_system_packages(self):
'시스템 팩키지 설치'
self.command_run('''
sudo apt-get update && \
sudo apt-get install -y git python3-pip python3-dev \
libpq-dev postgresql postgresql-contrib \
nginx \
libjpeg8-dev \
libpcre3 libpcre3-dev
''')
def init_db(self):
'Django 프로젝트 데이터베이스/계정 생성 및 암호 설정'
if self.kwargs['is_postgresql']:
for key in ('db_host', 'db_name', 'db_user', 'db_password'):
if not self.kwargs[key]:
raise RuntimeError('{} 설정을 해주세요.'.format(key))
if self.kwargs['db_host'] == '127.0.0.1':
# peer
self.command_run('''
sudo -u postgres dropdb --if-exists {db_name}
sudo -u postgres dropuser --if-exists {db_user}
sudo -u postgres createdb {db_name} && \
sudo -u postgres createuser {db_user} && \
echo "ALTER ROLE {db_user} WITH PASSWORD '{db_password}';" | sudo -u postgres psql
''')
else:
self.command_run('''
sudo -u postgres dropdb --if-exists --host={db_host} {db_name}
sudo -u postgres dropuser --if-exists --host={db_host} {db_user}
sudo -u postgres createdb --host={db_host} {db_name} && \
sudo -u postgres createuser --host={db_host} {db_user} && \
echo "ALTER ROLE {db_user} WITH PASSWORD '{db_password}';" | sudo -u postgres psql --host={db_host}
''')
def install_python_packages(self):
'Django 프로젝트, 파이썬 팩키지 설치'
self.command_run('sudo pip3 install -r {}'.format(self.req_path))
def make_logs_directory(self):
'Django 프로젝트 로그 디렉토리 생성'
self.command_run('''
sudo mkdir -p /var/log/{project_name} && \
sudo chown www-data:www-data /var/log/{project_name}
''')
def init_django_project(self):
self.command_run('''
python3 manage.py collectstatic --noinput --settings={django_settings_module} && \
python3 manage.py makemigrations --settings={django_settings_module} && \
python3 manage.py migrate --settings={django_settings_module}
''')
def init_nginx(self):
self.command_run('''
sudo mkdir -p {media_root} && \
sudo chown www-data:www-data {media_root}
''')
config = Template(self.NGINX_CONFIG_TEMPLATE).render(**self.kwargs)
open('/etc/nginx/nginx.conf', 'w', encoding='utf8').write(config)
def init_uwsgi(self):
config = Template(self.UWSGI_CONFIG_TEMPLATE).render(**self.kwargs)
open('/etc/init/{project_name}.conf'.format(**self.kwargs), 'w', encoding='utf8').write(config)
def service_start(self):
self.command_run('''
sudo service {project_name} start && \
sudo service nginx restart
''')
def command_run(self, command):
if isinstance(command, (list, tuple)):
command = '\n'.join(command)
command = command.format(**self.kwargs)
print()
for line in command.splitlines():
line = ' '.join(line.strip().split())
cprint(line, 'green', attrs=['bold'])
print()
result = os.system(command)
if result != 0:
raise RuntimeError('명령 수행 실패')
NGINX_CONFIG_TEMPLATE = '''
user www-data;
worker_processes 1;
pid /run/nginx.pid;
events {
worker_connections 1024;
}
http {
sendfile on;
include /etc/nginx/mime.types;
default_type application/octet-stream;
log_format main '$remote_addr - $remote_user [$time_local] "$request" '
'$status $body_bytes_sent "$http_referer" '
'"$http_user_agent" "$http_x_forwarded_for"';
server {
listen 80;
client_max_body_size 3M;
keepalive_timeout 5;
root {{ project_root }};
access_log /var/log/{{ project_name }}/nginx_access.log main;
rewrite ^/robots.txt(/?)+$ /static/robots.txt last;
location {{ static_url_prefix }} {
expires 1m;
autoindex off;
alias {{ static_root }};
}
location {{ media_url_prefix }} {
expires 1m;
autoindex off;
alias {{ media_root }};
}
location / {
expires epoch;
include uwsgi_params;
uwsgi_pass 127.0.0.1:8080;
uwsgi_param UWSGI_SCHEME $scheme;
}
}
}
'''
UWSGI_CONFIG_TEMPLATE = '''
description "{{ project_name }} uwsgi service"
start on runlevel [2345]
stop on runlevel [016]
respawn
env DJANGO_HOME={{ project_root }}
env DJANGO_SETTINGS_MODULE={{ django_settings_module }}
exec uwsgi --master \
--die-on-term \
--processes 1 \
--socket :8080 \
--harakiri 30 \
--harakiri-verbose \
--reload-on-rss 100 \
--logto /var/log/{{ project_name }}/uwsgi.log \
--pythonpath $DJANGO_HOME \
--wsgi-file $DJANGO_HOME/{{ project_name }}/wsgi.py \
--uid www-data --gid www-data
'''
if __name__ == '__main__':
try:
django_settings_module = sys.argv[1]
except IndexError:
cprint('Usage) {} <django_settings_module>'.format(sys.argv[0]), 'red', file=sys.stderr)
sys.exit(1)
try:
Installer(django_settings_module).run()
except RuntimeError as e:
cprint('Error) {}'.format(e), 'red', file=sys.stderr)
sys.exit(1)