chore(export): Added ability to export chart YAML files with Unicode …

…characters, fix apache#20331
xyb · May 22, 2024 · f87f1f9 · f87f1f9
1 parent 4fa7619
commit f87f1f9
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 6 deletions.
diff --git a/superset/commands/chart/export.py b/superset/commands/chart/export.py
@@ -71,7 +71,7 @@ def _file_content(model: Slice) -> str:
         if model.table:
             payload["dataset_uuid"] = str(model.table.uuid)
 
-        file_content = yaml.safe_dump(payload, sort_keys=False)
+        file_content = yaml.safe_dump(payload, sort_keys=False, allow_unicode=True)
         return file_content
 
     @staticmethod

diff --git a/superset/commands/dashboard/export.py b/superset/commands/dashboard/export.py
@@ -147,7 +147,7 @@ def _file_content(model: Dashboard) -> str:
 
         payload["version"] = EXPORT_VERSION
 
-        file_content = yaml.safe_dump(payload, sort_keys=False)
+        file_content = yaml.safe_dump(payload, sort_keys=False, allow_unicode=True)
         return file_content
 
     @staticmethod

diff --git a/superset/commands/database/export.py b/superset/commands/database/export.py
@@ -99,7 +99,7 @@ def _file_content(model: Database) -> str:
 
         payload["version"] = EXPORT_VERSION
 
-        file_content = yaml.safe_dump(payload, sort_keys=False)
+        file_content = yaml.safe_dump(payload, sort_keys=False, allow_unicode=True)
         return file_content
 
     @staticmethod
@@ -131,6 +131,9 @@ def _export(
                 yield (
                     file_path,
                     functools.partial(  # type: ignore
-                        yaml.safe_dump, payload, sort_keys=False
+                        yaml.safe_dump,
+                        payload,
+                        sort_keys=False,
+                        allow_unicode=True,
                     ),
                 )
diff --git a/superset/commands/dataset/export.py b/superset/commands/dataset/export.py
@@ -78,7 +78,7 @@ def _file_content(model: SqlaTable) -> str:
         payload["version"] = EXPORT_VERSION
         payload["database_uuid"] = str(model.database.uuid)
 
-        file_content = yaml.safe_dump(payload, sort_keys=False)
+        file_content = yaml.safe_dump(payload, sort_keys=False, allow_unicode=True)
         return file_content
 
     @staticmethod
@@ -122,4 +122,7 @@ def _export(
 
             payload["version"] = EXPORT_VERSION
 
-            yield file_path, lambda: yaml.safe_dump(payload, sort_keys=False)
+            yield (
+                file_path,
+                lambda: yaml.safe_dump(payload, sort_keys=False, allow_unicode=True),
+            )