Coverage for custom_components/supernotify/transports/media_player.py: 41%

1import logging 

2import urllib.parse 

3from typing import Any 

4 

5from homeassistant.const import ( 

6 ATTR_ENTITY_ID, 

7) 

8 

9from custom_components.supernotify import ( 

10 ATTR_MEDIA, 

11 ATTR_MEDIA_SNAPSHOT_URL, 

12 OPTION_TARGET_CATEGORIES, 

13 OPTION_TARGET_SELECT, 

14 TRANSPORT_MEDIA, 

15) 

16from custom_components.supernotify.envelope import Envelope 

17from custom_components.supernotify.model import DebugTrace, TransportConfig, TransportFeature 

18from custom_components.supernotify.transport import Transport 

19 

20RE_VALID_MEDIA_PLAYER = r"media_player\.[A-Za-z0-9_]+" 

21 

22_LOGGER = logging.getLogger(__name__) 

23 

24 

25class MediaPlayerTransport(Transport): 

26 name = TRANSPORT_MEDIA 

27 

28 def __init__(self, *args: Any, **kwargs: Any) -> None: 

29 super().__init__(*args, **kwargs) 

30 

31 @property 

32 def supported_features(self) -> TransportFeature: 

33 return TransportFeature.IMAGES | TransportFeature.VIDEO 

34 

35 @property 

36 def default_config(self) -> TransportConfig: 

37 config = TransportConfig() 

38 config.delivery_defaults.action = "media_player.play_media" 

39 config.delivery_defaults.options = { 

40 OPTION_TARGET_SELECT: [RE_VALID_MEDIA_PLAYER], 

41 OPTION_TARGET_CATEGORIES: [ATTR_ENTITY_ID], 

42 } 

43 return config 

44 

45 async def deliver(self, envelope: Envelope, debug_trace: DebugTrace | None = None) -> bool: # noqa: ARG002 

46 _LOGGER.debug("SUPERNOTIFY notify_media: %s", envelope.data) 

47 

48 data: dict[str, Any] = envelope.data or {} 

49 media_players: list[str] = envelope.target.entity_ids or [] 

50 media_type: str = data.get("media_content_type", "image") 

51 if not media_players: 

52 _LOGGER.debug("SUPERNOTIFY skipping media show, no targets") 

53 return False 

54 

55 snapshot_url = data.get(ATTR_MEDIA, {}).get(ATTR_MEDIA_SNAPSHOT_URL) 

56 if snapshot_url is None: 

57 # fallback to older idiosyncratic way for backward compatibility 

58 snapshot_url = data.get(ATTR_MEDIA_SNAPSHOT_URL) 

59 if snapshot_url is None: 

60 _LOGGER.debug("SUPERNOTIFY skipping media player, no snapshot url") 

61 return False 

62 # absolutize relative URL for external URl, probably preferred by Alexa Show etc 

63 snapshot_url = urllib.parse.urljoin(self.hass_api.external_url, snapshot_url) 

64 

65 action_data: dict[str, Any] = {"media": {"media_content_id": snapshot_url, "media_content_type": media_type}} 

66 if data and data.get("announce"): 

67 action_data["announce"] = data.get("announce") 

68 if data and data.get("enqueue"): 

69 action_data["enqueue"] = data.get("enqueue") 

70 

71 return await self.call_action(envelope, action_data=action_data, target_data={ATTR_ENTITY_ID: media_players})