Minor fix.
[fnpdjango.git] / fnpdjango / deploy / __init__.py
1 """
2 Generic fabric deployment script.
3 Create a fabfile.py in the project and start it with:
4
5     from fnpdjango.deploy import *
6
7 Then set up some env properties:
8     project_name: slug-like project name
9     hosts: list of target host names
10     user: remote user name
11     app_path: where does the app go
12     services: list of tasks to run after deployment
13     django_root_path (optional): path to the directory
14         containing django project, relative to the
15         root of the repository (defaults to '.')
16     localsettings_dst_path (optional): path indicating
17         where to copy the localsettings file, relative
18         to django_root_path (defaults to project_name/localsettings.py)
19     skip_collect_static (optional): if True, Django collectstatic command is not called
20 """
21 from os.path import abspath, dirname, exists, join
22 from django.utils.crypto import get_random_string
23 from fabric.api import *
24 from fabric.contrib import files
25 from fabric.tasks import Task, execute
26
27 env.virtualenv = '/usr/bin/virtualenv'
28 env.services = None
29
30
31 @task
32 def setup():
33     """
34     Setup all needed directories.
35     """
36     require('hosts', 'app_path')
37
38     if not files.exists(env.app_path):
39         run('mkdir -p %(app_path)s' % env, pty=True)
40     with cd(env.app_path):
41         for subdir in 'releases', 'packages', 'log', 'samples':
42             if not files.exists(subdir):
43                 run('mkdir -p %s' % subdir, pty=True)
44     with cd('%(app_path)s/releases' % env):
45         if not files.exists('current'):
46             run('ln -sfT . current', pty=True)
47         if not files.exists('previous'):
48             run('ln -sfT . previous', pty=True)
49     
50     upload_samples()
51
52
53 def check_localsettings():
54     return files.exists('%(app_path)s/localsettings.py' % env)
55
56
57 @task(default=True)
58 def deploy():
59     """
60     Deploy the latest version of the site to the servers,
61     install any required third party modules,
62     install the virtual host and then restart the webserver
63     """
64     require('hosts', 'app_path')
65
66     import time
67     env.release = time.strftime('%Y-%m-%dT%H%M')
68
69     setup()
70     if not check_localsettings():
71         abort('Setup is complete, but\n    %(app_path)s/localsettings.py\n'
72               'is needed for actual deployment.' % env)
73     upload_tar_from_git()
74     copy_localsettings()
75     install_requirements()
76     symlink_current_release()
77     migrate()
78     pre_collectstatic()
79     collectstatic()
80     restart()
81
82 @task
83 def rollback():
84     """
85     Limited rollback capability. Simple loads the previously current
86     version of the code. Rolling back again will swap between the two.
87     Warning: this will almost certainly go wrong, it there were any migrations
88     in the meantime!
89     """
90     require('hosts', 'app_path')
91     with cd(env.path):
92         run('mv releases/current releases/_previous;', pty=True)
93         run('mv releases/previous releases/current;', pty=True)
94         run('mv releases/_previous releases/previous;', pty=True)
95     collectstatic()
96     restart()
97
98 @task
99 def deploy_version(version):
100     """
101     Loads the specified version.
102     Warning: this will almost certainly go wrong, it there were any migrations
103     in the meantime!
104     """
105     "Specify a specific version to be made live"
106     require('hosts', 'app_path')
107     env.version = version
108     with cd(env.path):
109         run('rm releases/previous; mv releases/current releases/previous;', pty=True)
110         run('ln -s %(version)s releases/current' % env, pty=True)
111     collectstatic()
112     restart()
113
114 @task
115 def restart():
116     require('services')
117     for service in env.services or ():
118         execute(service)
119
120
121 # =====================================================================
122 # = Helpers. These are called by other functions rather than directly =
123 # =====================================================================
124
125 class Service(Task):
126     def upload_sample(self):
127         pass
128
129 class DebianGunicorn(Service):
130     def __init__(self, name):
131         super(Task, self).__init__()
132         self.name = name
133
134     def run(self):
135         print '>>> restart webserver using gunicorn-debian'
136         sudo('gunicorn-debian restart %s' % self.name, shell=False)
137
138     def upload_sample(self):
139         upload_sample('gunicorn')
140
141 class Apache(Service):
142     def run(self):
143         print '>>> restart webserver by touching WSGI'
144         with path('/sbin'):
145             run('touch %(app_path)s/%(project_name)s/wsgi.py' % env)
146
147 class Supervisord(Service):
148     def __init__(self, name):
149         super(Task, self).__init__()
150         self.name = name
151
152     def run(self):
153         print '>>> supervisord: restart %s' % self.name
154         sudo('supervisorctl restart %s' % self.name, shell=False)
155
156 class Command(Task):
157     def __init__(self, commands, working_dir):
158         if not hasattr(commands, '__iter__'):
159             commands = [commands]
160         self.name = 'Command: %s @ %s' % (commands, working_dir)
161         self.commands = commands
162         self.working_dir = working_dir
163
164     def run(self):
165         require('app_path')
166         with cd(join('%(app_path)s/releases/current' % env, self.working_dir)):
167             for command in self.commands:
168                 run(command)
169
170 def upload_samples():
171     upload_localsettings_sample()
172     upload_nginx_sample()
173     for service in env.services:
174         service.upload_sample()
175
176 def upload_sample(name, where="samples/"):
177     require('app_path', 'project_name')
178     upload_path = '%s/%s%s.sample' % (env['app_path'], where, name)
179     if files.exists(upload_path):
180         return
181     print '>>> upload %s template' % name
182     template = '%(project_name)s/' % env + name + '.template'
183     if not exists(template):
184         template = join(dirname(abspath(__file__)), 'templates/' + name + '.template')
185     files.upload_template(template, upload_path, env)
186
187 def upload_localsettings_sample():
188     "Fill out localsettings template and upload as a sample."
189     env.secret_key = get_random_string(50)
190     upload_sample('localsettings.py', where="")
191
192 upload_nginx_sample = lambda: upload_sample('nginx')
193
194 def upload_tar_from_git():
195     "Create an archive from the current Git branch and upload it"
196     print '>>> upload tar from git'
197     require('release', provided_by=[deploy])
198     require('app_path')
199     local('git-archive-all.sh --format tar %(release)s.tar' % env)
200     local('gzip %(release)s.tar' % env)
201     run('mkdir -p %(app_path)s/releases/%(release)s' % env, pty=True)
202     run('mkdir -p %(app_path)s/packages' % env, pty=True)
203     put('%(release)s.tar.gz' % env, '%(app_path)s/packages/' % env)
204     run('cd %(app_path)s/releases/%(release)s && tar zxf ../../packages/%(release)s.tar.gz' % env, pty=True)
205     local('rm %(release)s.tar.gz' % env)
206
207 def install_requirements():
208     "Install the required packages from the requirements file using pip"
209     print '>>> install requirements'
210     require('release', provided_by=[deploy])
211     require('app_path')
212     if not files.exists('%(app_path)s/ve' % env):
213         require('virtualenv')
214         run('%(virtualenv)s %(app_path)s/ve' % env, pty=True)
215     with cd('%(app_path)s/releases/%(release)s' % env):
216         run('%(app_path)s/ve/bin/pip install -r requirements.txt' % env, pty=True)
217     with cd(get_django_root_path(env['release'])):
218         # Install DB requirement
219         database_reqs = {
220             'django.db.backends.postgresql_psycopg2': 'psycopg2',
221             'django.db.backends.mysql': 'MySQL-python',
222         }
223         databases = run('''DJANGO_SETTINGS_MODULE=%(project_name)s.settings %(app_path)s/ve/bin/python -c 'from django.conf import settings;              print " ".join(set([d["ENGINE"] for d in settings.DATABASES.values()]))' ''' % env)
224         for database in databases.split():
225             if database in database_reqs:
226                 # TODO: set pip default pypi
227                 run('%(app_path)s/ve/bin/pip install ' % env + database_reqs[database])
228
229
230 def copy_localsettings():
231     "Copy localsettings.py from root directory to release directory (if this file exists)"
232     print ">>> copy localsettings"
233     require('release', provided_by=[deploy])
234     require('app_path', 'project_name')
235
236     with settings(warn_only=True):
237         copy_to = join(get_django_root_path(env['release']), env.get('localsettings_dst_path', env['project_name']))
238         run('cp %(app_path)s/localsettings.py ' % env + copy_to)
239
240 def symlink_current_release():
241     "Symlink our current release"
242     print '>>> symlink current release'
243     require('release', provided_by=[deploy])
244     require('app_path')
245     with cd(env.app_path):
246         run('rm releases/previous; mv releases/current releases/previous')
247         run('ln -s %(release)s releases/current' % env)
248
249 def migrate():
250     "Update the database"
251     print '>>> migrate'
252     require('app_path', 'project_name')
253     with cd(get_django_root_path('current')):
254         run('%(app_path)s/ve/bin/python manage.py syncdb --noinput' % env, pty=True)
255         run('%(app_path)s/ve/bin/python manage.py migrate' % env, pty=True)
256
257 def pre_collectstatic():
258     print '>>> pre_collectstatic'
259     for task in env.get('pre_collectstatic', []):
260         execute(task)
261
262 def collectstatic():
263     """Collect static files"""
264     print '>>> collectstatic'
265     if env.get('skip_collect_static', False):
266         print '... skipped'
267         return
268     require('app_path', 'project_name')
269     with cd(get_django_root_path('current')):
270         run('%(app_path)s/ve/bin/python manage.py collectstatic --noinput' % env, pty=True)
271
272
273 def get_django_root_path(release):
274     require('app_path')
275     path = '%(app_path)s/releases/%(release)s' % dict(app_path = env['app_path'], release = release)
276     if 'django_root_path' in env:
277         path = join(path, env['django_root_path'])
278     return path