add reform to party shade, fix party dataset shade updating

mysociety · Jul 5, 2024 · 16d4901 · 16d4901
1 parent 23d1264
commit 16d4901
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/hub/management/commands/base_importers.py b/hub/management/commands/base_importers.py
@@ -31,6 +31,7 @@
     "Sinn Féin": "#326760",
     "Social Democratic and Labour Party": "#2AA82C",
     "Reclaim": "#101122",
+    "Reform UK": "#3DBBE2",
     "No overall control": "#EEE",
     "Independents": "#DCDCDC",
 }

diff --git a/hub/management/commands/import_mps.py b/hub/management/commands/import_mps.py
@@ -282,12 +282,13 @@ def import_mps(self):
                             data=mp["Party"],
                         )
 
-        dataset = DataSet.objects.filter(name="party", options=list())
+        dataset = DataSet.objects.filter(name="party")
         if dataset:
             parties = list()
             all_parties = list(
                 PersonData.objects.filter(
-                    data_type__data_set__name="party"
+                    person__personarea__person_type="MP",
+                    data_type__data_set__name="party",
                 ).values_list("data")
             )