133 lines
5.8 KiB
Python
133 lines
5.8 KiB
Python
from flask import Response, request, redirect
|
|
import urllib.parse
|
|
import json
|
|
import datetime
|
|
|
|
from fHDHR.tools import humanized_time, channel_sort
|
|
|
|
|
|
class EPG():
|
|
"""Methods to create xmltv.xml"""
|
|
endpoints = ["/api/epg"]
|
|
endpoint_name = "api_epg"
|
|
endpoint_methods = ["GET", "POST"]
|
|
|
|
def __init__(self, fhdhr):
|
|
self.fhdhr = fhdhr
|
|
|
|
def __call__(self, *args):
|
|
return self.get(*args)
|
|
|
|
def get(self, *args):
|
|
|
|
method = request.args.get('method', default="get", type=str)
|
|
|
|
source = request.args.get('source', default=self.fhdhr.config.dict["epg"]["def_method"], type=str)
|
|
if source not in self.fhdhr.config.dict["epg"]["valid_methods"]:
|
|
return "%s Invalid xmltv method" % source
|
|
|
|
redirect_url = request.args.get('redirect', default=None, type=str)
|
|
|
|
if method == "get":
|
|
|
|
epgdict = self.fhdhr.device.epg.get_epg(source)
|
|
if source in ["blocks", "origin", self.fhdhr.config.dict["main"]["dictpopname"]]:
|
|
epgdict = epgdict.copy()
|
|
for c in list(epgdict.keys()):
|
|
chan_obj = self.fhdhr.device.channels.get_channel_obj("origin_id", epgdict[c]["id"])
|
|
epgdict[chan_obj.number] = epgdict.pop(c)
|
|
epgdict[chan_obj.number]["name"] = chan_obj.dict["name"]
|
|
epgdict[chan_obj.number]["callsign"] = chan_obj.dict["callsign"]
|
|
epgdict[chan_obj.number]["number"] = chan_obj.number
|
|
epgdict[chan_obj.number]["id"] = chan_obj.dict["origin_id"]
|
|
epgdict[chan_obj.number]["thumbnail"] = chan_obj.thumbnail
|
|
|
|
# Sort the channels
|
|
sorted_channel_list = channel_sort(list(epgdict.keys()))
|
|
sorted_chan_guide = {}
|
|
for channel in sorted_channel_list:
|
|
sorted_chan_guide[channel] = epgdict[channel]
|
|
|
|
epg_json = json.dumps(sorted_chan_guide, indent=4)
|
|
|
|
return Response(status=200,
|
|
response=epg_json,
|
|
mimetype='application/json')
|
|
|
|
elif method == "current":
|
|
|
|
nowtime = datetime.datetime.utcnow().timestamp()
|
|
|
|
chan_guide_list = []
|
|
|
|
whatson = self.fhdhr.device.epg.whats_on_allchans(source)
|
|
|
|
# Sort the channels
|
|
sorted_channel_list = channel_sort(list(whatson.keys()))
|
|
sorted_chan_guide = {}
|
|
for channel in sorted_channel_list:
|
|
sorted_chan_guide[channel] = whatson[channel]
|
|
|
|
for channel in list(sorted_chan_guide.keys()):
|
|
if sorted_chan_guide[channel]["listing"][0]["time_end"]:
|
|
remaining_time = humanized_time(sorted_chan_guide[channel]["listing"][0]["time_end"] - nowtime)
|
|
else:
|
|
remaining_time = "N/A"
|
|
|
|
chan_dict = {
|
|
"name": sorted_chan_guide[channel]["name"],
|
|
"number": sorted_chan_guide[channel]["number"],
|
|
"chan_thumbnail": sorted_chan_guide[channel]["thumbnail"],
|
|
"listing_title": sorted_chan_guide[channel]["listing"][0]["title"],
|
|
"listing_thumbnail": sorted_chan_guide[channel]["listing"][0]["thumbnail"],
|
|
"listing_description": sorted_chan_guide[channel]["listing"][0]["description"],
|
|
"listing_remaining_time": str(remaining_time)
|
|
}
|
|
|
|
for time_item in ["time_start", "time_end"]:
|
|
|
|
if not sorted_chan_guide[channel]["listing"][0][time_item]:
|
|
chan_dict["listing_%s" % time_item] = "N/A"
|
|
elif str(sorted_chan_guide[channel]["listing"][0][time_item]).endswith(tuple(["+0000", "+00:00"])):
|
|
chan_dict["listing_%s" % time_item] = str(sorted_chan_guide[channel]["listing"][0][time_item])
|
|
else:
|
|
chan_dict["listing_%s" % time_item] = str(datetime.datetime.fromtimestamp(sorted_chan_guide[channel]["listing"][0][time_item]))
|
|
|
|
if source in ["blocks", "origin", self.fhdhr.config.dict["main"]["dictpopname"]]:
|
|
chan_obj = self.fhdhr.device.channels.get_channel_obj("origin_id", sorted_chan_guide[channel]["id"])
|
|
|
|
chan_dict["name"] = chan_obj.dict["name"]
|
|
chan_dict["number"] = chan_obj.number
|
|
chan_dict["chan_thumbnail"] = chan_obj.thumbnail
|
|
chan_dict["enabled"] = chan_obj.dict["enabled"]
|
|
chan_dict["m3u_url"] = chan_obj.m3u_url
|
|
|
|
chan_dict["listing_thumbnail"] = chan_dict["listing_thumbnail"] or chan_obj.thumbnail
|
|
else:
|
|
if not chan_dict["listing_thumbnail"]:
|
|
chan_dict["listing_thumbnail"] = chan_dict["chan_thumbnail"]
|
|
if not chan_dict["listing_thumbnail"]:
|
|
chan_dict["listing_thumbnail"] = "/api/images?method=generate&type=channel&message=%s" % chan_dict["number"]
|
|
|
|
chan_guide_list.append(chan_dict)
|
|
|
|
epg_json = json.dumps(chan_guide_list, indent=4)
|
|
|
|
return Response(status=200,
|
|
response=epg_json,
|
|
mimetype='application/json')
|
|
|
|
elif method == "update":
|
|
self.fhdhr.device.epg.update(source)
|
|
|
|
elif method == "clearcache":
|
|
self.fhdhr.device.epg.clear_epg_cache(source)
|
|
|
|
else:
|
|
return "%s Invalid Method" % method
|
|
|
|
if redirect_url:
|
|
return redirect("%s?retmessage=%s" % (redirect_url, urllib.parse.quote("%s Success" % method)))
|
|
else:
|
|
return "%s Success" % method
|