From 00fb495492d033f1c971f06239a44f19d4e8781a Mon Sep 17 00:00:00 2001
From: Alexander Burdeiny <bonus_05@mail.ru>
Date: Thu, 14 Jul 2016 17:19:43 +0300
Subject: [PATCH] fabric

---
 fabfile.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/fabfile.py b/fabfile.py
index d697bdde..18240022 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -27,13 +27,21 @@ tickets = {}
 commands = {}
 
 def stage(stage_num):
-    def ticket(func):
+    def add_func(func):
         if not callable(func):
             raise NotImplementedError('func must be a callable')
+        func = ticket(func)
         stages.setdefault(stage_num, OrderedDict()).update({func.__name__: func})
-        tickets.update({func.__name__: func})
         return func
-    return ticket
+    return add_func
+
+
+def ticket(func):
+    if not callable(func):
+        raise NotImplementedError('func must be a callable')
+    # stages.setdefault(stage_num, OrderedDict()).update({func.__name__: func})
+    tickets.update({func.__name__: func})
+    return func
 
 
 # command_pattern
@@ -159,6 +167,11 @@ def t1460():
         run('python manage.py migrate newsletter')
 
 
+@ticket
+def update_crontab():
+    with cd(REMOTE_HOME_DIR):
+        run('python manage.py crontab remove')
+        run('python manage.py crontab add')
 
 # def stage3_pre_final():
 #     with cd(REMOTE_HOME_DIR):