More API testing.

[wolnelektury.git] / fabfile.py

diff --git a/fabfile.py b/fabfile.py
index bdf0fad..11442c7 100644 (file)
--- a/fabfile.py
+++ b/fabfile.py
@@ -1,4 +1,6 @@
+# -*- coding: utf-8 -*-
 from fnpdeploy import *
+
 try:
     from fabfile_local import *
 except ImportError:
@@ -8,6 +10,28 @@ except ImportError:
 env.project_name = 'wolnelektury'
 
 
+class ManageTask(Task):
+    def __init__(self, name, params='', **kwargs):
+        super(ManageTask, self).__init__(**kwargs)
+        self.name = name
+        self.params = params
+
+    def run(self):
+        require('app_path', 'project_name')
+        with cd(get_django_root_path('current')):
+            run('source %(ve)s/bin/activate && python manage.py %(task)s %(params)s' % {
+                've': env.ve,
+                'task': self.name,
+                'params': self.params,
+            }, pty=True)
+
+
+class Memcached(Service):
+    def run(self):
+        print '>>> memcached: restart'
+        sudo('/etc/init.d/memcached restart', shell=False)
+
+
 @task
 def production():
     env.hosts = ['giewont.icm.edu.pl']
@@ -15,9 +39,13 @@ def production():
     env.app_path = '/srv/wolnelektury.pl'
     env.django_root_path = 'src'
     env.requirements_file = 'requirements/requirements.txt'
+    env.pre_collectstatic = [
+        ManageTask('update_counters'),
+    ]
     env.services = [
         Supervisord('wolnelektury'),
         Supervisord('wolnelektury.celery'),
+        Memcached(),
     ]
 
 
@@ -29,6 +57,9 @@ def beta():
     env.ve = '/srv/wolnelektury.pl/ve'
     env.django_root_path = 'src'
     env.requirements_file = 'requirements/requirements.txt'
+    env.pre_collectstatic = [
+        ManageTask('update_counters'),
+    ]
     env.services = [
         Supervisord('beta'),
     ]