micke
/
cast
1
0
Fork 0
Code Issues 4 Pull Requests Projects Releases 3 Wiki Activity

Now lists all items from feed and caches links

Browse Source
main
Micke Nordin 2 years ago
parent 9d166d8b42
commit cc6c244769
Signed by: micke
GPG Key ID: 0DA0A7A5708FE257
4 changed files with 141 additions and 62 deletions
Show Stats Download Patch File Download Diff File

102
ChannelProvider/SVT/__init__.py
Unescape View File

@ -1,49 +1,97 @@
#!/usr/bin/env python3
import hashlib
import io
import json
import os
import pickle
import ChannelProvider
import feedparser
import requests
import wx
from bs4 import BeautifulSoup
from Items import Item
import json
class SVT(ChannelProvider.ChannelProvider):
m_cache: dict = dict()
m_cachefile = '/tmp/svt_cache'
def __init__(self) -> None:
super().__init__('SVT', 'http://www.svtplay.se/rss.xml')
if os.path.exists(self.m_cachefile):
print("Found cache file: {}".format(self.m_cachefile))
with open(self.m_cachefile, 'rb') as cachehandle:
self.m_cache = pickle.load(cachehandle)
self.m_items: list[Item] = self.parse_feed()
def refresh_items(self):
self.m_items: list[Item] = self.parse_feed()
def parse_feed(self) -> list[Item]:
feed = feedparser.parse(self.get_feed())
entries = feed['entries']
items: list[Item] = list()
for entry in entries:
thumbnail = ''
svt_id = ''
for link in entry['links']:
if str(link['type']).startswith('image/'):
thumbnail = str(link['href'])
page = requests.get(str(entry['link']))
soup = BeautifulSoup(page.text, 'html.parser')
for element in soup.find_all('a'):
href = element.get('href')
datart= element.get('data-rt')
if datart == 'top-area-play-button':
svt_id = href.split('=')[1].split('&')[0]
api = json.loads(requests.get('https://api.svt.se/video/{}'.format(svt_id)).text)
#print(api['videoReferences'])
resolved_link = ''
for reference in api['videoReferences']:
if reference['format'] == "dashhbbtv":
resolved_link = reference['url']
print(resolved_link)
item = Item(str(entry['description']), resolved_link,
self.m_provider_name, entry['published_parsed'],
thumbnail, str(entry['title']))
key = hashlib.sha256(entry['link'].encode('utf-8')).hexdigest()
if key in self.m_cache.keys():
print("Cache hit: {}".format(key))
thumbnail_link = self.m_cache[key]['thumbnail_link']
resolved_link = self.m_cache[key]['resolved_link']
description = self.m_cache[key]['description']
published_parsed = self.m_cache[key]['published_parsed']
title = self.m_cache[key]['title']
else:
svt_id = ''
for link in entry['links']:
if str(link['type']).startswith('image/'):
thumbnail_link = str(link['href'])
break
page = requests.get(str(entry['link']))
soup = BeautifulSoup(page.text, 'html.parser')
for element in soup.find_all('a'):
href = element.get('href')
datart = element.get('data-rt')
if datart == 'top-area-play-button':
svt_id = href.split('=')[1].split('&')[0]
api = json.loads(
requests.get(
'https://api.svt.se/video/{}'.format(svt_id)).text)
resolved_link = ''
for reference in api['videoReferences']:
if reference['format'] == "dashhbbtv":
resolved_link = reference['url']
print(resolved_link)
description = str(entry['description'])
published_parsed = entry['published_parsed']
title = str(entry['title'])
self.m_cache[key] = {'thumbnail_link': thumbnail_link}
self.m_cache[key]['resolved_link'] = resolved_link
self.m_cache[key]['description'] = description
self.m_cache[key]['published_parsed'] = published_parsed
self.m_cache[key]['title'] = title
res = requests.get(thumbnail_link)
content_bytes = io.BytesIO(res.content)
image = wx.Image(content_bytes,
type=wx.BITMAP_TYPE_ANY,
index=-1)
thumbnail = wx.Bitmap(image)
item = Item(description, resolved_link, self.m_provider_name,
published_parsed, thumbnail, title)
items.append(item)
# write to cache file
with open(self.m_cachefile, 'wb') as cachehandle:
pickle.dump(self.m_cache, cachehandle)
return items

2
ChannelProvider/__init__.py
Unescape View File

@ -9,7 +9,7 @@ class ChannelProvider:
def __init__(self, provider_name: str, feed: str) -> None:
self.m_provider_name = provider_name
self.m_feed = feed
self.m_items: Union[list[Item], None] = self.parse_feed()
self.m_items: Union[list[Item], None] = None
def get_feed(self) -> str:
return self.m_feed

4
Items/__init__.py
Unescape View File

