Fix target service attribute on Mastodon integration (#122546)

* Fix target

* Fix
This commit is contained in:
Andrew Jackson 2024-07-24 17:19:12 +01:00 committed by Franck Nijhof
parent 75f0384a15
commit 586a0b12ab
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3

View File

@ -3,7 +3,7 @@
from __future__ import annotations from __future__ import annotations
import mimetypes import mimetypes
from typing import Any from typing import Any, cast
from mastodon import Mastodon from mastodon import Mastodon
from mastodon.Mastodon import MastodonAPIError, MastodonUnauthorizedError from mastodon.Mastodon import MastodonAPIError, MastodonUnauthorizedError
@ -71,11 +71,15 @@ class MastodonNotificationService(BaseNotificationService):
def send_message(self, message: str = "", **kwargs: Any) -> None: def send_message(self, message: str = "", **kwargs: Any) -> None:
"""Toot a message, with media perhaps.""" """Toot a message, with media perhaps."""
target = None
if (target_list := kwargs.get(ATTR_TARGET)) is not None:
target = cast(list[str], target_list)[0]
data = kwargs.get(ATTR_DATA) data = kwargs.get(ATTR_DATA)
media = None media = None
mediadata = None mediadata = None
target = None
sensitive = False sensitive = False
content_warning = None content_warning = None
@ -87,7 +91,6 @@ class MastodonNotificationService(BaseNotificationService):
return return
mediadata = self._upload_media(media) mediadata = self._upload_media(media)
target = data.get(ATTR_TARGET)
sensitive = data.get(ATTR_MEDIA_WARNING) sensitive = data.get(ATTR_MEDIA_WARNING)
content_warning = data.get(ATTR_CONTENT_WARNING) content_warning = data.get(ATTR_CONTENT_WARNING)