49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
|
# Generated by Django 3.1.1 on 2020-09-15 19:19
|
||
|
|
|
||
|
|
from django.apps.registry import Apps
|
||
|
|
from django.db import migrations
|
||
|
|
|
||
|
|
|
||
|
|
def create_default_property_mappings(apps: Apps, schema_editor):
|
||
|
|
LDAPPropertyMapping = apps.get_model("passbook_sources_ldap", "LDAPPropertyMapping")
|
||
|
|
db_alias = schema_editor.connection.alias
|
||
|
|
mapping = {
|
||
|
|
"name": "name",
|
||
|
|
"first_name": "givenName",
|
||
|
|
"last_name": "sn",
|
||
|
|
"email": "mail",
|
||
|
|
}
|
||
|
|
for object_field, ldap_field in mapping.items():
|
||
|
|
expression = f"return ldap.get('{ldap_field}')"
|
||
|
|
LDAPPropertyMapping.objects.using(db_alias).get_or_create(
|
||
|
|
expression=expression,
|
||
|
|
object_field=object_field,
|
||
|
|
defaults={
|
||
|
|
"name": f"Autogenerated LDAP Mapping: {ldap_field} -> {object_field}"
|
||
|
|
},
|
||
|
|
)
|
||
|
|
ad_mapping = {
|
||
|
|
"username": "sAMAccountName",
|
||
|
|
"attributes.upn": "userPrincipalName",
|
||
|
|
}
|
||
|
|
for object_field, ldap_field in ad_mapping.items():
|
||
|
|
expression = f"return ldap.get('{ldap_field}')"
|
||
|
|
LDAPPropertyMapping.objects.using(db_alias).get_or_create(
|
||
|
|
expression=expression,
|
||
|
|
object_field=object_field,
|
||
|
|
defaults={
|
||
|
|
"name": f"Autogenerated Active Directory Mapping: {ldap_field} -> {object_field}"
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
("passbook_sources_ldap", "0005_auto_20200913_1947"),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.RunPython(create_default_property_mappings),
|
||
|
|
]
|