X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/cda38daafe58a438d629f00b3c17daf1bdedf3ba..64af15fe792f42d3a7dc3e857a45577d16806255:/fabfile.py

diff --git a/fabfile.py b/fabfile.py
index 5473b6dd4..11442c7ea 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -10,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']
@@ -17,19 +39,16 @@ 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(),
     ]
 
 
-def update_counters():
-    print '>>> update counters'
-    require('app_path', 'project_name')
-    with cd(get_django_root_path('current')):
-        run('%(ve)s/bin/python manage.py update_counters' % env, pty=True)
-
-
 @task
 def beta():
     env.hosts = ['giewont.icm.edu.pl']
@@ -39,7 +58,7 @@ def beta():
     env.django_root_path = 'src'
     env.requirements_file = 'requirements/requirements.txt'
     env.pre_collectstatic = [
-        update_counters,
+        ManageTask('update_counters'),
     ]
     env.services = [
         Supervisord('beta'),