From a3486e793da5facdcaace8d89d2bd9fefabb5c94 Mon Sep 17 00:00:00 2001
From: Tanner Prestegard <tanner.prestegard@ligo.org>
Date: Wed, 27 Mar 2019 11:33:17 -0500
Subject: [PATCH] Add account and cert for Fermi GBM follow-up
---
...8_create_gbm_followup_account_and_certs.py | 61 +++++++++++++++++++
1 file changed, 61 insertions(+)
create mode 100644 gracedb/ligoauth/migrations/0028_create_gbm_followup_account_and_certs.py
diff --git a/gracedb/ligoauth/migrations/0028_create_gbm_followup_account_and_certs.py b/gracedb/ligoauth/migrations/0028_create_gbm_followup_account_and_certs.py
new file mode 100644
index 000000000..edd1be88b
--- /dev/null
+++ b/gracedb/ligoauth/migrations/0028_create_gbm_followup_account_and_certs.py
@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.18 on 2019-03-27 17:12
+from __future__ import unicode_literals
+
+from django.conf import settings
+from django.db import migrations
+
+ACCOUNT = {
+ 'username': 'fermi-gbm-follow-up',
+ 'email': 'tyson.littenberg@ligo.org',
+ 'first_name': '',
+ 'last_name': 'Fermi-GBM follow-up',
+}
+
+CERTS = [
+ '/DC=org/DC=cilogon/C=US/O=LIGO/OU=Robots/CN=hippo.nsstc.uah.edu/CN=Fermi-GBM-follow-up/CN=Tyson Littenberg/CN=UID:tyson.littenberg.robot',
+]
+
+
+def create_account_and_certs(apps, schema_editor):
+ RobotUser = apps.get_model('ligoauth', 'RobotUser')
+ X509Cert = apps.get_model('ligoauth', 'X509Cert')
+ Group = apps.get_model('auth', 'Group')
+
+ # Create user
+ user = RobotUser.objects.create(**ACCOUNT)
+
+ # Add user to LVC group
+ group = Group.objects.get(name=settings.LVC_GROUP)
+ group.user_set.add(user)
+
+ # Create X509 certificates
+ for subject in CERTS:
+ user.x509cert_set.create(subject=subject)
+
+
+def delete_account_and_certs(apps, schema_editor):
+ RobotUser = apps.get_model('ligoauth', 'RobotUser')
+ X509Cert = apps.get_model('ligoauth', 'X509Cert')
+
+ # Get user
+ user = RobotUser.objects.get(username=ACCOUNT['username'])
+
+ # Delete certs
+ for cert in user.x509cert_set.filter(subject__in=CERTS):
+ cert.delete()
+
+ # Delete full User object
+ user.user_ptr.delete()
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('ligoauth', '0027_add_olib_cert'),
+ ]
+
+ operations = [
+ migrations.RunPython(create_account_and_certs,
+ delete_account_and_certs),
+ ]
--
GitLab