@ -1,5 +1,5 @@
#!/usr/bin/env python3
import wx
from datetime import datetime
@ -9,7 +9,7 @@ class Item(dict):
link: str,
provider_id: str,
published: datetime,
thumbnail: str,
thumbnail: wx.Bitmap,
title: str,
watchtime: int = 0):
self.__dict__['description'] = description

95
main.py
Unescape View File

@ -1,8 +1,8 @@
#!/usr/bin/env python3
import hashlib
from typing import NewType
import wx
import wx.lib.scrolledpanel as scrolled
import wx.media
from ChannelProvider import SVT, ChannelProvider
@ -11,53 +11,84 @@ ChannelProvider = NewType('ChannelProvider', ChannelProvider)
class Cast(wx.Frame):
m_channels: list[ChannelProvider] = list()
def __init__(self, *args, **kw):
"""__init__.
:param args:
:param kw:
"""
super().__init__(*args, **kw)
self.m_channels.append(SVT.SVT())
self.m_index = 0
self.m_sizer: wx.Sizer = wx.BoxSizer(wx.VERTICAL)
self.m_panel: wx.lib.scrolledpanel.ScrolledPanel = scrolled.ScrolledPanel(
self, -1, style=wx.VSCROLL)
self.m_control = None
self.m_panel.SetupScrolling()
self.m_panel.SetSizer(self.m_sizer)
self.m_channels: list[ChannelProvider] = [SVT.SVT()]
self.show_list(None)
def show_list(self, _):
self.m_sizer.Clear(delete_windows=True)
self.m_sizer = wx.BoxSizer(wx.VERTICAL)
for channel in self.m_channels:
for item in channel.get_items():
title = wx.StaticText(self.m_panel, -1, item['title'])
description = wx.StaticText(self.m_panel, -1,
item['description'])
bitmap = item['thumbnail']
btn = wx.BitmapButton(self.m_panel, id=self.m_index, bitmap=bitmap)
btn.Bind(wx.EVT_BUTTON,
lambda event, link=item['link']: self.show_player(
event, link))
self.m_sizer.Add(title)
self.m_sizer.Add(btn)
self.m_sizer.Add(description)
self.m_index = self.m_index + 1
self.m_panel.SetSizer(self.m_sizer)
self.m_panel.Layout()
def show_player(self, _, uri):
self.m_sizer.Clear(delete_windows=True)
self.m_sizer = wx.GridBagSizer()
self.m_control = wx.media.MediaCtrl(
self,
self.m_panel,
size=(480, 480),
style=wx.SIMPLE_BORDER,
szBackend=wx.media.MEDIABACKEND_GSTREAMER)
play_button = wx.Button(self, -1, "Play")
self.Bind(wx.EVT_BUTTON, self.play, play_button)
pause_button = wx.Button(self, -1, "Pause")
self.Bind(wx.EVT_BUTTON, self.pause, pause_button)
stop_button = wx.Button(self, -1, "Stop")
self.Bind(wx.EVT_BUTTON, self.quit, stop_button)
sizer = wx.GridBagSizer()
sizer.Add(self.m_control, (0, 0))
sizer.SetItemSpan(0, (0, 6))
sizer.Add(play_button, (1, 1))
sizer.Add(pause_button, (1, 2))
sizer.Add(stop_button, (1, 3))
self.SetSizer(sizer)
self.m_control.Bind(wx.media.EVT_MEDIA_LOADED, self.play)
self.m_control.Bind(wx.media.EVT_MEDIA_FINISHED, self.quit)
self.load_video(self.m_channels[0].get_items()[0]['link'])
self.Show(True)
sizer.Layout()
def load_video(self, uri):
play_button = wx.Button(self.m_panel, -1, "Play")
play_button.Bind(wx.EVT_BUTTON, self.play)
pause_button = wx.Button(self.m_panel, -1, "Pause")
pause_button.Bind(wx.EVT_BUTTON, self.pause)
back_button = wx.Button(self.m_panel, -1, "Back")
back_button.Bind(wx.EVT_BUTTON, self.show_list)
self.m_sizer.Add(self.m_control, (0, 0))
self.m_sizer.SetItemSpan(0, (0, 6))
self.m_sizer.Add(play_button, (1, 1))
self.m_sizer.Add(pause_button, (1, 2))
self.m_sizer.Add(back_button, (1, 3))
self.Bind(wx.media.EVT_MEDIA_LOADED, self.play)
self.Bind(wx.media.EVT_MEDIA_FINISHED, self.show_list)
self.load_uri(uri)
self.m_panel.SetSizer(self.m_sizer)
self.m_panel.Layout()
self.m_panel.ScrollChildIntoView(self.m_control)
def load_uri(self, uri):
self.m_control.LoadURI(uri)
def play(self, event):
def play(self, _):
self.m_control.Play()
def pause(self, event):
def pause(self, _):
self.m_control.Pause()
def quit(self, event):
def quit(self, _):
self.Destroy()

Write Preview
Loading…
Cancel
Save