Handle special case for states

MaRDI4NFDI · Jan 9, 2024 · 4018e5a · 4018e5a
1 parent a081191
commit 4018e5a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/src/zbmath_rest2oai/getAsXml.py b/src/zbmath_rest2oai/getAsXml.py
@@ -7,8 +7,15 @@ def final_xml2(de):
     headers = {'Accept': 'application/json'}
 
     r = requests.get('https://api.zbmath.org/v1/document/' + de, headers=headers)
-
-    return dict2xml.Converter(wrap="root").build(r.json(), closed_tags_for=[[], '', [None], None])
-
-
-print(final_xml2("6383667"))
+    json = r.json()
+    states = {}
+    for lst in json['result']['states']:
+        [k, v] = lst
+        states[k] = v
+    json['result']['states'] = states
+    return (
+        dict2xml.Converter(wrap="root")
+        .build(r.json(), closed_tags_for=[
+            [], '', [None], None
+        ])
+    )
diff --git a/tests/test_plain_xml.py b/tests/test_plain_xml.py
@@ -17,7 +17,7 @@ def test_similarity(self):
                 'ratio_mode': 'accurate'
             })
             essentials = list(filter(lambda e: not isinstance(e, MoveNode), diff))
-            self.assertLessEqual(len(essentials), 46)
+            self.assertLessEqual(len(essentials), 48)
             diff_text = main.diff_texts(expected_string, real_string, {
                 'ratio_mode': 'accurate'
             }, formatter=formatting.XMLFormatter(normalize=formatting.WS_BOTH))