修复glossary导出csv时的bug
This commit is contained in:
@@ -1,3 +1,3 @@
|
||||
# SPDX-FileCopyrightText: 2025 QinHan
|
||||
# SPDX-License-Identifier: MPL-2.0
|
||||
__version__="1.3.0"
|
||||
__version__="1.3.0b1"
|
||||
@@ -1,5 +1,8 @@
|
||||
# SPDX-FileCopyrightText: 2025 QinHan
|
||||
# SPDX-License-Identifier: MPL-2.0
|
||||
import csv
|
||||
from io import StringIO
|
||||
|
||||
from docutranslate.ir.document import Document
|
||||
|
||||
|
||||
@@ -24,9 +27,12 @@ class Glossary:
|
||||
return prompt
|
||||
else:
|
||||
return ""
|
||||
|
||||
@staticmethod
|
||||
def glossary_dict2csv(glossary_dict: dict[str, str], seperator=",", stem="glossary_gen") -> Document:
|
||||
content = f"src{seperator}dst\n"
|
||||
for src, dst in glossary_dict.items():
|
||||
content += f"{src}{seperator}{dst}\n"
|
||||
return Document.from_bytes(content=content.encode("utf-8"), suffix=".csv", stem=stem)
|
||||
def glossary_dict2csv(glossary_dict: dict[str, str], delimiter=",", stem="glossary_gen") -> Document:
|
||||
csv_rows = [[src, dst] for src, dst in glossary_dict.items()]
|
||||
content = StringIO()
|
||||
writer = csv.writer(content, delimiter=delimiter)
|
||||
writer.writerow(['src', 'dst'])
|
||||
writer.writerows(csv_rows)
|
||||
return Document.from_bytes(content=content.getvalue().encode("utf-8"), suffix=".csv", stem=stem)
|
||||
|
||||
Reference in New Issue
Block a user