micke/cast
1
0
Fork 0
Code Issues 4 Pull requests Projects Releases 3 Packages Wiki Activity Actions

Initial support for all programs A-Ö

Browse source 
This feature is not yet really usable since it takes a really long
time to load all programs sequentially and synchronously.
This commit is contained in:
Micke Nordin 2022-07-08 11:04:01 +02:00
parent 564f64106a
commit 30544ac026
Signed by untrusted user who does not match committer: micke
GPG key ID: 0DA0A7A5708FE257
5 changed files with 102 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
scripts/cast
View file

@ -162,6 +162,7 @@ class Cast(wx.Frame):
SVT.SVT("ch-svt24"),
SVT.SVT("ch-barnkanalen"),
SVT.SVT("ch-kunskapskanalen"),
SVT.SVT("allprograms"),
],
)

60
src/Channel/SVT/__init__.py
View file

@ -6,17 +6,17 @@ from datetime import datetime
import feedparser
import requests
import wx
from bs4 import BeautifulSoup
from Channel import Channel
from Items import Item
from Utils import (add_video, hash_string, make_bitmap_from_url,
from Utils import (add_video, get_all_svt_programs, get_svt_id, get_svt_thumbnail, hash_string, make_bitmap_from_url,
resolve_svt_channel, video_exists)
default_rss_url = 'http://www.svtplay.se/rss.xml'
class SVT(Channel):
def __init__(self, channel_id: str) -> None:
chan_dict = resolve_svt_channel(channel_id)
logo = chan_dict['thumbnail']
@ -24,7 +24,6 @@ class SVT(Channel):
super().__init__(channel_id, 'SVT', default_rss_url, logo, name)
def parse_feed(self) -> None:
#self.m_items: list[Item] = list()
resolved_link = str()
description = str()
title = str()
@ -33,7 +32,7 @@ class SVT(Channel):
published_parsed: datetime = datetime.now()
video_id = str()
if self.m_id == 'feed' :
if self.m_id == 'feed':
feed = feedparser.parse(self.get_feed())
entries = feed['entries']
for entry in entries:
@ -46,16 +45,7 @@ class SVT(Channel):
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]
svt_id = get_svt_id(str(entry['link']))
resolved_link = self.resolve_link(svt_id)
description = str(entry['description'])
published_parsed = entry['published_parsed']
@ -67,9 +57,40 @@ class SVT(Channel):
item = Item(description, resolved_link, self.m_provider_name,
published_parsed, thumbnail, title)
self.m_items.append(item)
add_video(video_id, self.m_id, self.m_provider_name, description,
resolved_link, published_parsed, thumbnail, title, 0)
add_video(video_id, self.m_id, self.m_provider_name,
description, resolved_link, published_parsed,
thumbnail, title, 0)
elif self.m_id == 'allprograms':
entries = get_all_svt_programs()
for entry in entries:
url = entry['item']['urls']['svtplay']
video_id = hash_string(url.split('/')[1])
svt_id = str()
link = "https://svtplay.se{}".format(url)
if video_exists(video_id, 'allprograms'):
continue
thumbnail_link = get_svt_thumbnail(link)
svt_id = get_svt_id(link)
resolved_link = self.resolve_link(svt_id)
title = str(entry['heading'])
type = str(entry['item']['__typename'])
published_parsed = datetime.now()
swe_only = "no"
if entry['item']['restrictions']['onlyAvailableInSweden']:
swe_only = 'yes'
description = "{}\nOnly available in Sweden:{}\nType of program:{}".format(title, swe_only, type)
if not resolved_link:
continue
thumbnail = make_bitmap_from_url(
thumbnail_link, wx.Size(int(self.m_screen_width), 150))
item = Item(description, resolved_link, self.m_provider_name,
published_parsed, thumbnail, title, sort_key = entry['selection_name'])
self.m_items.append(item)
add_video(video_id, self.m_id, self.m_provider_name,
description, resolved_link, published_parsed,
thumbnail, title, 0)
else:
chan_dict = resolve_svt_channel(self.m_id)
resolved_link = self.resolve_link(self.m_id)
@ -83,12 +104,10 @@ class SVT(Channel):
published_parsed, thumbnail, title)
self.m_items.append(item)
def resolve_link(self,svt_id) -> str:
def resolve_link(self, svt_id) -> str:
url = 'https://api.svt.se/video/{}'.format(svt_id)
print(url)
api = json.loads(
requests.get(url).text)
api = json.loads(requests.get(url).text)
resolved_link = ''
try:
@ -98,4 +117,3 @@ class SVT(Channel):
except KeyError:
pass
return resolved_link

1
src/ChannelProvider/__init__.py
View file

@ -1,6 +1,5 @@
#/usr/bin/env python3
import threading
import time
from os import path
import wx

6
src/Items/__init__.py
View file

@ -1,5 +1,6 @@
#!/usr/bin/env python3
from datetime import datetime
from typing import Union
import wx
@ -12,7 +13,9 @@ class Item(dict):
published: datetime,
thumbnail: wx.Bitmap,
title: str,
watchtime: int = 0):
watchtime: int = 0,
sort_key: Union[str, None] = None,
):
self.__dict__['description'] = description
self.__dict__['link'] = link
self.__dict__['provider_id'] = provider_id
@ -20,6 +23,7 @@ class Item(dict):
self.__dict__['thumbnail'] = thumbnail
self.__dict__['title'] = title
self.__dict__['watchtime'] = watchtime
self.__dict__['sort_key'] = sort_key
def __setitem__(self, key, item):
self.__dict__[key] = item

