76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
# Generated by Django 3.0.3 on 2020-02-17 16:15
|
|
|
|
from django.db import migrations
|
|
|
|
|
|
def cleanup_old_autogenerated(apps, schema_editor):
|
|
SAMLPropertyMapping = apps.get_model(
|
|
"passbook_providers_saml", "SAMLPropertyMapping"
|
|
)
|
|
db_alias = schema_editor.connection.alias
|
|
SAMLPropertyMapping.objects.using(db_alias).filter(
|
|
name__startswith="Autogenerated"
|
|
).delete()
|
|
|
|
|
|
def create_default_property_mappings(apps, schema_editor):
|
|
"""Create default SAML Property Mappings"""
|
|
SAMLPropertyMapping = apps.get_model(
|
|
"passbook_providers_saml", "SAMLPropertyMapping"
|
|
)
|
|
db_alias = schema_editor.connection.alias
|
|
defaults = [
|
|
{
|
|
"FriendlyName": "eduPersonPrincipalName",
|
|
"Name": "urn:oid:1.3.6.1.4.1.5923.1.1.1.6",
|
|
"Template": "{{ user.email }}",
|
|
},
|
|
{
|
|
"FriendlyName": "cn",
|
|
"Name": "urn:oid:2.5.4.3",
|
|
"Template": "{{ user.name }}",
|
|
},
|
|
{
|
|
"FriendlyName": "mail",
|
|
"Name": "urn:oid:0.9.2342.19200300.100.1.3",
|
|
"Template": "{{ user.email }}",
|
|
},
|
|
{
|
|
"FriendlyName": "displayName",
|
|
"Name": "urn:oid:2.16.840.1.113730.3.1.241",
|
|
"Template": "{{ user.username }}",
|
|
},
|
|
{
|
|
"FriendlyName": "uid",
|
|
"Name": "urn:oid:0.9.2342.19200300.100.1.1",
|
|
"Template": "{{ user.pk }}",
|
|
},
|
|
{
|
|
"FriendlyName": "member-of",
|
|
"Name": "member-of",
|
|
"Template": "[{% for group in user.groups.all() %}'{{ group.name }}',{% endfor %}]",
|
|
},
|
|
]
|
|
for default in defaults:
|
|
SAMLPropertyMapping.objects.using(db_alias).get_or_create(
|
|
saml_name=default["Name"],
|
|
friendly_name=default["FriendlyName"],
|
|
template=default["Template"],
|
|
defaults={
|
|
"name": f"Autogenerated SAML Mapping: {default['FriendlyName']} -> {default['Template']}"
|
|
},
|
|
)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
("passbook_providers_saml", "0004_auto_20200217_1526"),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(cleanup_old_autogenerated),
|
|
migrations.RemoveField(model_name="samlpropertymapping", name="values",),
|
|
migrations.RunPython(create_default_property_mappings),
|
|
]
|