first

apps/tickets/migrations/0014_merge_legacy_warehouse_units.py

@@ -0,0 +1,76 @@
+from django.db import migrations
+
+
+def merge_unit(apps, legacy_slug, canonical_slug, canonical_name, canonical_description):
+    Unit = apps.get_model("staff", "OrganizationalUnit")
+    Membership = apps.get_model("staff", "UnitMembership")
+    StaffProfile = apps.get_model("staff", "StaffProfile")
+    TicketCategory = apps.get_model("tickets", "TicketCategory")
+    Ticket = apps.get_model("tickets", "Ticket")
+    GroupDispatch = apps.get_model("tickets", "GroupDispatch")
+    ContactGroup = apps.get_model("customers", "ContactGroup")
+    CustomerProfile = apps.get_model("customers", "CustomerProfile")
+    SMSTemplate = apps.get_model("notifications", "SMSTemplate")
+    TicketMagicLink = apps.get_model("accounts", "TicketMagicLink")
+
+    legacy = Unit.objects.filter(slug=legacy_slug).first()
+    canonical = Unit.objects.filter(slug=canonical_slug).first()
+    if legacy is None:
+        return
+    if canonical is None:
+        legacy.slug = canonical_slug
+        legacy.name = canonical_name
+        legacy.description = canonical_description
+        legacy.is_active = True
+        legacy.accepts_external_tickets = True
+        legacy.show_in_customer_form = True
+        legacy.save()
+        return
+
+    canonical.permissions.add(*legacy.permissions.all())
+    TicketCategory.objects.filter(unit=legacy).update(unit=canonical)
+    Ticket.objects.filter(owning_unit=legacy).update(owning_unit=canonical)
+    Ticket.objects.filter(assigned_unit=legacy).update(assigned_unit=canonical)
+    GroupDispatch.objects.filter(unit=legacy).update(unit=canonical)
+    TicketMagicLink.objects.filter(created_unit=legacy).update(created_unit=canonical)
+    StaffProfile.objects.filter(primary_unit=legacy).update(primary_unit=canonical)
+    for membership in Membership.objects.filter(unit=legacy):
+        existing = Membership.objects.filter(user_id=membership.user_id, unit=canonical).first()
+        if existing is None:
+            membership.unit = canonical
+            membership.save(update_fields=["unit", "updated_at"])
+            continue
+        existing.permissions.add(*membership.permissions.all())
+        existing.is_active = existing.is_active or membership.is_active
+        existing.is_primary = existing.is_primary or membership.is_primary
+        if membership.ticket_scope == "UNIT":
+            existing.ticket_scope = "UNIT"
+        existing.save(update_fields=["is_active", "is_primary", "ticket_scope", "updated_at"])
+        membership.delete()
+    for group in ContactGroup.objects.filter(unit=legacy):
+        existing = ContactGroup.objects.filter(unit=canonical, name=group.name).first()
+        if existing is None:
+            group.unit = canonical
+            group.save(update_fields=["unit", "updated_at"])
+            continue
+        for profile in CustomerProfile.objects.filter(groups=group):
+            profile.groups.add(existing)
+        GroupDispatch.objects.filter(contact_group=group).update(contact_group=existing)
+        group.delete()
+    for template in SMSTemplate.objects.filter(unit=legacy):
+        if SMSTemplate.objects.filter(unit=canonical, event_type=template.event_type).exists():
+            template.delete()
+        else:
+            template.unit = canonical
+            template.save(update_fields=["unit", "updated_at"])
+    legacy.delete()
+
+
+def merge_legacy_warehouse_units(apps, schema_editor):
+    merge_unit(apps, "logistics", "warehouse-loading", "بارگیری انبار", "بارگیری، انبار و ارسال")
+    merge_unit(apps, "warehouse", "warehouse-loading", "بارگیری انبار", "بارگیری، انبار و ارسال")
+
+
+class Migration(migrations.Migration):
+    dependencies = [("tickets", "0013_merge_legacy_accounting_unit")]
+    operations = [migrations.RunPython(merge_legacy_warehouse_units, migrations.RunPython.noop)]