Merge pull request alphagov#80 from alphagov/add-whitehall-de-dupe

ajlanghorn · ajlanghorn · commit 632dd35a75dc · 2014-12-01T16:56:29.000Z
Add tasks for deduping stats announcements
diff --git a/fabfile.py b/fabfile.py
@@ -34,6 +34,7 @@
 import search
 import statsd
 import vm
+import whitehall
 
 HERE = os.path.dirname(__file__)
 SSH_DIR = os.path.join(HERE, '.ssh')
diff --git a/whitehall.py b/whitehall.py
@@ -0,0 +1,21 @@
+from fabric.api import task, hosts, sudo, cd, execute
+
+
+@task
+def dedupe_stats_announcement_from_file(filename):
+    """De-duplicate Whitehall statistics accouncements from a CSV file"""
+    with open(filename) as fd:
+        duplicates = [line.strip().split(',') for line in fd]
+        for duplicate_slug, authoritative_slug in duplicates:
+            execute(
+                dedupe_stats_announcement, duplicate_slug, authoritative_slug)
+
+
+@task
+@hosts('whitehall-backend-1.backend')
+def dedupe_stats_announcement(duplicate_slug, authoritative_slug):
+    """De-duplicate Whitehall statistics announcement"""
+    command = 'govuk_setenv whitehall ./script/dedupe_stats_announcement {} {}'
+    with cd('/var/apps/whitehall'):
+        sudo(command.format(duplicate_slug, authoritative_slug),
+             user='deploy')
