# 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), ]