summaryrefslogtreecommitdiff
path: root/archaeological_warehouse/migrations/0125_migrate_cached_town.py
blob: 49f914185353f6c902c23b0bef16949ec4798537 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# Generated by Django 2.2.24 on 2025-10-24 12:59

from django.db import migrations


def _generate_cached_town(Town, town):
    if town.precise_town_id:
        return Town.objects.get(id=town.precise_town_id).name
    return town.town or ""


def migrate_cached_towns(apps, __):
    Town = apps.get_model("ishtar_common", "Town")
    Warehouse = apps.get_model("archaeological_warehouse", "Warehouse")
    for warehouse in Warehouse.objects.all():
        Warehouse.objects.filter(pk=warehouse.pk).update(
            cached_town=_generate_cached_town(Town, warehouse),
        )


class Migration(migrations.Migration):

    dependencies = [
        ('archaeological_warehouse', '0124_warehouse_cached_town'),
    ]

    operations = [
        migrations.RunPython(migrate_cached_towns)
    ]