add Issuer model (#56819)

mellon/migrations/0003_add_issuer_model.py

@@ -0,0 +1,47 @@
+# Generated by Django 2.2.19 on 2021-09-14 18:53
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mellon', '0002_sessionindex'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Issuer',
+            fields=[
+                (
+                    'id',
+                    models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID'),
+                ),
+                (
+                    'entity_id',
+                    models.TextField(unique=True, verbose_name='IdP Entity ID'),
+                ),
+                ('slug', models.TextField(null=True, unique=True, verbose_name='IdP slug')),
+            ],
+            options={
+                'verbose_name': 'SAML IdP',
+                'verbose_name_plural': 'SAML IdPs',
+            },
+        ),
+        migrations.AlterField(
+            model_name='usersamlidentifier',
+            name='issuer',
+            field=models.TextField(null=True, verbose_name='Issuer'),
+        ),
+        migrations.AddField(
+            model_name='usersamlidentifier',
+            name='issuer_fk',
+            field=models.ForeignKey(
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                to='mellon.Issuer',
+                verbose_name='Issuer',
+            ),
+        ),
+    ]

mellon/models.py

@@ -28,9 +28,12 @@ class UserSAMLIdentifier(models.Model):
         related_name='saml_identifiers',
         on_delete=models.CASCADE,
     )
-    issuer = models.TextField(verbose_name=_('Issuer'))
+    issuer = models.TextField(verbose_name=_('Issuer'), null=True)
     name_id = models.TextField(verbose_name=_('SAML identifier'))
     created = models.DateTimeField(verbose_name=_('created'), auto_now_add=True)
+    issuer_fk = models.ForeignKey(
+        'mellon.Issuer', verbose_name=_('Issuer'), null=True, on_delete=models.CASCADE
+    )
 
     class Meta:
         verbose_name = _('user SAML identifier')
@@ -60,3 +63,12 @@ class SessionIndex(models.Model):
         verbose_name = _('SAML SessionIndex')
         verbose_name_plural = _('SAML SessionIndexes')
         unique_together = (('saml_identifier', 'session_index', 'session_key'),)
+
+
+class Issuer(models.Model):
+    entity_id = models.TextField(verbose_name=_('IdP Entity ID'), unique=True)
+    slug = models.TextField(verbose_name=_('IdP slug'), unique=True, null=True)
+
+    class Meta:
+        verbose_name = _('SAML IdP')
+        verbose_name_plural = _('SAML IdPs')