From b705e68e3bafd7cffc4efa77881ffc3a064734e1 Mon Sep 17 00:00:00 2001 From: Ramon Wenger Date: Tue, 31 Aug 2021 14:57:16 +0200 Subject: [PATCH] Use migration without helper class --- server/portfolio/migrations/0006_auto_20210810_1348.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/server/portfolio/migrations/0006_auto_20210810_1348.py b/server/portfolio/migrations/0006_auto_20210810_1348.py index e19eb25a..cc3d026f 100644 --- a/server/portfolio/migrations/0006_auto_20210810_1348.py +++ b/server/portfolio/migrations/0006_auto_20210810_1348.py @@ -5,12 +5,13 @@ from django.db import migrations def set_default_school_class(apps, schema_editor): Project = apps.get_model('portfolio', 'Project') - User = apps.get_model('users', 'User') + UserSetting = apps.get_model('users', 'UserSetting') projects = Project.objects.all() bulk_projects = [] for project in projects: - owner = User.objects.get(id=project.student.id) - selected_class = owner._get_selected_class() + owner = project.student + settings = UserSetting.objects.get(user=owner) + selected_class = settings.selected_class if selected_class is not None: project.school_class = selected_class bulk_projects.append(project)