ArchiveBox/archivebox/machine/migrations/0011_remove_binary_output_dir.py

from django.db import migrations


def remove_output_dir_if_exists(apps, schema_editor):
    cursor = schema_editor.connection.cursor()
    cursor.execute("PRAGMA table_info(machine_binary)")
    columns = {row[1] for row in cursor.fetchall()}

    if "output_dir" not in columns:
        return

    Binary = apps.get_model("machine", "Binary")
    schema_editor.remove_field(Binary, Binary._meta.get_field("output_dir"))


class Migration(migrations.Migration):
    dependencies = [
        ("machine", "0010_alter_process_process_type"),
    ]

    operations = [
        migrations.SeparateDatabaseAndState(
            database_operations=[
                migrations.RunPython(remove_output_dir_if_exists, migrations.RunPython.noop),
            ],
            state_operations=[
                migrations.RemoveField(
                    model_name="binary",
                    name="output_dir",
                ),
            ],
        ),
    ]