update import mp engagement to not use external_id

mysociety · Feb 19, 2024 · ae93110 · ae93110
1 parent b5f2292
commit ae93110
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/hub/management/commands/import_mp_engagement.py b/hub/management/commands/import_mp_engagement.py
@@ -4,7 +4,7 @@
 import pandas as pd
 from tqdm import tqdm
 
-from hub.models import DataSet, DataType, Person, PersonData
+from hub.models import DataSet, DataType, PersonData
 
 
 class Command(BaseCommand):
@@ -19,8 +19,9 @@ def handle(self, quiet=False, *args, **options):
 
     def get_person_from_id(self, id):
         try:
-            return Person.objects.get(external_id=id)
-        except Person.DoesNotExist:
+            data = PersonData.objects.get(data_type__name="parlid", data=id)
+            return data.person
+        except PersonData.DoesNotExist:
             return None
 
     def get_df(self):