67
src/Utils/__init__.py
View file

@ -1,6 +1,7 @@
#!/usr/bin/env python3
import hashlib
import io
import json
import sqlite3
import time
from datetime import datetime
@ -9,12 +10,13 @@ from typing import Callable, Union
import requests
import wx
from bs4 import BeautifulSoup
from Items import Item
HEIGHT = int(1440 / 2)
BTN_HEIGHT = 40
SIZE = wx.Size(68,100)
SIZE = wx.Size(68, 100)
SCREEN_WIDTH = int(720 / 2)
BASEPATH = path.join(str(environ.get("HOME")), '.config/cast')
DB_FILE_NAME = 'cast.db'
@ -67,7 +69,27 @@ def add_video(video_id: str,
con.commit()
def get_default_logo(providerid: str = 'default', path: str = '/usr/share/cast') -> wx.Bitmap:
def get_all_svt_programs() -> list:
url = 'https://www.svtplay.se/program'
res = requests.get(url)
soup = BeautifulSoup(res.text, features="lxml")
data = json.loads(
soup.find(id="__NEXT_DATA__").string)["props"]["urqlState"]
programs = list()
for key in data.keys():
temp = json.loads(data[key]["data"])
if "programAtillO" in temp.keys():
for selection in temp["programAtillO"]["selections"]:
for item in selection["items"]:
item['selection_name'] = selection['name']
programs.append(item)
return programs
def get_default_logo(providerid: str = 'default',
path: str = '/usr/share/cast') -> wx.Bitmap:
if providerid == 'SVT':
return wx.Bitmap('{}/assets/SVT.png'.format(path))
else:
@ -140,6 +162,29 @@ def get_subscriptions(basepath: str = BASEPATH,
return subscriptions
def get_svt_id(link: str) -> str:
svt_id = str()
page = requests.get(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]
return svt_id
def get_svt_thumbnail(link: str) -> str:
page = requests.get(link)
soup = BeautifulSoup(page.text, 'html.parser')
meta = soup.find(property="og:image")
image_link = meta["content"]
return image_link
def get_videos(channel_id: str,
basepath: str = BASEPATH,
filename: str = DB_FILE_NAME) -> list[Item]:
@ -181,7 +226,7 @@ def hash_string(string: str) -> str:
def make_sized_button(parent_pnl: wx.Panel, bitmap_or_str: Union[wx.Bitmap,
str],
text: str, callback: Callable) -> wx.BoxSizer:
btn_sizer = wx.StaticBoxSizer(wx.HORIZONTAL,parent_pnl)
btn_sizer = wx.StaticBoxSizer(wx.HORIZONTAL, parent_pnl)
if type(bitmap_or_str) == type(str):
if bitmap_or_str.startswith('http'): # type: ignore
bitmap = make_bitmap_from_url(bitmap_or_str) # type: ignore
@ -194,7 +239,7 @@ def make_sized_button(parent_pnl: wx.Panel, bitmap_or_str: Union[wx.Bitmap,
wx.ID_ANY,
bitmap,
style=btn_style,
size=wx.Size(100,68))
size=wx.Size(100, 68))
btn_logo.SetToolTip(text)
btn_sizer.Add(btn_logo, 0, wx.EXPAND, 1)
@ -202,8 +247,7 @@ def make_sized_button(parent_pnl: wx.Panel, bitmap_or_str: Union[wx.Bitmap,
wx.ID_ANY,
text,
style=wx.BORDER_NONE | wx.BU_AUTODRAW,
size=wx.Size(SCREEN_WIDTH - 100,
SIZE.GetHeight()))
size=wx.Size(SCREEN_WIDTH - 100, SIZE.GetHeight()))
btn_text.SetToolTip(text)
btn_sizer.Add(btn_text, 0, wx.EXPAND, 1)
parent_pnl.Bind(wx.EVT_BUTTON, callback, btn_logo)
@ -255,8 +299,7 @@ def resolve_svt_channel(svt_id: str, path: str = '/usr/share/cast') -> dict:
make_bitmap_from_file('{}/assets/SVT2.png'.format(path))
},
"ch-svt24": {
"name":
"SVT 24",
"name": "SVT 24",
"thumbnail":
make_bitmap_from_file('{}/assets/SVT24.png'.format(path))
},
@ -264,14 +307,18 @@ def resolve_svt_channel(svt_id: str, path: str = '/usr/share/cast') -> dict:
"name":
"Kunskapskanalen",
"thumbnail":
make_bitmap_from_file(
'{}/assets/Kunskapskanalen.png'.format(path))
make_bitmap_from_file('{}/assets/Kunskapskanalen.png'.format(path))
},
"feed": {
"name": "Senaste program",
"thumbnail":
make_bitmap_from_file('{}/assets/SVT.png'.format(path))
},
"allprograms": {
"name": "Alla program",
"thumbnail":
make_bitmap_from_file('{}/assets/SVT.png'.format(path))
},
}
return channels[svt_id]