updating iDQ robots with new e-mails and certificates; deleting old certs

ligoauth/migrations/0012_update_iDQ_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': 'idq-la',
+           'newcert': '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=idqLLO/ldas-grid.ligo-la.caltech.edu',
+           'oldcert': [
+                       '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=idq/ldas-pcdev1.ligo-la.caltech.edu',
+                       '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=idq/ldas-pcdev2.ligo-la.caltech.edu'
+                      ],
+            'newemail': 'ressick@mit.edu',
+            'oldemail': 'rvaulin@mit.edu',
+          },
+          {'username': 'idq-wa',
+           'newcert': '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=idqLHO/ldas-grid.ligo-wa.caltech.edu',
+           'oldcert': [
+                      '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=idq/ldas-pcdev1.ligo-wa.caltech.edu',
+                      '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=idq/ldas-pcdev2.ligo-wa.caltech.edu'
+                      ],
+           'newemail': 'ressick@mit.edu',
+           'oldemail': 'rvaulin@mit.edu',
+          },
+]
+
+def create_robots(apps, schema_editor):
+    LocalUser = apps.get_model('ligoauth','LocalUser')
+    X509Cert = apps.get_model('ligoauth','X509Cert')
+    Group = apps.get_model('auth','Group')
+    lvc_group = Group.objects.get(name=settings.LVC_GROUP)
+
+    # 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', '0011_add_O2_hwinj_logger_account'),
+    ]
+
+    operations = [
+        migrations.RunPython(create_robots, delete_robots)
+    ]
+
+# End of file