Fix empty relationships being None in LAYOUT_FIGURE

Belval · Belval · commit 5df8e780f96b · 2024-03-21T12:12:01.000Z
diff --git a/textractor/parsers/response_parser.py b/textractor/parsers/response_parser.py
@@ -752,7 +752,7 @@ def _create_layout_objects(
                 )
             )
             parsed_blocks.add(block["Id"])
-            for relationship in block.get("Relationships", []):
+            for relationship in (block.get("Relationships", []) or []):
                 if relationship["Type"] != "CHILD":
                     continue
                 for leaf_id in relationship["Ids"]:
@@ -770,7 +770,7 @@ def _create_layout_objects(
                         )
                     )
                     container_layouts[-1].children[-1].raw_object = block
-                    for relationship in block.get("Relationships", []):
+                    for relationship in (block.get("Relationships", []) or []):
                         if relationship["Type"] != "CHILD":
                             continue
                         container_layouts[-1].children[-1].add_children(
@@ -789,7 +789,7 @@ def _create_layout_objects(
                 )
             )
             leaf_layouts[-1].raw_object = block
-            for relationship in block.get("Relationships", []):
+            for relationship in (block.get("Relationships", []) or []):
                 if relationship["Type"] != "CHILD":
                     continue
                 leaf_layouts[-1].add_children(

Original file line number	Diff line number	Diff line change
`@@ -752,7 +752,7 @@ def _create_layout_objects(`
`752`	`752`	`)`
`753`	`753`	`)`
`754`	`754`	`parsed_blocks.add(block["Id"])`
`755`		`- for relationship in block.get("Relationships", []):`
	`755`	`+ for relationship in (block.get("Relationships", []) or []):`
`756`	`756`	`if relationship["Type"] != "CHILD":`
`757`	`757`	`continue`
`758`	`758`	`for leaf_id in relationship["Ids"]:`
`@@ -770,7 +770,7 @@ def _create_layout_objects(`
`770`	`770`	`)`
`771`	`771`	`)`
`772`	`772`	`container_layouts[-1].children[-1].raw_object = block`
`773`		`- for relationship in block.get("Relationships", []):`
	`773`	`+ for relationship in (block.get("Relationships", []) or []):`
`774`	`774`	`if relationship["Type"] != "CHILD":`
`775`	`775`	`continue`
`776`	`776`	`container_layouts[-1].children[-1].add_children(`
`@@ -789,7 +789,7 @@ def _create_layout_objects(`
`789`	`789`	`)`
`790`	`790`	`)`
`791`	`791`	`leaf_layouts[-1].raw_object = block`
`792`		`- for relationship in block.get("Relationships", []):`
	`792`	`+ for relationship in (block.get("Relationships", []) or []):`
`793`	`793`	`if relationship["Type"] != "CHILD":`
`794`	`794`	`continue`
`795`	`795`	`leaf_layouts[-1].add_children(`