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