From f3483868ed9c01498f909630e4bcff8dcfff5edc Mon Sep 17 00:00:00 2001 From: nswain Date: Wed, 9 Sep 2020 17:32:09 -0600 Subject: [PATCH] Use the ORM to perform migration query. --- mfa/migrations/0009_user_keys_owned_by_enterprise.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/mfa/migrations/0009_user_keys_owned_by_enterprise.py b/mfa/migrations/0009_user_keys_owned_by_enterprise.py index aa9d1aa..9185fcc 100644 --- a/mfa/migrations/0009_user_keys_owned_by_enterprise.py +++ b/mfa/migrations/0009_user_keys_owned_by_enterprise.py @@ -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) ]