Merge pull request #19 from brc-dd/fix/18
Fix character decoding issues with text-like files
This commit is contained in:
commit
ed91e8b534
4 changed files with 20 additions and 3 deletions
|
|
@ -38,6 +38,7 @@ dependencies = [
|
||||||
"youtube-transcript-api",
|
"youtube-transcript-api",
|
||||||
"SpeechRecognition",
|
"SpeechRecognition",
|
||||||
"pathvalidate",
|
"pathvalidate",
|
||||||
|
"charset-normalizer",
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.urls]
|
[project.urls]
|
||||||
|
|
|
||||||
|
|
@ -26,6 +26,7 @@ import pptx
|
||||||
import puremagic
|
import puremagic
|
||||||
import requests
|
import requests
|
||||||
from bs4 import BeautifulSoup
|
from bs4 import BeautifulSoup
|
||||||
|
from charset_normalizer import from_path
|
||||||
|
|
||||||
# Optional Transcription support
|
# Optional Transcription support
|
||||||
try:
|
try:
|
||||||
|
|
@ -161,9 +162,7 @@ class PlainTextConverter(DocumentConverter):
|
||||||
elif "text/" not in content_type.lower():
|
elif "text/" not in content_type.lower():
|
||||||
return None
|
return None
|
||||||
|
|
||||||
text_content = ""
|
text_content = str(from_path(local_path).best())
|
||||||
with open(local_path, "rt", encoding="utf-8") as fh:
|
|
||||||
text_content = fh.read()
|
|
||||||
return DocumentConverterResult(
|
return DocumentConverterResult(
|
||||||
title=None,
|
title=None,
|
||||||
text_content=text_content,
|
text_content=text_content,
|
||||||
|
|
|
||||||
4
tests/test_files/test_mskanji.csv
Normal file
4
tests/test_files/test_mskanji.csv
Normal file
|
|
@ -0,0 +1,4 @@
|
||||||
|
–¼‘O,”N—î,<2C>Z<EFBFBD>Š
|
||||||
|
<EFBFBD>²“¡‘¾˜Y,30,“Œ‹ž
|
||||||
|
ŽO–؉pŽq,25,‘å<E28098>ã
|
||||||
|
îà‹´<EFBFBD>~,35,–¼ŒÃ‰®
|
||||||
|
|
|
@ -87,6 +87,13 @@ SERP_TEST_EXCLUDES = [
|
||||||
"data:image/svg+xml,%3Csvg%20width%3D",
|
"data:image/svg+xml,%3Csvg%20width%3D",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
CSV_CP932_TEST_STRINGS = [
|
||||||
|
"名前,年齢,住所",
|
||||||
|
"佐藤太郎,30,東京",
|
||||||
|
"三木英子,25,大阪",
|
||||||
|
"髙橋淳,35,名古屋",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(
|
@pytest.mark.skipif(
|
||||||
skip_remote,
|
skip_remote,
|
||||||
|
|
@ -164,6 +171,12 @@ def test_markitdown_local() -> None:
|
||||||
for test_string in SERP_TEST_STRINGS:
|
for test_string in SERP_TEST_STRINGS:
|
||||||
assert test_string in text_content
|
assert test_string in text_content
|
||||||
|
|
||||||
|
## Test non-UTF-8 encoding
|
||||||
|
result = markitdown.convert(os.path.join(TEST_FILES_DIR, "test_mskanji.csv"))
|
||||||
|
text_content = result.text_content.replace("\\", "")
|
||||||
|
for test_string in CSV_CP932_TEST_STRINGS:
|
||||||
|
assert test_string in text_content
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(
|
@pytest.mark.skipif(
|
||||||
skip_exiftool,
|
skip_exiftool,
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue