update non standard importers to convert to new area type

mysociety · Dec 11, 2023 · 31d415e · 31d415e
1 parent f366081
commit 31d415e
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 20 deletions.
diff --git a/hub/management/commands/import_hnh_polling_data.py b/hub/management/commands/import_hnh_polling_data.py
@@ -6,6 +6,7 @@
 from tqdm import tqdm
 
 from hub.models import Area, AreaData, DataSet, DataType
+from hub.transformers import DataTypeConverter
 
 from .base_importers import BaseImportFromDataFrameCommand
 
@@ -408,6 +409,11 @@ def extract_and_save_data(self):
                             defaults={"data": value},
                         )
 
+            converter = DataTypeConverter()
+            for data_type in data_types.values():
+                if data_type.area_type == converter.old_area_type:
+                    converter.convert_datatype_to_new_geography(data_type)
+
     def delete_data(self):
         self.log("Deleting existing AreaData objects for DataTypes:")
         for file in self.files:

diff --git a/hub/management/commands/import_onward_polling_data.py b/hub/management/commands/import_onward_polling_data.py
@@ -42,35 +42,31 @@ class Command(BaseImportFromDataFrameCommand):
 
     data_sets = {
         "constituency_nz_support": {
-            "defaults": defaults.copy().extend(
-                {
-                    "description": "Estimated percentage of constituents who support net zero."
-                }
-            ),
+            "defaults": {
+                **defaults,
+                "description": "Estimated percentage of constituents who support net zero.",
+            },
             "col": "support net zero",
         },
         "constituency_nz_neutral": {
-            "defaults": defaults.extend(
-                {
-                    "description": "Estimated percentage of constituents who neither support nor oppose net zero."
-                }
-            ),
+            "defaults": {
+                **defaults,
+                "description": "Estimated percentage of constituents who neither support nor oppose net zero.",
+            },
             "col": "neither support nor oppose net zero",
         },
         "constituency_nz_oppose": {
-            "defaults": defaults.extend(
-                {
-                    "description": "Estimated percentage of constituents who oppose net zero."
-                }
-            ),
+            "defaults": {
+                **defaults,
+                "description": "Estimated percentage of constituents who oppose net zero.",
+            },
             "col": "oppose net zero",
         },
         "constituency_cc_high": {
-            "defaults": defaults.copy().extend(
-                {
-                    "description": "Estimated percentage of constituents who consider climate change a high priority."
-                }
-            ),
+            "defaults": {
+                **defaults,
+                "description": "Estimated percentage of constituents who consider climate change a high priority.",
+            },
             "col": "consider climate change a high priority",
         },
     }

diff --git a/hub/management/commands/import_red_blue_wall_constituencies.py b/hub/management/commands/import_red_blue_wall_constituencies.py
@@ -31,6 +31,7 @@ class Command(BaseImportFromDataFrameCommand):
         ],
         "data_url": "",
         "comparators": DataSet.in_comparators(),
+        "data_type": "text",
     }
 
     data_sets = {