diff --git a/awx/main/migrations/0123_drop_hg_support.py b/awx/main/migrations/0123_drop_hg_support.py index 2a0ed1d1fb..089c6bba6f 100644 --- a/awx/main/migrations/0123_drop_hg_support.py +++ b/awx/main/migrations/0123_drop_hg_support.py @@ -1,6 +1,5 @@ -# Generated by Django 2.2.16 on 2020-12-01 18:50 - from django.db import migrations, models +from awx.main.migrations._hg_removal import delete_hg_scm class Migration(migrations.Migration): @@ -10,6 +9,7 @@ class Migration(migrations.Migration): ] operations = [ + migrations.RunPython(delete_hg_scm), migrations.AlterField( model_name='project', name='scm_type', diff --git a/awx/main/migrations/_hg_removal.py b/awx/main/migrations/_hg_removal.py new file mode 100644 index 0000000000..7b9c878198 --- /dev/null +++ b/awx/main/migrations/_hg_removal.py @@ -0,0 +1,12 @@ +import logging + +logger = logging.getLogger('awx.main.migrations') + + +def delete_hg_scm(apps, schema_editor): + set_current_apps(apps) + Project = apps.get_model('main', 'Project') + ProjectUpdate = apps.get_model('main', 'ProjectUpdate') + + ProjectUpdate.objects.filter(project__scm_type='hg').delete() + Project.objects.filter(scm_type='hg').delete()