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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
# Generated by Django 2.2.24 on 2024-03-07 17:52
from django.db import migrations
from django.db.models.functions import Length
from ishtar_common.data import FRENCH_STATES, FRENCH_TOM_STATES
def migrate(apps, __):
Area = apps.get_model('ishtar_common', 'Area')
Department = apps.get_model('ishtar_common', 'Department')
State = apps.get_model('ishtar_common', 'State')
Town = apps.get_model('ishtar_common', 'Town')
if not State.objects.filter(label="Nord-Pas-de-Calais").count():
# already OK or not french
return
State.objects.filter(pk__isnull=False).delete()
Area.objects.filter(txt_idx__startswith="dep-").delete()
Area.objects.filter(txt_idx__startswith="state-").delete()
for lbl, code, departments in (FRENCH_STATES + FRENCH_TOM_STATES):
state = State.objects.create(label=lbl, number=code)
state_area = Area.objects.create(label=lbl, txt_idx=f'state-{code}')
for dep_code in departments:
q = Department.objects.filter(number=dep_code)
if not q.count():
print(f"{dep_code} department is missing")
continue
dep = q.all()[0]
dep.state = state
dep.save()
dep_area = Area.objects.create(label=dep.label, txt_idx=f'dep-{dep.number}')
dep_area.parent = state_area
dep_area.save()
dep_area.towns.clear()
q2 = Town.objects.annotate(insee_len=Length('numero_insee')).filter(
numero_insee__startswith=dep_code, insee_len=5)
dep_area.towns.add(*list(q2.all()))
class Migration(migrations.Migration):
dependencies = [
('ishtar_common', '0240_state_number_max_len'),
]
operations = [
migrations.RunPython(migrate)
]
|