X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/029fd918963a5cebf7cb722e2ba39c27639a976f..e79ef640becb4b59dba11c1dfcce0ea87000919b:/fabfile.py

diff --git a/fabfile.py b/fabfile.py
index 87954e834..11442c7ea 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -10,24 +10,26 @@ except ImportError:
 env.project_name = 'wolnelektury'
 
 
-def update_counters():
-    print '>>> updating 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)
+class ManageTask(Task):
+    def __init__(self, name, params='', **kwargs):
+        super(ManageTask, self).__init__(**kwargs)
+        self.name = name
+        self.params = params
 
-
-def compile_messages():
-    print '>>> compiling messages'
-    require('app_path', 'project_name')
-    with cd(get_django_root_path('current')):
-        run('source %(ve)s/bin/activate && python manage.py localepack -c' % env, pty=True)
+    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('service memcached restart', shell=False)
+        sudo('/etc/init.d/memcached restart', shell=False)
 
 
 @task
@@ -38,8 +40,7 @@ def production():
     env.django_root_path = 'src'
     env.requirements_file = 'requirements/requirements.txt'
     env.pre_collectstatic = [
-        update_counters,
-        compile_messages,
+        ManageTask('update_counters'),
     ]
     env.services = [
         Supervisord('wolnelektury'),
@@ -57,8 +58,7 @@ def beta():
     env.django_root_path = 'src'
     env.requirements_file = 'requirements/requirements.txt'
     env.pre_collectstatic = [
-        update_counters,
-        compile_messages,
+        ManageTask('update_counters'),
     ]
     env.services = [
         Supervisord('beta'),