From: Oleg Broytman <phd@phdru.name>
Date: Sun, 12 Nov 2023 18:19:58 +0000 (+0300)
Subject: Fix(robots): Process response without `Content-Type`
X-Git-Tag: 5.0.0~42
X-Git-Url: https://git.phdru.name/?p=bookmarks_db.git;a=commitdiff_plain;h=9f2e541f950d245198f893936de4ebece69894ba

Fix(robots): Process response without `Content-Type`

Try to recognize HTML.
---

diff --git a/Robots/bkmk_robot_base.py b/Robots/bkmk_robot_base.py
index 9a74cb4..b5cac69 100644
--- a/Robots/bkmk_robot_base.py
+++ b/Robots/bkmk_robot_base.py
@@ -126,6 +126,13 @@ class robot_base(Robot):
                 try:
                     content_type = headers["Content-Type"]
                     self.log("   Content-Type: %s" % content_type)
+                    if content_type is None:
+                        if 'html' in content.lower():
+                            content_type = 'text/html'
+                        else:
+                            content_type = 'text/plain'
+                        self.log("   Set Content-Type to: %s"
+                                 % content_type)
                     try:
                         # extract charset from
                         # "text/html; foo; charset=UTF-8, bar; baz;"
@@ -187,9 +194,11 @@ class robot_base(Robot):
                                 icons[icon_url] = None
                             else:
                                 content_type = icon_headers["Content-Type"]
-                                if content_type.startswith("application/") \
-                                   or content_type.startswith("image/") \
-                                   or content_type.startswith("text/plain"):
+                                if content_type and (
+                                    content_type.startswith("application/")
+                                    or content_type.startswith("image/")
+                                    or content_type.startswith("text/plain")
+                                ):
                                     bookmark.icon_href = icon_url
                                     self.log("   got icon       : %s"
                                              % content_type)