Django migrations: Run Python function that takes optional parameter when reversing migration

migration_with_python.py

from django.db import migrations
from myapp.models import MyModel

def convert_mymodel_status(apps, schema_editor, reverse: bool | None = False):
    """
    Convert "OLD STATUS" status to "NEW STATUS"
    Or reverses these changes
    """
    status = "NEW STATUS" if reverse else "OLD STATUS"
    updated_status = "OLD STATUS" if reverse else "NEW STATUS"

    MyModel.objects.filter(status=status).update(status=updated_status)


class Migration(migrations.Migration):
    dependencies = []

    operations = [
        migrations.RunPython(
            code=convert_mymodel_status,
            reverse_code=lambda apps, schema_editor: convert_mymodel_status(
                apps, schema_editor, True
            ),
        )
    ]