fix: site icon not showing (#11094)

This commit is contained in:
Hash Brown 2024-11-26 10:23:03 +08:00 committed by GitHub
parent 17ee731546
commit 0f85e3557b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -31,12 +31,12 @@ class AppIconUrlField(fields.Raw):
if obj is None:
return None
from models.model import App, IconType
from models.model import App, IconType, Site
if isinstance(obj, dict) and "app" in obj:
obj = obj["app"]
if isinstance(obj, App) and obj.icon_type == IconType.IMAGE.value:
if isinstance(obj, App | Site) and obj.icon_type == IconType.IMAGE.value:
return file_helpers.get_signed_file_url(obj.icon)
return None