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

Code format

Browse source
This commit is contained in:
Micke Nordin 2022-07-08 13:21:30 +02:00
parent 4f719ac781
commit 0528d20f1d
Signed by untrusted user who does not match committer: micke
GPG key ID: 0DA0A7A5708FE257
1 changed files with 33 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
src/Utils/__init__.py
View file

@ -68,44 +68,53 @@ def add_video(video_id: str,
con.commit()
def get_svt_thumb_from_id_changed(id: str, changed: str, size: str = "480") -> str:
return "https://www.svtstatic.se/image/custom/{}/{}/{}".format(size, id, changed)
def get_svt_thumb_from_id_changed(id: str,
changed: str,
size: str = "480") -> str:
return "https://www.svtstatic.se/image/custom/{}/{}/{}".format(
size, id, changed)
def get_all_svt_categories() -> list:
categories: list = list()
url = "https://www.svtplay.se/kategori"
data = get_svt_data(url)
data = get_svt_data(url)
for entry in data:
if 'genres' in entry.keys():
categories = entry['genres']
break
return categories
def get_svt_category(category: str) -> list:
url = 'https://www.svtplay.se/kategori/{}?tab=all'.format(category)
data = get_svt_data(url)
programs = list()
data = get_svt_data(url)
programs = list()
for entry in data:
if "categoryPage" in entry.keys():
for tab in entry["categoryPage"]["lazyLoadedTabs"]:
if "selections" in tab.keys():
for selection in tab["selections"]:
programs += selection['items']
break
if "categoryPage" in entry.keys():
for tab in entry["categoryPage"]["lazyLoadedTabs"]:
if "selections" in tab.keys():
for selection in tab["selections"]:
programs += selection['items']
break
return programs
def get_svt_data(url: str) -> list:
result:list = list()
result: list = list()
res = requests.get(url)
soup = BeautifulSoup(res.text, features="lxml")
data = json.loads(soup.find(id="__NEXT_DATA__").string)["props"]["urqlState"]
soup = BeautifulSoup(res.text, features="lxml")
data = json.loads(
soup.find(id="__NEXT_DATA__").string)["props"]["urqlState"]
for key in data.keys():
result.append(json.loads(data[key]["data"]))
result.append(json.loads(data[key]["data"]))
return result
def get_all_svt_programs() -> list:
url = 'https://www.svtplay.se/program'
data = get_svt_data(url)
data = get_svt_data(url)
programs = list()
for entry in data:
if "programAtillO" in entry.keys():
@ -352,7 +361,14 @@ def resolve_svt_channel(svt_id: str, path: str = '/usr/share/cast') -> dict:
}
for category in get_all_svt_categories():
channels[category['id']] = {"name": category["name"], "thumbnail": make_bitmap_from_url(get_svt_thumb_from_id_changed(category['image']['id'], category['image']['changed']))}
channels[category['id']] = {
"name":
category["name"],
"thumbnail":
make_bitmap_from_url(
get_svt_thumb_from_id_changed(category['image']['id'],
category['image']['changed']))
}
return channels[svt_id]