MT#56508 repoapi: fix jbi rows with projectname with '-gerrit', take two

Change-Id: I292d5d838d033570541718dbd69f166809e3d845

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),
     ]