From c9af2f4e973581691fa9eb7acc3d663527ce2076 Mon Sep 17 00:00:00 2001
From: Tanner Prestegard <tanner.prestegard@ligo.org>
Date: Sun, 5 Feb 2017 16:38:23 -0600
Subject: [PATCH] updating robot cert for LIB robot
---
ligoauth/migrations/0014_update_lib_robot.py | 65 ++++++++++++++++++++
1 file changed, 65 insertions(+)
create mode 100644 ligoauth/migrations/0014_update_lib_robot.py
diff --git a/ligoauth/migrations/0014_update_lib_robot.py b/ligoauth/migrations/0014_update_lib_robot.py
new file mode 100644
index 000000000..c05298c39
--- /dev/null
+++ b/ligoauth/migrations/0014_update_lib_robot.py
@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+# Default imports
+from __future__ import unicode_literals
+from django.db import migrations, models
+from django.conf import settings
+
+ROBOTS = [{'username': 'LIB',
+ 'newcert': '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=LIB/ldas-pcdev5.ligo.caltech.edu',
+ 'oldcert': '/DC=org/DC=ligo/O=LIGO/OU=Services/CN=LIB/ldas-pcdev1.ligo.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.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.
+ try:
+ cert = X509Cert.objects.get(subject=entry['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.save()
+
+ # Create oldcerts, add to user
+ cert, created = X509Cert.objects.get_or_create(subject=entry['oldcert'])
+ cert.users.add(user)
+ cert.save()
+
+ # Delete newcert.
+ try:
+ X509Cert.objects.get(subject=entry['newcert']).delete()
+ except:
+ pass
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('ligoauth', '0013_update_bayeswave_robot'),
+ ]
+
+ operations = [
+ migrations.RunPython(create_robots, delete_robots)
+ ]
+
+# End of file
--
GitLab