Commit f1979c278c5e22227b9daa49249395ea1299362e
1 parent
8089173b
Exists in
master
and in
39 other branches
Removed fabfile
Showing
1 changed file
with
0 additions
and
338 deletions
Show diff stats
fabfile.py
... | ... | @@ -1,338 +0,0 @@ |
1 | -# coding: utf-8 | |
2 | - | |
3 | -import os | |
4 | - | |
5 | -from fabric import colors | |
6 | -from fabric.utils import error | |
7 | -from fabric.decorators import task | |
8 | -from fabric.api import env, run, sudo, local | |
9 | -from fabric.contrib.files import exists | |
10 | -from fabric.context_managers import prefix, cd, settings, shell_env | |
11 | - | |
12 | -DEBIAN_FAMILY = ['debian', 'ubuntu'] | |
13 | -REDHAT_FAMILY = ['centos', 'fedora'] | |
14 | - | |
15 | -DISTRO_CMD = { | |
16 | - 'debian': ('apt-get', { | |
17 | - 'install': '-y', | |
18 | - 'update': '-y', | |
19 | - }), | |
20 | - 'redhat': ('yum', { | |
21 | - 'install': '-y', | |
22 | - 'update': '-y', | |
23 | - }) | |
24 | -} | |
25 | - | |
26 | -APP_USER = APP_NAME = VENV_NAME = 'colab' | |
27 | -REPO_URL = 'https://github.com/colab-community/colab.git' | |
28 | -REPO_BRANCH = 'master' | |
29 | - | |
30 | - | |
31 | -environments = { | |
32 | - 'dev': { | |
33 | - 'hosts': ['127.0.0.1'], | |
34 | - 'key_filename': '~/.vagrant.d/insecure_private_key', | |
35 | - 'port': 2222, | |
36 | - 'is_vagrant': True, | |
37 | - 'superuser': 'vagrant', | |
38 | - }, | |
39 | -} | |
40 | -DEFAULT_ENVIRONMENT = 'dev' | |
41 | - | |
42 | -env.user = APP_USER | |
43 | -env.use_shell = False | |
44 | - | |
45 | -PROJECT_PATH = os.path.join(os.path.dirname(__file__)) | |
46 | -REPO_PATH = '/home/{}/{}'.format(APP_USER, APP_NAME) | |
47 | -SOURCE_VENV = 'source /usr/local/bin/virtualenvwrapper.sh' | |
48 | - | |
49 | -WORKON_ENV = '{} && workon {}'.format(SOURCE_VENV, VENV_NAME) | |
50 | -MANAGE_PATH = os.path.join(REPO_PATH, 'src') | |
51 | -SETTINGS_PATH = os.path.join(MANAGE_PATH, APP_NAME) | |
52 | - | |
53 | - | |
54 | -def get_distro_family(): | |
55 | - cmd = 'python -c "import platform; print platform.dist()[0]"' | |
56 | - linux_name = run(cmd).lower() | |
57 | - if linux_name in DEBIAN_FAMILY: | |
58 | - return 'debian' | |
59 | - elif linux_name in REDHAT_FAMILY: | |
60 | - return 'redhat' | |
61 | - else: | |
62 | - error(colors.red('Distribuiton `{}` not supported'.format(linux_name))) | |
63 | - exit(1) | |
64 | - | |
65 | - | |
66 | -def cmd(family, command, args=''): | |
67 | - pkgmanager, commands = DISTRO_CMD[family] | |
68 | - return ' '.join([pkgmanager, command, commands[command], args]) | |
69 | - | |
70 | - | |
71 | -@task | |
72 | -def environment(name=DEFAULT_ENVIRONMENT): | |
73 | - """Set the environment where the tasks will be executed""" | |
74 | - global REPO_URL | |
75 | - global REPO_BRANCH | |
76 | - | |
77 | - try: | |
78 | - import project_cfg | |
79 | - except ImportError: | |
80 | - pass | |
81 | - else: | |
82 | - REPO_URL = project_cfg.repository_url | |
83 | - REPO_BRANCH = project_cfg.branch | |
84 | - environments.update(project_cfg.environments) | |
85 | - | |
86 | - if name not in environments: | |
87 | - error(colors.red('Environment `{}` does not exist.'.format(name))) | |
88 | - | |
89 | - env.update(environments[name]) | |
90 | - env.environment = name | |
91 | - | |
92 | - | |
93 | -def package_install(pkg): | |
94 | - family = get_distro_family() | |
95 | - sudo(cmd(family, 'install', pkg)) | |
96 | - | |
97 | - | |
98 | -def install_requirements(): | |
99 | - with cd(REPO_PATH), prefix(WORKON_ENV): | |
100 | - run('pip install -U distribute') | |
101 | - if not env.is_vagrant: | |
102 | - run('pip install -r requirements.txt') | |
103 | - return | |
104 | - | |
105 | - if exists('requirements-{}.txt'.format(env.environment)): | |
106 | - run('pip install -r requirements-{}.txt'.format(env.environment)) | |
107 | - else: | |
108 | - run('pip install -r requirements.txt') | |
109 | - | |
110 | - | |
111 | -def mkvirtualenv(): | |
112 | - if not exists('~/.virtualenvs/' + VENV_NAME): | |
113 | - | |
114 | - with prefix(SOURCE_VENV): | |
115 | - run('mkvirtualenv ' + VENV_NAME) | |
116 | - return True | |
117 | - | |
118 | - | |
119 | -def manage(command): | |
120 | - django_settings = env.get('django_settings') | |
121 | - env_vars = {} | |
122 | - if django_settings: | |
123 | - env_vars.update({'DJANGO_SETTINGS_MODULE': django_settings}) | |
124 | - | |
125 | - with shell_env(**env_vars): | |
126 | - with cd(MANAGE_PATH), prefix(WORKON_ENV): | |
127 | - run('python manage.py {}'.format(command)) | |
128 | - | |
129 | - | |
130 | -def migrate(): | |
131 | - manage('migrate') | |
132 | - manage('loaddata super_archives/fixture/initial_data.json') | |
133 | - | |
134 | - | |
135 | -def collectstatic(): | |
136 | - sudo('mkdir -p /usr/share/nginx/{}'.format(APP_NAME)) | |
137 | - sudo('chown {} /usr/share/nginx/{}'.format(env.user, APP_NAME)) | |
138 | - manage('collectstatic --noinput') | |
139 | - | |
140 | - | |
141 | -def create_local_settings(): | |
142 | - with cd(SETTINGS_PATH): | |
143 | - env_local_settings = 'local_settings-{}.py'.format(env.environment) | |
144 | - | |
145 | - if not exists('local_settings.py') and exists(env_local_settings): | |
146 | - run('ln -s {} {}'.format(env_local_settings, 'local_settings.py')) | |
147 | - run('chown {} local_settings.py'.format(env.user)) | |
148 | - | |
149 | - | |
150 | -def update_code(): | |
151 | - if env.is_vagrant: | |
152 | - if not exists(REPO_PATH): | |
153 | - run('ln -s /vagrant/ {}'.format(REPO_PATH)) | |
154 | - return | |
155 | - | |
156 | - if not exists(REPO_PATH): | |
157 | - run('git clone {} {} -b {}'.format(REPO_URL, REPO_PATH, REPO_BRANCH)) | |
158 | - else: | |
159 | - with cd(REPO_PATH): | |
160 | - run('git checkout {}'.format(REPO_BRANCH)) | |
161 | - run('git pull') | |
162 | - | |
163 | - | |
164 | -@task | |
165 | -def bootstrap(): | |
166 | - """Bootstrap machine to run fabric tasks""" | |
167 | - | |
168 | - with settings(user=env.superuser): | |
169 | - family = get_distro_family() | |
170 | - if(family == 'debian'): | |
171 | - sudo(cmd(family, 'update')) | |
172 | - else: | |
173 | - package_install('wget') | |
174 | - | |
175 | - if not exists('/usr/bin/git'): | |
176 | - package_install('git-core') | |
177 | - | |
178 | - if env.is_vagrant: | |
179 | - groups = ['sudo', 'vagrant'] | |
180 | - local('chmod -fR g+w {}'.format(PROJECT_PATH)) | |
181 | - else: | |
182 | - groups = ['sudo'] | |
183 | - | |
184 | - for group in groups: | |
185 | - sudo('groupadd -f {}'.format(group)) | |
186 | - | |
187 | - command = 'useradd {} -G {} -m -s /bin/bash' | |
188 | - sudo(command.format(APP_USER, ','.join(groups))) | |
189 | - | |
190 | - ssh_dir = '/home/{0}/.ssh/'.format(APP_USER) | |
191 | - if not exists(ssh_dir): | |
192 | - sudo('mkdir -p {0}'.format(ssh_dir)) | |
193 | - sudo('chmod 700 {0}'.format(ssh_dir)) | |
194 | - sudo('cp ~{}/.ssh/authorized_keys /home/{}/.ssh/'.format( | |
195 | - env.superuser, | |
196 | - APP_USER | |
197 | - )) | |
198 | - sudo('chown -fR {0}:{0} {1}'.format(APP_USER, ssh_dir)) | |
199 | - | |
200 | - sudoers_file = os.path.join('/etc/sudoers.d/', APP_USER) | |
201 | - tmp_file = os.path.join('/tmp', APP_USER) | |
202 | - if not exists(sudoers_file): | |
203 | - sudo('echo "{} ALL=NOPASSWD: ALL" > {}'.format(APP_USER, tmp_file)) | |
204 | - sudo('chown root:root {}'.format(tmp_file)) | |
205 | - sudo('chmod 440 {}'.format(tmp_file)) | |
206 | - sudo('mv {} {}'.format(tmp_file, sudoers_file)) | |
207 | - | |
208 | - | |
209 | -@task | |
210 | -def provision(): | |
211 | - """Run puppet""" | |
212 | - | |
213 | - update_code() | |
214 | - | |
215 | - puppet_path = os.path.join(REPO_PATH, 'puppet/') | |
216 | - modules_path = os.path.join(puppet_path, 'modules') | |
217 | - puppet_modules = '{}:/etc/puppet/modules'.format(modules_path) | |
218 | - | |
219 | - with cd(puppet_path): | |
220 | - run('sudo python bootstrap.py') | |
221 | - | |
222 | - if env.is_vagrant: | |
223 | - cmd = os.path.join(puppet_path, 'manifests', 'site.pp') | |
224 | - else: | |
225 | - cmd = '-e "include {}"'.format(APP_NAME) | |
226 | - | |
227 | - if not exists('/usr/bin/puppet'): | |
228 | - print(colors.red('Please install `puppet` before continue.')) | |
229 | - return | |
230 | - | |
231 | - sudo('puppet apply --modulepath={} {}'.format(puppet_modules, cmd)) | |
232 | - | |
233 | - | |
234 | -@task | |
235 | -def ssh_keygen(): | |
236 | - """Create SSH credentials""" | |
237 | - | |
238 | - if not exists('~/.ssh/id_rsa'): | |
239 | - run("ssh-keygen -f ~/.ssh/id_rsa -N '' -b 1024 -q") | |
240 | - key = run('cat ~/.ssh/id_rsa.pub') | |
241 | - | |
242 | - print('Public key:') | |
243 | - print(colors.yellow(key)) | |
244 | - print('') | |
245 | - print('Add the key above to your github repository deploy keys') | |
246 | - | |
247 | - | |
248 | -@task | |
249 | -def deploy(noprovision=False): | |
250 | - """Deploy and run the new code (master branch)""" | |
251 | - | |
252 | - if noprovision is False: | |
253 | - provision() | |
254 | - else: | |
255 | - update_code() | |
256 | - | |
257 | - fix_path() | |
258 | - | |
259 | - install_solr() | |
260 | - | |
261 | - mkvirtualenv() | |
262 | - | |
263 | - sudo('supervisorctl stop all') | |
264 | - | |
265 | - install_requirements() | |
266 | - create_local_settings() | |
267 | - collectstatic() | |
268 | - migrate() | |
269 | - | |
270 | - build_schema() | |
271 | - | |
272 | - sudo('supervisorctl start all') | |
273 | - | |
274 | - | |
275 | -def fix_path(): | |
276 | - global SOURCE_VENV | |
277 | - global WORKON_ENV | |
278 | - if exists('/usr/bin/virtualenvwrapper.sh'): | |
279 | - SOURCE_VENV = 'source /usr/bin/virtualenvwrapper.sh' | |
280 | - WORKON_ENV = '{} && workon {}'.format(SOURCE_VENV, VENV_NAME) | |
281 | - | |
282 | - | |
283 | -@task | |
284 | -def install_solr(): | |
285 | - """Install Solr""" | |
286 | - | |
287 | - SOLR_PKG = 'https://archive.apache.org/dist/lucene/solr/4.6.1/solr-4.6.1.tgz' | |
288 | - | |
289 | - if not exists('~/solr-4.6.1'): | |
290 | - run('wget {} -O /tmp/solr-4.6.1.tgz'.format(SOLR_PKG)) | |
291 | - run('tar xzf /tmp/solr-4.6.1.tgz -C /tmp/') | |
292 | - run('cp -rf /tmp/solr-4.6.1 ~/solr-4.6.1') | |
293 | - run('mv ~/solr-4.6.1/example ~/solr-4.6.1/colab') | |
294 | - run('chmod +x ~/solr-4.6.1/colab/start.jar') | |
295 | - run('rm /tmp/solr-4.6.1.tgz') | |
296 | - | |
297 | - with cd('~/solr-4.6.1/colab/solr/collection1/conf/'): | |
298 | - if not exists('stopwords_en.txt'): | |
299 | - run('cp stopwords.txt stopwords_en.txt') | |
300 | - | |
301 | - | |
302 | -@task | |
303 | -def solr(port=8983): | |
304 | - """Start Solr""" | |
305 | - with cd('~/solr-4.6.1/colab'), settings(user='colab'): | |
306 | - run('java -jar start.jar -Djetty.port={}'.format(port)) | |
307 | - | |
308 | - | |
309 | -@task | |
310 | -def rebuild_index(age=None, batch=None): | |
311 | - """Rebuild the solr index""" | |
312 | - age_arg = '' | |
313 | - if age: | |
314 | - age_arg = '--age={}'.format(age) | |
315 | - | |
316 | - batch_arg = '' | |
317 | - if batch: | |
318 | - batch_arg = '--batch-size={}'.format(batch) | |
319 | - | |
320 | - manage('rebuild_index {} {}'.format(age_arg, batch_arg)) | |
321 | - | |
322 | - | |
323 | -@task | |
324 | -def update_index(): | |
325 | - """Update solr index""" | |
326 | - manage('update_index') | |
327 | - | |
328 | - | |
329 | -@task | |
330 | -def build_schema(): | |
331 | - """Build solr schema""" | |
332 | - solr_schema_file = '~/solr-4.6.1/colab/solr/collection1/conf/schema.xml' | |
333 | - manage('build_solr_schema -f {}'.format(solr_schema_file)) | |
334 | - run(r'sed -i "s/<fields>/<fields>\n<field name=\"_version_\" type=\"long\" indexed=\"true\" stored =\"true\"\/>/" {}'.format(solr_schema_file)) | |
335 | - | |
336 | - | |
337 | -# Main | |
338 | -environment() |