CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: Exception when unable to identify favicon mimetype

pull/888/head
Raphaël Vinot 2024-02-26 20:36:03 +01:00
parent b76d28de27
commit 29907dc379
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lookyloo/lookyloo.py
View File

@ -34,7 +34,7 @@ from lacuscore import (LacusCore,
CaptureSettings as CaptureSettingsCore)
from PIL import Image, UnidentifiedImageError
from playwrightcapture import get_devices
from puremagic import from_string # type: ignore[import-untyped]
from puremagic import from_string, PureError # type: ignore[import-untyped]
from pylacus import (PyLacus,
CaptureStatus as CaptureStatusPy
# CaptureResponse as CaptureResponsePy,
@ -867,8 +867,12 @@ class Lookyloo():
favicon = fav.getvalue()
if not favicon:
return '', ''
mimetype = from_string(favicon, mime=True)
return mimetype, base64.b64encode(favicon).decode()
try:
mimetype = from_string(favicon, mime=True)
return mimetype, base64.b64encode(favicon).decode()
except PureError:
self.logger.warning(f'Unable to get the mimetype of the favicon for {capture_uuid}.')
return '', ''
return fav
def get_html(self, capture_uuid: str, /, all_html: bool=False) -> BytesIO: