Merge pull request #3 from lindell/play_random

Added play_random

Author: lindell

README.md

@@ -25,6 +25,16 @@ Play the latest video or clip from a specific program. There exist an option to
     category: Intervjuer # Optional
 ```
 
+### Play random
+Play a random video or clip from a specific program. There exist an option to just random from specific categories.
+```yaml
+- service: svt_play.play_random
+  entity_id: media_player.living_room_tv
+  data:
+    program_name: skavlan
+    category: Intervjuer # Optional
+```
+
 ### Play Channel
 Play one of the svt channels.
 ```yaml
@@ -40,7 +50,7 @@ Play one of the svt channels.
 
 Copy the `custom_components/svt_play` folder in this repository to `<home assistant config>/custom_components/svt_play`
 
-Or: 
+Or:
 
 Install via [Home Assistant Community Store](https://hacs.xyz/)
 

custom_components/svt_play/__init__.py

@@ -1,7 +1,7 @@
 import logging
 import homeassistant.helpers.config_validation as cv
 import voluptuous as vol
-from .video_url_fetch.video_fetch import video_url_by_video_id, video_id_by_time, video_url_by_channel, suggested_video_id
+from .video_url_fetch.video_fetch import video_url_by_video_id, video_id_by_time, video_url_by_channel, suggested_video_id, random_video_id
 from .validation import category_names
 
 DOMAIN = "svt_play"
@@ -26,6 +26,13 @@
     CONF_CATEGORY: category_names,
 })
 
+SERVICE_PLAY_RANDOM = 'play_random'
+SERVICE_PLAY_RANDOM_SCHEMA = vol.Schema({
+    CONF_ENTITY_ID: cv.entity_ids,
+    CONF_PROGRAM_NAME: str,
+    CONF_CATEGORY: category_names,
+})
+
 SERVICE_PLAY_CHANNEL = 'play_channel'
 SERVICE_PLAY_CHANNEL_SCHEMA = vol.Schema({
     CONF_ENTITY_ID: cv.entity_ids,
@@ -75,6 +82,26 @@ async def play_latest(service):
         DOMAIN, SERVICE_PLAY_LATEST, play_latest, SERVICE_PLAY_LATEST_SCHEMA
     )
 
+    async def play_random(service):
+        """Play a random svt play video from a specified program"""
+
+        entity_id = service.data.get(CONF_ENTITY_ID)
+        program_name = service.data.get(CONF_PROGRAM_NAME)
+        category = service.data.get(CONF_CATEGORY)
+
+        video_url = video_url_by_video_id(
+            random_video_id(program_name, categories=category)
+        )
+
+        await hass.services.async_call('media_player', 'play_media', {
+            'entity_id': entity_id,
+            'media_content_id': video_url,
+            'media_content_type': 'video'
+        })
+    hass.services.async_register(
+        DOMAIN, SERVICE_PLAY_RANDOM, play_random, SERVICE_PLAY_RANDOM_SCHEMA
+    )
+
     async def play_channel(service):
         """Play the specified channel"""
 

custom_components/svt_play/video_url_fetch/live_test.py

@@ -1,4 +1,4 @@
-from video_fetch import video_url_by_channel, video_id_by_time, video_url_by_video_id, suggested_video_id
+from video_fetch import video_url_by_channel, video_id_by_time, video_url_by_video_id, suggested_video_id, random_video_id
 import pytest
 
 
@@ -32,3 +32,14 @@ def test_not_found_by_suggested():
 def test_not_found_channel():
     with pytest.raises(Exception, match="Could not fetch video url: Not found"):
         video_url_by_channel("svt1337")
+
+
+def test_random():
+    id = random_video_id('skavlan')
+    same_counter = 0
+    for _ in range(5):
+        new_id = random_video_id('skavlan')
+        print(new_id)
+        if id == new_id:
+            same_counter += 1
+    assert same_counter != 5

custom_components/svt_play/video_url_fetch/video_fetch.py

@@ -2,6 +2,7 @@
 from requests import post
 from datetime import datetime
 from datetime import timezone
+import random
 import iso8601
 
 default_formats = ['dashhbbtv', 'hls-cmaf', 'hls']
@@ -66,6 +67,19 @@ def video_id_by_time(program_id, index=0, categories=None):
     return videos[index]['item']['videoSvtId']
 
 
+def random_video_id(program_id, categories=None):
+    "Get a random video id from the specified program"
+    program_data = information_by_program_id(program_id)
+
+    videos = []
+    for content in program_data['associatedContent']:
+        if categories is None or content['name'] in categories:
+            videos += content['items']
+
+    index = random.randint(0, len(videos) - 1)
+    return videos[index]['item']['videoSvtId']
+
+
 def suggested_video_id(program_id):
     "Get the video id if the suggested video"
     program_data = information_by_program_id(program_id)