diff --git a/gracedb/ligoauth/migrations/0054_update_pycbclive_cert.py b/gracedb/ligoauth/migrations/0054_update_pycbclive_cert.py new file mode 100644 index 0000000000000000000000000000000000000000..cea9b363007c183fec2e65ae13afbb3d0127fe5d --- /dev/null +++ b/gracedb/ligoauth/migrations/0054_update_pycbclive_cert.py @@ -0,0 +1,43 @@ +# -*- coding: utf-8 -*- +# Generated by Django 1.11.20 on 2019-06-03 20:10 +from __future__ import unicode_literals + +from django.db import migrations + + +ACCOUNT = { + 'name': 'virgo_detchar', + 'new_cert': '/DC=org/DC=cilogon/C=US/O=LIGO/OU=Robots/CN=ldas-grid.ligo.caltech.edu/CN=pycbclive/CN=Tito Canton/CN=UID:tito.canton.robot', +} + + +def add_cert(apps, schema_editor): + RobotUser = apps.get_model('auth', 'User') + + # Get user + user = RobotUser.objects.get(username=ACCOUNT['name']) + + # Create new certificate + user.x509cert_set.create(subject=ACCOUNT['new_cert']) + + +def delete_cert(apps, schema_editor): + RobotUser = apps.get_model('auth', 'User') + + # Get user + user = RobotUser.objects.get(username=ACCOUNT['name']) + + # Delete new certificate + cert = user.x509cert_set.get(subject=ACCOUNT['new_cert']) + cert.delete() + + +class Migration(migrations.Migration): + + dependencies = [ + ('ligoauth', '0053_update_virgodetchar_cert'), + ] + + operations = [ + migrations.RunPython(add_cert, delete_cert), + ] \ No newline at end of file