mirror of
https://github.com/fHDHR/fHDHR_NextPVR.git
synced 2025-12-06 10:26:57 -05:00
110 lines
4.8 KiB
Python
110 lines
4.8 KiB
Python
from flask import Response, request, redirect
|
|
import urllib.parse
|
|
import json
|
|
|
|
from fHDHR.tools import channel_sort
|
|
|
|
|
|
class W3U():
|
|
endpoints = ["/api/w3u"]
|
|
endpoint_name = "api_w3u"
|
|
endpoint_methods = ["GET", "POST"]
|
|
|
|
def __init__(self, fhdhr):
|
|
self.fhdhr = fhdhr
|
|
|
|
def __call__(self, *args):
|
|
return self.get(*args)
|
|
|
|
def get(self, *args):
|
|
|
|
base_url = request.url_root[:-1]
|
|
|
|
method = request.args.get('method', default="get", type=str)
|
|
channel = request.args.get('channel', default="all", type=str)
|
|
redirect_url = request.args.get('redirect', default=None, type=str)
|
|
|
|
if method == "get":
|
|
|
|
origin_methods = self.fhdhr.origins.valid_origins
|
|
origin = request.args.get('origin', default=None, type=str)
|
|
if origin and origin not in origin_methods:
|
|
return "%s Invalid channels origin" % origin
|
|
|
|
channel_info_m3u = {
|
|
"name": self.fhdhr.config.dict["fhdhr"]["friendlyname"],
|
|
"image": '%s/favicon.ico' % base_url,
|
|
"epg": '%s/api/xmltv' % base_url,
|
|
"stations": []
|
|
}
|
|
|
|
channel_items = []
|
|
|
|
if origin:
|
|
if channel == "all":
|
|
fileName = "channels.m3u"
|
|
for fhdhr_id in [x["id"] for x in self.fhdhr.device.channels.get_channels(origin)]:
|
|
channel_obj = self.fhdhr.device.channels.get_channel_obj("id", fhdhr_id, origin)
|
|
if channel_obj.enabled:
|
|
channel_items.append(channel_obj)
|
|
elif str(channel) in [str(x) for x in self.fhdhr.device.channels.get_channel_list("number", origin)]:
|
|
channel_obj = self.fhdhr.device.channels.get_channel_obj("number", channel, origin)
|
|
fileName = "%s.m3u" % channel_obj.number
|
|
if channel_obj.enabled:
|
|
channel_items.append(channel_obj)
|
|
else:
|
|
return "Channel Disabled"
|
|
elif channel != "all" and str(channel) in [str(x) for x in self.fhdhr.device.channels.get_channel_list("id", origin)]:
|
|
channel_obj = self.fhdhr.device.channels.get_channel_obj("id", channel, origin)
|
|
fileName = "%s.m3u" % channel_obj.number
|
|
if channel_obj.enabled:
|
|
channel_items.append(channel_obj)
|
|
else:
|
|
return "Channel Disabled"
|
|
elif not origin and channel != "all" and str(channel) in [str(x) for x in self.fhdhr.device.channels.get_channel_list("id")]:
|
|
channel_obj = self.fhdhr.device.channels.get_channel_obj("id", channel)
|
|
fileName = "%s.m3u" % channel_obj.number
|
|
if channel_obj.enabled:
|
|
channel_items.append(channel_obj)
|
|
else:
|
|
return "Channel Disabled"
|
|
else:
|
|
return "Invalid Channel"
|
|
|
|
channels_info = {}
|
|
|
|
for channel_obj in channel_items:
|
|
|
|
if self.fhdhr.config.dict["epg"]["images"] == "proxy" or not channel_obj.thumbnail:
|
|
logourl = ('%s/api/images?method=get&type=channel&id=%s' %
|
|
(base_url, str(channel_obj.dict['origin_id'])))
|
|
else:
|
|
logourl = channel_obj.thumbnail
|
|
|
|
channels_info[channel_obj.number] = {
|
|
"name": str(channel_obj.dict['name']),
|
|
"url": "%s%s" % (base_url, channel_obj.api_stream_url),
|
|
"epgId": str(channel_obj.dict['origin_id']),
|
|
"image": logourl,
|
|
}
|
|
|
|
# Sort the channels
|
|
sorted_channel_list = channel_sort(list(channels_info.keys()))
|
|
for channel in sorted_channel_list:
|
|
channel_info_m3u["stations"].append(channels_info[channel])
|
|
|
|
channels_info_json = json.dumps(channel_info_m3u, indent=4)
|
|
|
|
resp = Response(status=200, response=channels_info_json, mimetype='application/json')
|
|
resp.headers["content-disposition"] = "attachment; filename=%s" % fileName
|
|
return resp
|
|
|
|
return Response(status=200,
|
|
response=channels_info_json,
|
|
mimetype='application/json')
|
|
|
|
if redirect_url:
|
|
return redirect("%s?retmessage=%s" % (redirect_url, urllib.parse.quote("%s Success" % method)))
|
|
else:
|
|
return "%s Success" % method
|