Co-authored-by: Copilot <copilot@github.com>
This commit is contained in:
2026-04-24 21:58:27 +03:00
parent 3112e6d4bf
commit 43a0e578c1

View File

@@ -857,13 +857,17 @@ class Limoka(loader.Module):
self._invalid_banners.add(url) self._invalid_banners.add(url)
return None return None
ct = response.headers.get("Content-Type", "").lower() ct = response.headers.get("Content-Type", "").lower()
mime = None
if ct.startswith("image/"):
return url
if not ct: # Some servers don't respond to HEAD requests with Content-Type, so instead we will try guess mime from content if not ct: # Some servers don't respond to HEAD requests with Content-Type, so instead we will try guess mime from content
data = open(response.data, "rb").read(2048) async with session.get(url, timeout=5) as get_response:
mime = filetype.guess_mime(data, mime=True) data = await get_response.read(2048)
if not ct.startswith("image/") or not mime.startswith("image/"): mime = filetype.guess_mime(data, mime=True)
self._invalid_banners.add(url) if mime and mime.startswith("image/"):
return None return url
return url self._invalid_banners.add(url)
return None
except Exception: except Exception:
if url: if url:
self._invalid_banners.add(url) self._invalid_banners.add(url)
@@ -1712,7 +1716,8 @@ class Limoka(loader.Module):
) )
try: try:
result = SearchIndex(args.lower(), self.ix).search() result = SearchIndex(args.lower(), self.ix).search()
except Exception: except Exception as e:
logger.exception(f"Error occurred while searching: {e}")
return await utils.answer(message, self.strings["?"]) return await utils.answer(message, self.strings["?"])
if not result: if not result:
return await utils.answer(message, self.strings["404"].format(query=args)) return await utils.answer(message, self.strings["404"].format(query=args))