josh/django-mfa2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use the ORM to perform migration query.

Browse Source
This commit is contained in:
nswain
2020-09-09 17:32:09 -06:00
parent f5218bdbdb
commit f3483868ed
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mfa/migrations/0009_user_keys_owned_by_enterprise.py
View File

@@ -4,6 +4,12 @@ from __future__ import unicode_literals
from django.db import models, migrations
from django.conf import settings
def update_owned_by_enterprise(apps, schema_editor):
user_keys = apps.get_model('mfa', 'user_keys')
user_keys.objects.filter(key_type='FIDO2').update(owned_by_enterprise=getattr(settings,"MFA_OWNED_BY_ENTERPRISE",False))
class Migration(migrations.Migration):
dependencies = [
@@ -16,5 +22,5 @@ class Migration(migrations.Migration):
name='owned_by_enterprise',
field=models.NullBooleanField(default=None),
),
migrations.RunSQL("update mfa_user_keys set owned_by_enterprise = %s where key_type='FIDO2'"%(getattr(settings,"MFA_OWNED_BY_ENTERPRISE",False)))
migrations.RunPython(update_owned_by_enterprise)
]