Merge pull request #73 from wey-gu/escape_new_line_ng_load

feat: escape newline for ng_load
wey-gu · Aug 16, 2024 · 17d8a96 · 17d8a96
2 parents 41dc190 + 247a069
commit 17d8a96
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
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="[email protected]",
     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="[email protected]",
     description="Jupyter extension for NebulaGraph",