]> git.phdru.name Git - bookmarks_db.git/commitdiff
Fix(Py3): Guess input file encoding
authorOleg Broytman <phd@phdru.name>
Mon, 13 Nov 2023 15:13:14 +0000 (18:13 +0300)
committerOleg Broytman <phd@phdru.name>
Mon, 13 Nov 2023 19:02:26 +0000 (22:02 +0300)
Robots/bkmk_rurllib_py3.py

index 2b73347568813505a270d9308e1595172aa98819..1da8ec879e96925e40a8ab9cb9fe2e54407ffb7c 100644 (file)
@@ -106,10 +106,31 @@ class robot_urllib_py3(robot_base):
                     # Remove Accept-Charset
                     del opener.addheaders[-1]
 
-            infile = open(fname, 'rt')
-            content = infile.read()
-            infile.close()
-
+            possible_encodings = []
+            for encoding in (
+                    bookmark.charset,
+                    sys.getfilesystemencoding(),
+                    'utf-8',
+            ):
+                if encoding and encoding not in possible_encodings:
+                    possible_encodings.append(encoding)
+            content = None
+            for encoding in possible_encodings:
+                infile = open(fname, 'rt', encoding=encoding)
+                try:
+                    content = infile.read()
+                except UnicodeDecodeError:
+                    infile.close()
+                    continue
+                else:
+                    break
+                infile.close()
+
+            if content is None:
+                return (
+                    'ERROR: File encoding was not recognized',
+                    None, None, None, None
+                )
             return None, None, None, headers, content
 
         except RedirectException as e: