From d4d026bf6afc05b5a742c0fb30915cc9cd332d23 Mon Sep 17 00:00:00 2001 From: Jens Langhammer Date: Fri, 18 Sep 2020 18:58:07 +0200 Subject: [PATCH] stages/user_write: add migration that removes unintended data --- .../migrations/0002_auto_20200918_1653.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 passbook/stages/user_write/migrations/0002_auto_20200918_1653.py diff --git a/passbook/stages/user_write/migrations/0002_auto_20200918_1653.py b/passbook/stages/user_write/migrations/0002_auto_20200918_1653.py new file mode 100644 index 000000000..1be48e285 --- /dev/null +++ b/passbook/stages/user_write/migrations/0002_auto_20200918_1653.py @@ -0,0 +1,27 @@ +# Generated by Django 3.1.1 on 2020-09-18 16:53 + +from django.apps.registry import Apps +from django.db import migrations +from django.db.backends.base.schema import BaseDatabaseSchemaEditor + + +def remove_unintended_attributes(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): + db_alias = schema_editor.connection.alias + User = apps.get_model("passbook_core", "User") + for user in User.objects.using(db_alias).all(): + if "password_repeat" in user.attributes: + del user.attributes["password_repeat"] + if "password" in user.attributes: + del user.attributes["password"] + user.save() + + +class Migration(migrations.Migration): + + dependencies = [ + ("passbook_stages_user_write", "0001_initial"), + ] + + operations = [ + migrations.RunPython(remove_unintended_attributes), + ]