| @@ -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 | |||