Download records as excel fix

code-kern-ai · Nov 4, 2024 · f922923 · f922923
1 parent 0456f99
commit f922923
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/controller/transfer/export_parser.py b/controller/transfer/export_parser.py
@@ -13,6 +13,8 @@
 
 from util.sql_helper import parse_sql_text
 
+ILLEGAL_CHARACTER_REG_EX = r"[\000-\010]|[\013-\014]|[\016-\037]"
+
 
 def parse(
     project_id: str,
@@ -43,6 +45,10 @@ def parse(
     elif file_type == enums.RecordExportFileTypes.CSV.value:
         df.to_csv(file_path, index=False)
     elif file_type == enums.RecordExportFileTypes.XLSX.value:
+        for column in df.columns:
+            type_name = df[column].dtype.name
+            if type_name == "object":
+                df[column] = df[column].str.replace(ILLEGAL_CHARACTER_REG_EX, "")
         df.to_excel(file_path)
     else:
         message = f"File type {file_type} not supported."

diff --git a/submodules/model b/submodules/model