From a1c9706424e9c4569bf958d416e3d636ac32379f Mon Sep 17 00:00:00 2001
From: Wil Clouser <wclouser@mozilla.com>
Date: Wed, 17 Aug 2011 14:36:02 -0700
Subject: [PATCH] Update deploy script; prepare for manage.py commands

---
 scripts/deploy.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/scripts/deploy.py b/scripts/deploy.py
index e22ce654..cec5e523 100755
--- a/scripts/deploy.py
+++ b/scripts/deploy.py
@@ -36,6 +36,27 @@ def schematic(ctx):
         ctx.local("python2.6 ./vendor/src/schematic/schematic migrations")
 
 
+@task
+def disable_cron(ctx):
+    ctx.local("mv /etc/cron.d/builder-prod-maint /tmp/builder-prod-maint")
+
+
+@task
+def enable_cron(ctx):
+    with ctx.lcd(FLIGHTDECK_DIR):
+        ctx.local("cp scripts/crontab/prod /etc/cron.d/builder-prod-maint")
+
+
+def manage_cmd(ctx, command):
+    """Call a manage.py command."""
+    with ctx.lcd(FLIGHTDECK_DIR):
+        ctx.local("python2.6 manage.py %s" % command)
+@task
+def make_crons(ctx):
+    with ctx.lcd(FLIGHTDECK_DIR):
+        ctx.local("python2.6 ./scripts/crontab/make-crons.py")
+
+
 def _git_checkout_tag(ctx, tag):
     ctx.local("git fetch -t origin")
     ctx.local("git checkout %s" % tag)
@@ -46,6 +67,7 @@ def _git_checkout_tag(ctx, tag):
 @task
 def start_update(ctx, tag):
     """Updates code to `tag`"""
+    disable_cron()
     with ctx.lcd(FLIGHTDECK_DIR):
         _git_checkout_tag(ctx, tag)
 
@@ -53,6 +75,13 @@ def start_update(ctx, tag):
 @task
 def update_flightdeck(ctx):
     """Deploys code to the webservers and restarts celery"""
+    # BEGIN: The normal update/push cycle.
+    make_crons()
     schematic()
     deploy_code()
     restart_celery()
+    enable_cron()
+    # END: The normal update/push cycle.
+
+    # Run management commands like this:
+    # manage_cmd(ctx, 'cmd')