Add convert str2object (#326)

memgraph · Sep 7, 2023 · 9fdeb79 · 9fdeb79
1 parent 6000bbb
commit 9fdeb79
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/e2e/convert_test/test_str2object1/file1 b/e2e/convert_test/test_str2object1/file1
@@ -0,0 +1,4 @@
+"_id","_labels","a","b","prop","did_it","_start","_end","_type"
+"0",":Dog","3","[""1"", ""2"", 3]","2","","","",""
+"1",":Human","","[[""1"",{""key"": 5}]]","","yes","","",""
+"5",":Cat","","{""prop"": {""prop2"": {""key"": 10}}}","","","0","1","loves"
diff --git a/e2e/convert_test/test_str2object1/input.cyp b/e2e/convert_test/test_str2object1/input.cyp
diff --git a/e2e/convert_test/test_str2object1/test.yml b/e2e/convert_test/test_str2object1/test.yml
@@ -0,0 +1,8 @@
+query: >
+    LOAD CSV FROM "/_file1" WITH header AS row
+    RETURN convert.str2object(row.b) AS list;
+
+output: 
+  - list: [ "1","2", 3]
+  - list: [["1",{"key": 5}]]
+  - list: {"prop": {"prop2": {"key": 10}}}
diff --git a/python/convert.py b/python/convert.py
@@ -0,0 +1,9 @@
+import mgp
+from json import loads
+
+
+@mgp.function
+def str2object(string: str) -> mgp.Any:
+    if string:
+        return loads(string)
+    return None