from django.db import migrations

def update_statuses(apps, schema_editor):
    EditSubmission = apps.get_model('moderation', 'EditSubmission')
    PhotoSubmission = apps.get_model('moderation', 'PhotoSubmission')

    # Update EditSubmissions
    EditSubmission.objects.filter(status='NEW').update(status='PENDING')

    # Update PhotoSubmissions
    PhotoSubmission.objects.filter(status='NEW').update(status='PENDING')
    PhotoSubmission.objects.filter(status='AUTO_APPROVED').update(status='APPROVED')

def reverse_statuses(apps, schema_editor):
    EditSubmission = apps.get_model('moderation', 'EditSubmission')
    PhotoSubmission = apps.get_model('moderation', 'PhotoSubmission')

    # Reverse EditSubmissions
    EditSubmission.objects.filter(status='PENDING').update(status='NEW')

    # Reverse PhotoSubmissions
    PhotoSubmission.objects.filter(status='PENDING').update(status='NEW')

class Migration(migrations.Migration):

    dependencies = [
        ('moderation', '0002_alter_editsubmission_status_and_more'),
    ]

    operations = [
        migrations.RunPython(update_statuses, reverse_statuses),
    ]