Merge pull request #73 from wey-gu/escape_new_line_ng_load

wey-gu · web-flow · commit 17d8a96bc64d · 2024-08-16T16:07:36.000+08:00
feat: escape newline for ng_load
diff --git a/ngql/ng_load.py b/ngql/ng_load.py
@@ -266,7 +266,11 @@ def safe_load_file(source, file_type, header_option=None, limit=None):
                                 )
                             prop_str += "NULL, "
                         elif prop_schema_map[prop_name]["type"] == "string":
-                            raw_prop_str = prop_value.strip('"').replace('"', '\\"')
+                            raw_prop_str = (
+                                prop_value.strip('"')
+                                .replace('"', '\\"')
+                                .replace("\n", "\\n")
+                            )
                             prop_str += f"{QUOTE}{raw_prop_str}{QUOTE}, "
                         elif prop_schema_map[prop_name]["type"] == "date":
                             prop_str += f"date({QUOTE}{prop_value}{QUOTE}), "
@@ -346,7 +350,11 @@ def safe_load_file(source, file_type, header_option=None, limit=None):
                                 )
                             prop_str += "NULL, "
                         elif prop_schema_map[prop_name]["type"] == "string":
-                            raw_prop_str = prop_value.strip('"').replace('"', '\\"')
+                            raw_prop_str = (
+                                prop_value.strip('"')
+                                .replace('"', '\\"')
+                                .replace("\n", "\\n")
+                            )
                             prop_str += f"{QUOTE}{raw_prop_str}{QUOTE}, "
                         elif prop_schema_map[prop_name]["type"] == "date":
                             prop_str += f"date({QUOTE}{prop_value}{QUOTE}), "
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="jupyter_nebulagraph",
-    version="0.14.1",
+    version="0.14.2",
     author="Wey Gu",
     author_email="weyl.gu@gmail.com",
     description="Jupyter extension for NebulaGraph",
diff --git a/setup_ipython.py b/setup_ipython.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="ipython-ngql",
-    version="0.14.1",
+    version="0.14.2",
     author="Wey Gu",
     author_email="weyl.gu@gmail.com",
     description="Jupyter extension for NebulaGraph",
