Merge pull request alphagov#153 from alphagov/emergency_publishing

mattbostock · mattbostock · commit 8528c5166ded · 2015-08-26T10:28:02.000+01:00
Emergency publishing
diff --git a/campaigns.py b/campaigns.py
@@ -6,52 +6,89 @@
 
 from jinja2 import Template
 
-import vm
+import app
+
+
+env['eagerly_disconnect'] = True
+
+
+APPLICATIONS = ['frontend', 'static']
+CAMPAIGN_CLASSES = ['red', 'black', 'green']
+
 
 def validate_classes(campaign_class):
     """Checks that the campaign class is valid"""
-    if campaign_class in ['red', 'black', 'green']:
+    if campaign_class in CAMPAIGN_CLASSES:
         return campaign_class
-    raise Exception, "Invalid class %s, valid values are 'red', 'black', 'green'" % campaign_class
+    raise Exception, "Invalid class {}, valid values are {}".format(campaign_class, CAMPAIGN_CLASSES)
 
 
 @runs_once
-def homepage_template():
-    """Promotes a campaign to the homepage of GOV.UK"""
-    context = {
-        'heading': prompt("Heading for campaign:"),
-        'extra_info': prompt("Extra information for campaign:"),
-        'more_info': prompt("Link for more information:"),
-        'campaign_class': prompt("Campaign class:", validate=validate_classes)
+def set_context():
+    env['context'] = {
+        'heading': prompt("Heading for campaign:", 'heading'),
+        'extra_info': prompt("Extra information for campaign:", 'extra_info'),
+        'more_info': prompt("Link for more information:", 'more_info'),
+        'campaign_class': prompt("Campaign class:", 'campaign_class', validate=validate_classes)
     }
 
-    template = Template("""<div id="campaign" class="{{ campaign_class }}">
-  <div class="campaign-inner">
-    <h1>{{ heading|e }}</h1>
-    <p>{{ extra_info|e }}</p>
-    <a href="{{ more_info|e }}">More information</a>
-  </div>
-</div>""")
-    env['template_contents'] = template.render(context)
 
-    print "Template contents:\n%s" % env['template_contents']
+def template(app):
+    if app == 'frontend':
+      template = Template("""<div id="campaign" class="{{ campaign_class }}">
+    <div class="campaign-inner">
+      <h1>{{ heading|e }}</h1>
+      <p>{{ extra_info|e }}</p>
+      <a href="{{ more_info|e }}">More information</a>
+    </div>
+  </div>""")
+    elif app == 'static':
+      template = Template("""<p>{{ heading|e }}<br />
+    {{ extra_info|e }}</p>
+  <a href="#" class="right">{{ more_info|e }}</a>""")
+
+    env['template_contents'] = template.render(env.context)
+
 
 @task
 @roles('class-frontend')
-def deploy_to_homepage():
-    execute(homepage_template)
+def deploy_banner(application):
+    execute(template, application)
+    if application == 'frontend':
+        remote_filename = '/var/apps/frontend/app/views/root/_campaign_notification.html.erb'
+    elif application == 'static':
+        remote_filename = "/var/apps/static/app/views/notifications/banner_{}.erb".format(env.campaign_class)
+    content = env['template_contents']
+    put(StringIO.StringIO(content), remote_filename, use_sudo=True, mirror_local_mode=True)
+    sudo('chown deploy:deploy %s' % remote_filename)
+    execute(app.reload, application)
 
-    contents = env['template_contents']
+def remove_banner(application):
+    if application == 'frontend':
+        remote_filenames = ['/var/apps/frontend/app/views/root/_campaign_notification.html.erb']
+    elif application == 'static':
+        remote_filenames = ["/var/apps/static/app/views/notifications/banner_%s.erb" % i for i in CAMPAIGN_CLASSES]
+    content = ''
+    for remote_filename in remote_filenames:
+        put(StringIO.StringIO(content), remote_filename, use_sudo=True, mirror_local_mode=True)
+        sudo('chown deploy:deploy %s' % remote_filename)
+    execute(app.reload, application)
 
-    remote_filename = '/var/apps/frontend/app/views/root/_campaign_notification.html.erb'
-    put(StringIO.StringIO(contents), remote_filename, use_sudo=True, mirror_local_mode=True)
-    sudo('chown deploy:deploy %s' % remote_filename)
-    execute(vm.reload_unicorn, name='frontend')
 
 @task
 @roles('class-frontend')
-def remove_from_homepage():
-    remote_filename = '/var/apps/frontend/app/views/root/_campaign_notification.html.erb'
-    put(StringIO.StringIO(''), remote_filename, use_sudo=True, mirror_local_mode=True)
-    sudo('chown deploy:deploy %s' % remote_filename)
-    execute(vm.reload_unicorn, name='frontend')
+def deploy_emergency_banner():
+    """Deploy an emergency banner to GOV.UK"""
+    execute(set_context)
+    for application in APPLICATIONS:
+        # Black banners are only placed on the homepage i.e.'frontend'.
+        # Don't deploy a black banner to the static application.
+        if not (env.campaign_class == 'black' and application == 'static'):
+            deploy_banner(application)
+
+@task
+@roles('class-frontend')
+def remove_emergency_banner():
+    """Remove all banners from GOV.UK"""
+    for application in APPLICATIONS:
+        remove_banner(application)
diff --git a/cdn.py b/cdn.py
@@ -8,8 +8,13 @@
 @roles('class-cache')
 def fastly_purge(*args):
     "Purge items from Fastly, eg \"/one,/two,/three\". Wildcards not supported."
-    govuk_fastly = 'http://www-gov-uk.map.fastly.net'
-    hostnames_to_purge = ['www.gov.uk', 'assets.digital.cabinet-office.gov.uk']
+    if env.environment == 'production':
+        hostnames_to_purge = ['www.gov.uk', 'assets.digital.cabinet-office.gov.uk']
+    elif env.environment == 'staging':
+        # FIXME - Staging Fastly service should point to non alphagov.co.uk
+        # domains and amended here after the miration to Carrenza
+        hostnames_to_purge = ['www.staging.alphagov.co.uk', 'assets.staging.alphagov.co.uk']
+
     for path in args:
         if "*" in path:
             abort("Sorry, purging paths containing wildcards is not supported "
@@ -18,7 +23,7 @@ def fastly_purge(*args):
                 % path)
     for govuk_path in args:
         for hostname in hostnames_to_purge:
-            run("curl -s -X PURGE -H 'Host: {0}' {1}{2} | grep 'ok'".format(hostname, govuk_fastly, govuk_path.strip()))
+            run("curl -s -X PURGE {0}{1} | grep 'ok'".format(hostname, govuk_path.strip()))
 
 @task
 def purge_all(*args):
diff --git a/fabfile.py b/fabfile.py
@@ -231,21 +231,25 @@ def help(name):
 @task
 def production():
     """Select production environment"""
+    env['environment'] = 'production'
     _set_gateway('production.alphagov.co.uk')
 
 @task
 def production_migration():
     """Select production migration environment"""
+    env['environment'] = 'production'
     _set_gateway('publishing.service.gov.uk')
 
 @task
 def staging():
     """Select staging environment"""
+    env['environment'] = 'staging'
     _set_gateway('staging.publishing.service.gov.uk')
 
 @task
 def preview():
     """Select preview environment"""
+    env['environment'] = 'preview'
     _set_gateway('preview.alphagov.co.uk')
 
 @task
