Add script to update Mapit database

alexmuller · alexmuller · commit bd24cbd2d55a · 2015-06-19T12:38:06.000+01:00
diff --git a/fabfile.py b/fabfile.py
@@ -26,6 +26,7 @@
 import logstream
 import mongo
 import mainstream_slugs
+import mapit
 import mysql
 import nagios
 import nginx
diff --git a/mapit.py b/mapit.py
@@ -0,0 +1,24 @@
+from fabric.api import task, settings, sudo, env
+from fabric.tasks import execute
+
+import app
+import nginx
+import puppet
+
+@task
+def update_database():
+    """Update a Mapit database using a new database dump"""
+
+    if len(env.hosts) > 1:
+      exit('This command should only be run on one Mapit machine at a time')
+
+    execute(nginx.gracefulstop)
+    execute(app.stop, 'mapit')
+    sudo('service collectd stop')
+
+    sudo('rm /data/vhost/mapit/data/mapit.sql.gz')
+
+    with settings(sudo_user='postgres'):
+      sudo("psql -c 'DROP DATABASE mapit;'")
+
+    execute(puppet.agent, '--test')
