From 69ac1b74c6480615764f0afe2b06a1ee8fda761e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laura=20Kl=C3=BCnder?= <laura@codingcatgirl.de>
Date: Wed, 10 May 2017 21:24:31 +0200
Subject: [PATCH] convert all remaining AreaLocations

---
 .../0088_locationgroup_compiled_area.py       | 20 ++++++
 .../migrations/0089_assign_location_area.py   | 62 +++++++++++++++++++
 src/c3nav/mapdata/models/locations.py         |  1 +
 3 files changed, 83 insertions(+)
 create mode 100644 src/c3nav/mapdata/migrations/0088_locationgroup_compiled_area.py
 create mode 100644 src/c3nav/mapdata/migrations/0089_assign_location_area.py

diff --git a/src/c3nav/mapdata/migrations/0088_locationgroup_compiled_area.py b/src/c3nav/mapdata/migrations/0088_locationgroup_compiled_area.py
new file mode 100644
index 00000000..0fca875e
--- /dev/null
+++ b/src/c3nav/mapdata/migrations/0088_locationgroup_compiled_area.py
@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10.7 on 2017-05-10 18:57
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mapdata', '0087_assign_location_roomsegment_poi'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='locationgroup',
+            name='compiled_area',
+            field=models.BooleanField(default=False, verbose_name='is a compiled area'),
+        ),
+    ]
diff --git a/src/c3nav/mapdata/migrations/0089_assign_location_area.py b/src/c3nav/mapdata/migrations/0089_assign_location_area.py
new file mode 100644
index 00000000..64bd63ad
--- /dev/null
+++ b/src/c3nav/mapdata/migrations/0089_assign_location_area.py
@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10.7 on 2017-05-10 19:13
+from __future__ import unicode_literals
+
+from django.db import migrations
+
+from c3nav.mapdata.utils.geometry import assert_multipolygon
+
+
+def assign_locations_area(apps, schema_editor):
+    Space = apps.get_model('mapdata', 'Space')
+    Area = apps.get_model('mapdata', 'Area')
+    AreaLocation = apps.get_model('mapdata', 'AreaLocation')
+    LocationSlug = apps.get_model('mapdata', 'LocationSlug')
+    LocationGroup = apps.get_model('mapdata', 'LocationGroup')
+    for obj in AreaLocation.objects.filter(location_type='area').order_by('slug'):
+        spaces = [s for s in Space.objects.filter(section=obj.section, level='')
+                  if s.geometry.intersection(obj.geometry).area / s.geometry.area > 0.10]
+
+        if not spaces:
+            obj.delete()
+            continue
+
+        total_spaces = [s for s in spaces
+                        if s.geometry.intersection(obj.geometry).area / s.geometry.area > 0.95]
+        partial_spaces = [s for s in spaces if s not in total_spaces]
+
+        to_obj = LocationGroup()
+        to_obj.locationslug_ptr = LocationSlug.objects.create(slug=obj.slug)
+        to_obj.titles = obj.titles
+        to_obj.can_search = obj.can_search
+        to_obj.can_describe = obj.can_describe
+        to_obj.color = obj.color
+        to_obj.compiled_area = True
+        to_obj.save()
+
+        for space in total_spaces:
+            space.groups.add(to_obj)
+
+        for space in partial_spaces:
+            for polygon in assert_multipolygon(space.geometry.intersection(obj.geometry)):
+                area = Area()
+                area.locationslug_ptr = LocationSlug.objects.create()
+                area.geometry = polygon
+                area.space = space
+                area.can_search = False
+                area.can_describe = False
+                area.save()
+                area.groups.add(to_obj)
+
+        obj.delete()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mapdata', '0088_locationgroup_compiled_area'),
+    ]
+
+    operations = [
+        migrations.RunPython(assign_locations_area),
+    ]
diff --git a/src/c3nav/mapdata/models/locations.py b/src/c3nav/mapdata/models/locations.py
index 3245b036..045eac53 100644
--- a/src/c3nav/mapdata/models/locations.py
+++ b/src/c3nav/mapdata/models/locations.py
@@ -74,6 +74,7 @@ class LegacyLocation:
 
 class LocationGroup(Location, EditorFormMixin, models.Model):
     compiled_room = models.BooleanField(default=False, verbose_name=_('is a compiled room'))
+    compiled_area = models.BooleanField(default=False, verbose_name=_('is a compiled area'))
 
     class Meta:
         verbose_name = _('Location Group')