karrio-server-pricing 2025.5rc14__py3-none-any.whl → 2025.5rc16__py3-none-any.whl

karrio/server/pricing/migrations/0065_fix_surcharge_type_enum.py

@@ -0,0 +1,36 @@
+# Generated by Django
+
+from django.db import migrations
+
+
+def fix_surcharge_types(apps, schema_editor):
+    """Convert surcharge_type from symbols to enum names"""
+    Surcharge = apps.get_model('pricing', 'Surcharge')
+
+    # Update $ to AMOUNT
+    updated_amount = Surcharge.objects.filter(surcharge_type='$').update(surcharge_type='AMOUNT')
+
+    # Update % to PERCENTAGE
+    updated_percentage = Surcharge.objects.filter(surcharge_type='%').update(surcharge_type='PERCENTAGE')
+
+
+def reverse_surcharge_types(apps, schema_editor):
+    """Reverse the changes - convert back to symbols"""
+    Surcharge = apps.get_model('pricing', 'Surcharge')
+
+    # Reverse AMOUNT to $
+    Surcharge.objects.filter(surcharge_type='AMOUNT').update(surcharge_type='$')
+
+    # Reverse PERCENTAGE to %
+    Surcharge.objects.filter(surcharge_type='PERCENTAGE').update(surcharge_type='%')
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('pricing', '0064_alter_surcharge_carriers_alter_surcharge_services'),
+    ]
+
+    operations = [
+        migrations.RunPython(fix_surcharge_types, reverse_surcharge_types),
+    ]