adding bayeswave robot certificate

ligoauth/migrations/0013_update_bayeswave_robot.py

+# -*- coding: utf-8 -*-
+# Default imports
+from __future__ import unicode_literals
+from django.db import migrations, models
+from django.conf import settings
+
+ROBOTS = [{'username': 'bayeswave',
+           'newcert': '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=bayeswave_online/ldas-grid.ligo.caltech.edu',
+           'oldcert': '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=bayeswave/ldas-grid.ligo.caltech.edu',
+           'newemail': 'margaret.millhouse@montana.edu',
+           'oldemail': 'jkanner@caltech.edu',
+          },
+]
+
+def create_robots(apps, schema_editor):
+    LocalUser = apps.get_model('ligoauth','LocalUser')
+    X509Cert = apps.get_model('ligoauth','X509Cert')
+
+    # Get/create new user, get/create new cert, associate user with cert.
+    for entry in ROBOTS:
+        # get user
+        user = LocalUser.objects.get(username=entry['username'])
+        user.email = entry['newemail']
+        user.save()
+
+        # get or create certificate, add user
+        cert, c_created = X509Cert.objects.get_or_create(subject=entry['newcert'])
+        cert.users.add(user)
+        cert.save()
+
+        # Delete old certs.
+        for oldcert in entry['oldcert']:
+            try:
+                cert = X509Cert.objects.get(subject=oldcert)
+                cert.delete()
+            except:
+                pass
+
+def delete_robots(apps, schema_editor):
+    LocalUser = apps.get_model('ligoauth','LocalUser')
+    X509Cert = apps.get_model('ligoauth','X509Cert')
+
+    # Delete users.
+    for entry in ROBOTS:
+        user = LocalUser.objects.get(username=entry['username'])
+        user.email = entry['oldemail']
+        user.save()
+
+        # Create oldcerts, add to user
+        for oldcert in entry['oldcert']:
+            cert, created = X509Cert.objects.get_or_create(subject=oldcert)
+            cert.users.add(user)
+            cert.save()
+
+        # Delete newcert.
+        X509Cert.objects.get(subject=entry['newcert']).delete()
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ligoauth', '0012_update_iDQ_robot'),
+    ]
+
+    operations = [
+        migrations.RunPython(create_robots, delete_robots)
+    ]
+
+# End of file