add support for non-ascii character in string

tigergraph · Nov 5, 2024 · 7b2eee7 · 7b2eee7
1 parent 4392510
commit 7b2eee7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/pyTigerGraph/pyTigerGraphLoading.py b/pyTigerGraph/pyTigerGraphLoading.py
@@ -161,8 +161,13 @@ def runLoadingJobWithData(self, data: str, fileTag: str, jobName: str, sep: str
         if eol is not None:
             params["eol"] = eol
 
-        res = self._req("POST", self.restppUrl + "/ddl/" + self.graphname, params=params, data=data,
-                        headers={"RESPONSE-LIMIT": str(sizeLimit), "GSQL-TIMEOUT": str(timeout)})
+        if isinstance(data, str):
+            data = data.encode("utf-8")
+            res = self._req("POST", self.restppUrl + "/ddl/" + self.graphname, params=params, data=data,
+                            headers={"Content-Type": "application/x-www-form-urlencoded; Charset=utf-8", "RESPONSE-LIMIT": str(sizeLimit), "GSQL-TIMEOUT": str(timeout)})
+        else:
+            res = self._req("POST", self.restppUrl + "/ddl/" + self.graphname, params=params, data=data,
+                            headers={"RESPONSE-LIMIT": str(sizeLimit), "GSQL-TIMEOUT": str(timeout)})
 
         if logger.level == logging.DEBUG:
             logger.debug("return: " + str(res))

diff --git a/pyTigerGraph/pytgasync/pyTigerGraphLoading.py b/pyTigerGraph/pytgasync/pyTigerGraphLoading.py
@@ -161,8 +161,13 @@ async def runLoadingJobWithData(self, data: str, fileTag: str, jobName: str, sep
         if eol is not None:
             params["eol"] = eol
 
-        res = await self._req("POST", self.restppUrl + "/ddl/" + self.graphname, params=params, data=data,
-                              headers={"RESPONSE-LIMIT": str(sizeLimit), "GSQL-TIMEOUT": str(timeout)})
+        if isinstance(data, str):
+            data = data.encode("utf-8")
+            res = await self._req("POST", self.restppUrl + "/ddl/" + self.graphname, params=params, data=data,
+                            headers={"Content-Type": "application/x-www-form-urlencoded; Charset=utf-8", "RESPONSE-LIMIT": str(sizeLimit), "GSQL-TIMEOUT": str(timeout)})
+        else:
+            res = await self._req("POST", self.restppUrl + "/ddl/" + self.graphname, params=params, data=data,
+                            headers={"RESPONSE-LIMIT": str(sizeLimit), "GSQL-TIMEOUT": str(timeout)})
 
         if logger.level == logging.DEBUG:
             logger.debug("return: " + str(res))