From 49d928cabc15399bea89e11cc2e8db2e7df218b0 Mon Sep 17 00:00:00 2001 From: Victor Seva Date: Wed, 8 Feb 2023 15:49:11 +0100 Subject: [PATCH] MT#56508 repoapi: fix jbi rows with projectname with '-gerrit', take two Change-Id: I292d5d838d033570541718dbd69f166809e3d845 --- repoapi/migrations/0014_fix_jbi_gerrit.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/repoapi/migrations/0014_fix_jbi_gerrit.py b/repoapi/migrations/0014_fix_jbi_gerrit.py index bdaabf6..e109450 100644 --- a/repoapi/migrations/0014_fix_jbi_gerrit.py +++ b/repoapi/migrations/0014_fix_jbi_gerrit.py @@ -5,11 +5,22 @@ from django.db import migrations RE_GERRIT = re.compile("-gerrit$") +def revert_func(apps, schema_editor): + pass + + +def remove_gerrit(model): + qs = model.objects.filter(projectname__endswith="-gerrit") + for jbi in qs: + jbi.projectname = RE_GERRIT.sub("", jbi.projectname) + jbi.save() + + def forwards_func(apps, schema_editor): JenkinsBuildInfo = apps.get_model("repoapi", "JenkinsBuildInfo") - qs = JenkinsBuildInfo.objects.filter(projectname__endswith="-gerrit") - for jbi in qs: - jbi.update(projectname=RE_GERRIT.sub("", jbi.projectname)) + remove_gerrit(JenkinsBuildInfo) + BuildInfo = apps.get_model("buildinfo", "BuildInfo") + remove_gerrit(BuildInfo) class Migration(migrations.Migration): @@ -19,5 +30,5 @@ class Migration(migrations.Migration): ] operations = [ - migrations.RunPython(forwards_func), + migrations.RunPython(forwards_func, revert_func), ]