diff --git a/gracedb/ligoauth/migrations/0021_create_deepclean_account_and_certs.py b/gracedb/ligoauth/migrations/0021_create_deepclean_account_and_certs.py index 172d0102da4c34ddb211127e27cbac86204ff059..fb51749df5a0d8ff89acbe84faa6aa23d14f2af5 100644 --- a/gracedb/ligoauth/migrations/0021_create_deepclean_account_and_certs.py +++ b/gracedb/ligoauth/migrations/0021_create_deepclean_account_and_certs.py @@ -2,6 +2,7 @@ # Generated by Django 1.11.16 on 2019-01-04 18:39 from __future__ import unicode_literals +from django.conf import settings from django.db import migrations ACCOUNT = { @@ -20,10 +21,15 @@ CERTS = [ 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) diff --git a/gracedb/ligoauth/migrations/0024_create_gwhen_account_and_certs.py b/gracedb/ligoauth/migrations/0024_create_gwhen_account_and_certs.py index 15f842aa073e47e8b777ce8d1c094569e0d4ea0f..22f514ce78ed7ef01c1467cea8a85ffc01976547 100644 --- a/gracedb/ligoauth/migrations/0024_create_gwhen_account_and_certs.py +++ b/gracedb/ligoauth/migrations/0024_create_gwhen_account_and_certs.py @@ -2,6 +2,7 @@ # Generated by Django 1.11.18 on 2019-02-18 19:50 from __future__ import unicode_literals +from django.conf import settings from django.db import migrations ACCOUNT = { @@ -19,10 +20,15 @@ CERTS = [ 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)