|
|
|
@ -1,6 +1,8 @@
|
|
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
import sys
|
|
|
|
|
import threading
|
|
|
|
|
import time
|
|
|
|
|
import os
|
|
|
|
|
from typing import Callable
|
|
|
|
|
|
|
|
|
|
import pychromecast
|
|
|
|
@ -11,7 +13,8 @@ from vlc import Instance
|
|
|
|
|
|
|
|
|
|
from Channel import SVT
|
|
|
|
|
from ChannelProvider import ChannelProvider
|
|
|
|
|
from Utils import MYPATH, make_bitmap_from_file, make_sized_button
|
|
|
|
|
from Utils import (get_all_svt_categories, make_bitmap_from_file,
|
|
|
|
|
make_sized_button, BASEPATH)
|
|
|
|
|
|
|
|
|
|
WIDTH = int(720 / 2)
|
|
|
|
|
HEIGHT = int(1440 / 2)
|
|
|
|
@ -27,12 +30,23 @@ FLAGS.Center()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Cast(wx.Frame):
|
|
|
|
|
|
|
|
|
|
def __init__(self, *args, **kw):
|
|
|
|
|
"""__init__.
|
|
|
|
|
:param args:
|
|
|
|
|
:param kw:
|
|
|
|
|
"""
|
|
|
|
|
self.m_mode = 'normal'
|
|
|
|
|
self.asset_path: str = '/usr/share/cast'
|
|
|
|
|
url = None
|
|
|
|
|
if "url" in kw:
|
|
|
|
|
self.m_mode = 'single'
|
|
|
|
|
url = kw['url']
|
|
|
|
|
del kw['url']
|
|
|
|
|
super().__init__(*args, **kw)
|
|
|
|
|
if not os.path.isdir(BASEPATH):
|
|
|
|
|
os.mkdir(BASEPATH)
|
|
|
|
|
|
|
|
|
|
self.m_selected_chromecast = None
|
|
|
|
|
self.SetSizeHints(WIDTH, HEIGHT, maxW=WIDTH)
|
|
|
|
|
self.m_style = self.GetWindowStyle()
|
|
|
|
@ -40,7 +54,6 @@ class Cast(wx.Frame):
|
|
|
|
|
args=(),
|
|
|
|
|
kwargs={})
|
|
|
|
|
self.m_vlc = Instance()
|
|
|
|
|
self.m_vlc_medialist = self.m_vlc.media_list_new()
|
|
|
|
|
self.m_vlc_listplayer = self.m_vlc.media_list_player_new()
|
|
|
|
|
self.m_chromecast_thr.start()
|
|
|
|
|
self.m_sizer: wx.Sizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
@ -49,7 +62,12 @@ class Cast(wx.Frame):
|
|
|
|
|
self.m_panel.SetupScrolling(rate_y=SCROLL_RATE, scrollToTop=True)
|
|
|
|
|
self.m_panel.SetSizer(self.m_sizer)
|
|
|
|
|
self.m_providers: list[ChannelProvider] = self.get_providers()
|
|
|
|
|
self.show_channel_list(None, 0)
|
|
|
|
|
if url:
|
|
|
|
|
self.show_player(None, url, 0)
|
|
|
|
|
elif len(self.m_providers) > 1:
|
|
|
|
|
self.show_provider_list(None)
|
|
|
|
|
else:
|
|
|
|
|
self.show_channel_list(None, 0)
|
|
|
|
|
|
|
|
|
|
def add_back_button(self, callback: Callable) -> None:
|
|
|
|
|
backbtn = wx.Button(self.m_panel,
|
|
|
|
@ -68,6 +86,7 @@ class Cast(wx.Frame):
|
|
|
|
|
self.m_chromecasts, self.m_browser = pychromecast.get_chromecasts()
|
|
|
|
|
|
|
|
|
|
def get_player_controls(self, channel_index: int, uri: str) -> wx.BoxSizer:
|
|
|
|
|
outer_sizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
|
inner_sizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
|
|
|
play_button = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
@ -79,28 +98,34 @@ class Cast(wx.Frame):
|
|
|
|
|
"Pause",
|
|
|
|
|
size=(WIDTH / 4, BTN_HEIGHT))
|
|
|
|
|
|
|
|
|
|
back_button = wx.Button(self.m_panel,
|
|
|
|
|
stop_button = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
"Back",
|
|
|
|
|
"Stop",
|
|
|
|
|
size=(WIDTH / 4, BTN_HEIGHT))
|
|
|
|
|
back_button.Bind(
|
|
|
|
|
wx.EVT_BUTTON,
|
|
|
|
|
lambda event, cindex=channel_index: self.show_video_list(
|
|
|
|
|
event, cindex),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
stop_button = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
"Stop",
|
|
|
|
|
size=(WIDTH / 4, BTN_HEIGHT))
|
|
|
|
|
if self.m_mode == 'normal':
|
|
|
|
|
back_button = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
"Go Back",
|
|
|
|
|
size=(WIDTH, BTN_HEIGHT))
|
|
|
|
|
back_button.Bind(
|
|
|
|
|
wx.EVT_BUTTON,
|
|
|
|
|
lambda event, cindex=channel_index: self.show_video_list(
|
|
|
|
|
event, cindex),
|
|
|
|
|
)
|
|
|
|
|
else:
|
|
|
|
|
back_button = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
label="Close",
|
|
|
|
|
size=(WIDTH, BTN_HEIGHT))
|
|
|
|
|
back_button.Bind(wx.EVT_BUTTON, lambda event: self.Destroy())
|
|
|
|
|
outer_sizer.Add(back_button, FLAGS)
|
|
|
|
|
inner_sizer.Add(play_button, FLAGS)
|
|
|
|
|
inner_sizer.Add(pause_button, FLAGS)
|
|
|
|
|
inner_sizer.Add(back_button, FLAGS)
|
|
|
|
|
inner_sizer.Add(stop_button, FLAGS)
|
|
|
|
|
|
|
|
|
|
if not self.m_chromecast_thr.is_alive(
|
|
|
|
|
) and not self.m_selected_chromecast:
|
|
|
|
|
btm = make_bitmap_from_file('{}/assets/Cast.png'.format(MYPATH), wx.Size(24,24))
|
|
|
|
|
if self.has_usable_chromecasts():
|
|
|
|
|
btm = make_bitmap_from_file(
|
|
|
|
|
'{}/assets/Cast.png'.format(self.asset_path), wx.Size(24, 24))
|
|
|
|
|
cast_button = wx.BitmapButton(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
bitmap=btm,
|
|
|
|
@ -112,7 +137,6 @@ class Cast(wx.Frame):
|
|
|
|
|
)
|
|
|
|
|
inner_sizer.Add(cast_button, FLAGS)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if self.m_selected_chromecast:
|
|
|
|
|
self.cast(wx.media.EVT_MEDIA_LOADED, uri),
|
|
|
|
|
play_button.Bind(wx.EVT_BUTTON, self.play_cast)
|
|
|
|
@ -128,42 +152,75 @@ class Cast(wx.Frame):
|
|
|
|
|
play_button.Bind(wx.EVT_BUTTON, self.play)
|
|
|
|
|
pause_button.Bind(wx.EVT_BUTTON, self.pause)
|
|
|
|
|
stop_button.Bind(wx.EVT_BUTTON, self.stop)
|
|
|
|
|
outer_sizer.Add(inner_sizer, FLAGS)
|
|
|
|
|
inner_sizer.Fit(self)
|
|
|
|
|
inner_sizer.Layout()
|
|
|
|
|
outer_sizer.Fit(self)
|
|
|
|
|
outer_sizer.Layout()
|
|
|
|
|
|
|
|
|
|
return inner_sizer
|
|
|
|
|
return outer_sizer
|
|
|
|
|
|
|
|
|
|
def get_providers(self) -> list[ChannelProvider]:
|
|
|
|
|
|
|
|
|
|
providers = list()
|
|
|
|
|
svt = ChannelProvider(
|
|
|
|
|
"SVT",
|
|
|
|
|
channels=[
|
|
|
|
|
SVT.SVT("feed"),
|
|
|
|
|
SVT.SVT("ch-svt1"),
|
|
|
|
|
SVT.SVT("ch-svt2"),
|
|
|
|
|
SVT.SVT("ch-svt24"),
|
|
|
|
|
SVT.SVT("ch-barnkanalen"),
|
|
|
|
|
SVT.SVT("ch-kunskapskanalen"),
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
providers.append(svt)
|
|
|
|
|
chandict = {
|
|
|
|
|
"kanaler": {
|
|
|
|
|
"channels": [
|
|
|
|
|
"ch-svt1", "ch-svt2", "ch-svt24", "ch-barnkanalen",
|
|
|
|
|
"ch-kunskapskanalen"
|
|
|
|
|
],
|
|
|
|
|
"displayname":
|
|
|
|
|
"SVT Channel Streams"
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
chandict["program"] = {
|
|
|
|
|
"channels": ["feed", "allprograms"],
|
|
|
|
|
"displayname": "SVT Shows - Latest and A-Ö"
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
categories = list()
|
|
|
|
|
for category in get_all_svt_categories():
|
|
|
|
|
categories.append(category['id'])
|
|
|
|
|
chandict["kategorier"] = {
|
|
|
|
|
"channels": categories,
|
|
|
|
|
"displayname": "SVT All Categories"
|
|
|
|
|
}
|
|
|
|
|
for provider in ["kanaler", "program", "kategorier"]:
|
|
|
|
|
channels = list()
|
|
|
|
|
for id in chandict[provider]["channels"]:
|
|
|
|
|
channels.append(SVT.SVT(id))
|
|
|
|
|
svt = ChannelProvider(chandict[provider]["displayname"],
|
|
|
|
|
channels=channels)
|
|
|
|
|
providers.append(svt)
|
|
|
|
|
|
|
|
|
|
return providers
|
|
|
|
|
|
|
|
|
|
def has_usable_chromecasts(self) -> bool:
|
|
|
|
|
if self.m_chromecast_thr.is_alive():
|
|
|
|
|
return False
|
|
|
|
|
if self.m_selected_chromecast:
|
|
|
|
|
return False
|
|
|
|
|
result = False
|
|
|
|
|
for cast in self.m_chromecasts:
|
|
|
|
|
if cast.cast_type != 'audio':
|
|
|
|
|
result = True
|
|
|
|
|
break
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
|
|
def show_channel_list(self, _, provider_index) -> None:
|
|
|
|
|
self.m_selected_provider_index = provider_index
|
|
|
|
|
self.m_selected_provider = self.m_providers[provider_index]
|
|
|
|
|
self.m_sizer.Clear(delete_windows=True)
|
|
|
|
|
self.m_sizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
|
closebtn = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
label="Close",
|
|
|
|
|
size=(WIDTH, BTN_HEIGHT))
|
|
|
|
|
closebtn.Bind(wx.EVT_BUTTON, lambda event: self.Destroy())
|
|
|
|
|
self.m_sizer.Add(closebtn, 0, wx.ALL, 1)
|
|
|
|
|
if len(self.m_providers) > 1:
|
|
|
|
|
back_callback = lambda event: self.show_provider_list(event)
|
|
|
|
|
self.add_back_button(back_callback)
|
|
|
|
|
else:
|
|
|
|
|
closebtn = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
label="Close",
|
|
|
|
|
size=(WIDTH, BTN_HEIGHT))
|
|
|
|
|
closebtn.Bind(wx.EVT_BUTTON, lambda event: self.Destroy())
|
|
|
|
|
self.m_sizer.Add(closebtn, 0, wx.ALL, 1)
|
|
|
|
|
|
|
|
|
|
channel_index = 0
|
|
|
|
|
|
|
|
|
@ -182,6 +239,30 @@ class Cast(wx.Frame):
|
|
|
|
|
self.m_sizer.Fit(self)
|
|
|
|
|
self.m_sizer.Layout()
|
|
|
|
|
|
|
|
|
|
def show_provider_list(self, _) -> None:
|
|
|
|
|
self.m_sizer.Clear(delete_windows=True)
|
|
|
|
|
self.m_sizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
|
closebtn = wx.Button(self.m_panel,
|
|
|
|
|
-1,
|
|
|
|
|
label="Close",
|
|
|
|
|
size=(WIDTH, BTN_HEIGHT))
|
|
|
|
|
closebtn.Bind(wx.EVT_BUTTON, lambda event: self.Destroy())
|
|
|
|
|
self.m_sizer.Add(closebtn, 0, wx.ALL, 1)
|
|
|
|
|
provider_index = 0
|
|
|
|
|
for provider in self.m_providers:
|
|
|
|
|
bitmap = provider.get_logo_as_bitmap()
|
|
|
|
|
callback = lambda event, pindex=provider_index: self.show_channel_list(
|
|
|
|
|
event, pindex)
|
|
|
|
|
btn_sizer: wx.BoxSizer = make_sized_button(self.m_panel, bitmap,
|
|
|
|
|
provider.get_name(),
|
|
|
|
|
callback)
|
|
|
|
|
self.m_sizer.Add(btn_sizer, 0, wx.ALL, 1)
|
|
|
|
|
provider_index += 1
|
|
|
|
|
self.m_panel.SetupScrolling(rate_y=SCROLL_RATE, scrollToTop=True)
|
|
|
|
|
self.m_panel.SetSizer(self.m_sizer)
|
|
|
|
|
self.m_sizer.Fit(self)
|
|
|
|
|
self.m_sizer.Layout()
|
|
|
|
|
|
|
|
|
|
def show_video_list(self, _, channel_index) -> None:
|
|
|
|
|
self.SetWindowStyle(self.m_style)
|
|
|
|
|
self.m_selected_channel = self.m_selected_provider.get_channel_by_index(
|
|
|
|
@ -191,7 +272,11 @@ class Cast(wx.Frame):
|
|
|
|
|
back_callback = lambda event: self.show_channel_list(
|
|
|
|
|
event, self.m_selected_provider_index)
|
|
|
|
|
self.add_back_button(back_callback)
|
|
|
|
|
if self.m_selected_provider.get_name() == 'SVT' and self.m_selected_channel.get_id() == 'feed':
|
|
|
|
|
if self.m_selected_provider.get_name().startswith(
|
|
|
|
|
'SVT') and self.m_selected_channel.get_id() not in [
|
|
|
|
|
"ch-svt1", "ch-svt2", "ch-svt24", "ch-barnkanalen",
|
|
|
|
|
"ch-kunskapskanalen"
|
|
|
|
|
]:
|
|
|
|
|
|
|
|
|
|
def refresh_callback(event):
|
|
|
|
|
self.m_selected_channel.refresh()
|
|
|
|
@ -206,9 +291,13 @@ class Cast(wx.Frame):
|
|
|
|
|
|
|
|
|
|
if self.m_selected_channel.wait():
|
|
|
|
|
with wx.BusyInfo("Please wait, working..."):
|
|
|
|
|
number_of_waits = 0
|
|
|
|
|
while self.m_selected_channel.wait():
|
|
|
|
|
number_of_waits += 1
|
|
|
|
|
time.sleep(1)
|
|
|
|
|
wx.GetApp().Yield()
|
|
|
|
|
if number_of_waits > 10:
|
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
btnindex = 0
|
|
|
|
|
for item in self.m_selected_channel.get_items(): # type: ignore
|
|
|
|
@ -271,8 +360,10 @@ class Cast(wx.Frame):
|
|
|
|
|
:param uri str: the link to the video stream
|
|
|
|
|
"""
|
|
|
|
|
media = self.m_vlc.media_new(uri)
|
|
|
|
|
self.m_vlc_medialist.add_media(media)
|
|
|
|
|
self.m_vlc_listplayer.set_media_list(self.m_vlc_medialist)
|
|
|
|
|
medialist = self.m_vlc.media_list_new()
|
|
|
|
|
|
|
|
|
|
medialist.add_media(media)
|
|
|
|
|
self.m_vlc_listplayer.set_media_list(medialist)
|
|
|
|
|
|
|
|
|
|
self.m_sizer.Clear(delete_windows=True)
|
|
|
|
|
self.m_sizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
@ -298,7 +389,13 @@ class Cast(wx.Frame):
|
|
|
|
|
self.m_sizer.Add(cancel_btn)
|
|
|
|
|
|
|
|
|
|
for cast in self.m_chromecasts:
|
|
|
|
|
friendly_name = cast.device.friendly_name
|
|
|
|
|
if cast.cast_type == 'audio':
|
|
|
|
|
continue
|
|
|
|
|
friendly_name = "Unknown Chromecast"
|
|
|
|
|
try:
|
|
|
|
|
friendly_name = cast.device.friendly_name
|
|
|
|
|
except AttributeError:
|
|
|
|
|
friendly_name = cast.cast_info.friendly_name
|
|
|
|
|
btn = wx.Button(self.m_panel,
|
|
|
|
|
id=-1,
|
|
|
|
|
label=friendly_name,
|
|
|
|
@ -388,7 +485,12 @@ class Cast(wx.Frame):
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
# When this module is run (not imported) then create the app, the
|
|
|
|
|
# frame, show it, and start the event loop.
|
|
|
|
|
url = None
|
|
|
|
|
app: wx.App = wx.App()
|
|
|
|
|
frm: Cast = Cast(None, title="Cast")
|
|
|
|
|
if len(sys.argv) > 1:
|
|
|
|
|
url = sys.argv[1]
|
|
|
|
|
frm: Cast = Cast(None, title="Cast", url=url)
|
|
|
|
|
else:
|
|
|
|
|
frm: Cast = Cast(None, title="Cast")
|
|
|
|
|
frm.Show()
|
|
|
|
|
app.MainLoop()
|
|
|
|
|