make uri migration more robust

acdh-oeaw · Sep 27, 2023 · de308ee · de308ee
1 parent 5e9dac2
commit de308ee
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/apis_ontology/management/commands/import.py b/apis_ontology/management/commands/import.py
@@ -76,10 +76,12 @@ def handle(self, *args, **options):
             for result in data["results"]:
                 print(result["url"])
                 newuri, created = Uri.objects.get_or_create(uri=result["uri"])
-                try:
-                    result["root_object"] = RootObject.objects.get(pk=result["entity"]["id"])
-                    for attribute in result:
-                        if hasattr(newuri, attribute):
-                            setattr(newuri, attribute, result[attribute])
-                except RootObject.DoesNotExist:
-                    pass
+                if hasattr(result["entity"], "id"):
+                    try:
+                        result["root_object"] = RootObject.objects.get(pk=result["entity"]["id"])
+                        for attribute in result:
+                            if hasattr(newuri, attribute):
+                                setattr(newuri, attribute, result[attribute])
+                        newuri.save()
+                    except RootObject.DoesNotExist:
+                        pass