ecb26968da
Hotstar is currently throwing errors when getting tracks, will be fixed with next commit. Fixed error when loading Widevine Device.
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import os
|
|
import re
|
|
from copy import copy
|
|
|
|
from vinetrimmer.services.BaseService import BaseService
|
|
|
|
SERVICE_MAP = {}
|
|
|
|
from vinetrimmer.services.amazon import Amazon
|
|
from vinetrimmer.services.appletvplus import AppleTVPlus
|
|
from vinetrimmer.services.max import Max
|
|
from vinetrimmer.services.netflix import Netflix
|
|
from vinetrimmer.services.peacock import Peacock
|
|
from vinetrimmer.services.hotstar import Hotstar
|
|
from vinetrimmer.services.jio import Jio
|
|
from vinetrimmer.services.moviesanywhere import MoviesAnywhere
|
|
from vinetrimmer.services.sonyliv import Sonyliv
|
|
|
|
|
|
# Above is necessary since dynamic imports like below fuck up nuitak
|
|
|
|
# Below dynamic imports fuck with compiling when using Nuitka - exec() call is the problem
|
|
#for service in os.listdir(os.path.dirname(__file__)):
|
|
# if service.startswith("_") or not service.endswith(".py"):
|
|
# continue
|
|
|
|
# service = os.path.splitext(service)[0]
|
|
|
|
# if service in ("__init__", "BaseService"):
|
|
# continue
|
|
|
|
# with open(os.path.join(os.path.dirname(__file__), f"{service}.py"), encoding="utf-8") as fd:
|
|
# code = ""
|
|
# for line in fd.readlines():
|
|
# if re.match(r"\s*(?:import(?! click)|from)\s", line):
|
|
# continue
|
|
# code += line
|
|
# if re.match(r"\s*super\(\)\.__init__\(", line):
|
|
# break
|
|
# exec(code)
|
|
|
|
for x in copy(globals()).values():
|
|
if isinstance(x, type) and issubclass(x, BaseService) and x != BaseService:
|
|
SERVICE_MAP[x.__name__] = x.ALIASES
|
|
|
|
|
|
def get_service_key(value):
|
|
"""
|
|
Get the Service Key name (e.g. DisneyPlus, not dsnp, disney+, etc.) from the SERVICE_MAP.
|
|
Input value can be of any case-sensitivity and can be either the key itself or an alias.
|
|
"""
|
|
value = value.lower()
|
|
for key, aliases in SERVICE_MAP.items():
|
|
if value in map(str.lower, aliases) or value == key.lower():
|
|
return key
|