144 lines
7.0 KiB
Python
144 lines
7.0 KiB
Python
import datetime
|
|
|
|
|
|
class OriginEPG():
|
|
|
|
def __init__(self, fhdhr):
|
|
self.fhdhr = fhdhr
|
|
|
|
self.base_api = 'https://valencia-app-mds.xumo.com/v2/'
|
|
|
|
def update_epg(self, fhdhr_channels):
|
|
programguide = {}
|
|
|
|
todaydate = datetime.datetime.utcnow().date()
|
|
self.remove_stale_cache(todaydate)
|
|
|
|
for fhdhr_id in list(fhdhr_channels.list.keys()):
|
|
c = fhdhr_channels.list[fhdhr_id].dict
|
|
|
|
if str(c["number"]) not in list(programguide.keys()):
|
|
programguide[str(c["number"])] = {
|
|
"callsign": c["callsign"],
|
|
"name": c["name"],
|
|
"number": c["number"],
|
|
"id": c["origin_id"],
|
|
"thumbnail": c["thumbnail"],
|
|
"listing": [],
|
|
}
|
|
cached_items = self.get_cached(c["origin_id"])
|
|
print(len(cached_items))
|
|
for cached_item in cached_items:
|
|
|
|
for asset in cached_item["assets"]:
|
|
|
|
content_id = asset["id"]
|
|
content_cache = self.get_cached_content(content_id)
|
|
|
|
timestart = self.xumo_xmltime(asset['timestamps']["start"])
|
|
timeend = self.xumo_xmltime(asset['timestamps']["end"])
|
|
|
|
clean_prog_dict = {
|
|
"time_start": timestart,
|
|
"time_end": timeend,
|
|
"duration_minutes": str((asset['timestamps']["end"] - asset['timestamps']["start"]) / 60),
|
|
"thumbnail": None,
|
|
"title": content_cache['title'] or "Unavailable",
|
|
"sub-title": "Unavailable",
|
|
"description": self.getDescription(content_cache["descriptions"]) or "Unavailable",
|
|
"rating": "N/A",
|
|
"episodetitle": "Unavailable",
|
|
"releaseyear": None,
|
|
"genres": None,
|
|
"seasonnumber": None,
|
|
"episodenumber": None,
|
|
"isnew": None,
|
|
"id": str(content_id)
|
|
}
|
|
|
|
programguide[str(c["number"])]["listing"].append(clean_prog_dict)
|
|
|
|
return programguide
|
|
|
|
def getDescription(self, description):
|
|
if 'large' in description:
|
|
return description['large']
|
|
elif 'medium' in description:
|
|
return description['medium']
|
|
elif 'small' in description:
|
|
return description['small']
|
|
elif 'tiny' in description:
|
|
return description['tiny']
|
|
else:
|
|
return None
|
|
|
|
def xumo_xmltime(self, tm):
|
|
tm = datetime.datetime.fromtimestamp(tm/1000.0)
|
|
tm = str(tm.strftime('%Y%m%d%H%M%S')) + " +0000"
|
|
return tm
|
|
|
|
def get_cached_content(self, content_id):
|
|
cacheitem = self.fhdhr.db.get_cacheitem_value(str(content_id), "content_cache", "origin")
|
|
if cacheitem:
|
|
self.fhdhr.logger.info('FROM CACHE: ' + str(content_id))
|
|
return cacheitem
|
|
else:
|
|
content_url = "%sassets/asset/%s.json?f=title&f=providers&f=descriptions&f=runtime&f=availableSince" % (self.base_api, content_id)
|
|
self.fhdhr.logger.info('Fetching: ' + content_url)
|
|
try:
|
|
resp = self.fhdhr.web.session.get(content_url)
|
|
except self.fhdhr.web.exceptions.HTTPError:
|
|
self.fhdhr.logger.info('Got an error! Ignoring it.')
|
|
return None
|
|
result = resp.json()
|
|
self.fhdhr.db.set_cacheitem_value(str(content_id), "offline_cache", result, "origin")
|
|
return result
|
|
|
|
def get_cached(self, channel_id):
|
|
for hour_num in range(0, 24):
|
|
lineup_url = "%schannels/channel/%s/broadcast.json?hour=%s" % (self.base_api, channel_id, hour_num)
|
|
self.get_cached_item(channel_id, hour_num, lineup_url)
|
|
cache_list = self.fhdhr.db.get_cacheitem_value("cache_list", "offline_cache", "origin") or []
|
|
return [self.fhdhr.db.get_cacheitem_value(x, "offline_cache", "origin") for x in cache_list if x.startswith(channel_id)]
|
|
|
|
def get_cached_item(self, channel_id, cache_key, url):
|
|
cache_key = datetime.datetime.today().replace(hour=cache_key).timestamp()
|
|
cache_key = str(channel_id) + "_" + str(cache_key)
|
|
cacheitem = self.fhdhr.db.get_cacheitem_value(str(cache_key), "offline_cache", "origin")
|
|
if cacheitem:
|
|
self.fhdhr.logger.info('FROM CACHE: ' + str(cache_key))
|
|
return cacheitem
|
|
else:
|
|
self.fhdhr.logger.info('Fetching: ' + url)
|
|
try:
|
|
resp = self.fhdhr.web.session.get(url)
|
|
except self.fhdhr.web.exceptions.HTTPError:
|
|
self.fhdhr.logger.info('Got an error! Ignoring it.')
|
|
return
|
|
result = resp.json()
|
|
|
|
self.fhdhr.db.set_cacheitem_value(str(cache_key), "offline_cache", result, "origin")
|
|
cache_list = self.fhdhr.db.get_cacheitem_value("cache_list", "offline_cache", "origin") or []
|
|
cache_list.append(str(cache_key))
|
|
self.fhdhr.db.set_cacheitem_value("cache_list", "offline_cache", cache_list, "origin")
|
|
|
|
def remove_stale_cache(self, todaydate):
|
|
cache_clear_time = todaydate.strftime('%Y-%m-%dT%H:00:00')
|
|
cache_clear_time = datetime.datetime.strptime(cache_clear_time, '%Y-%m-%dT%H:%M:%S').timestamp()
|
|
cache_list = self.fhdhr.db.get_cacheitem_value("cache_list", "offline_cache", "origin") or []
|
|
cache_to_kill = []
|
|
for cacheitem in cache_list:
|
|
cachetime = str(cacheitem).split("_")[-1]
|
|
if float(cachetime) < cache_clear_time:
|
|
cache_to_kill.append(cacheitem)
|
|
self.fhdhr.db.delete_cacheitem_value(str(cacheitem), "offline_cache", "origin")
|
|
self.fhdhr.logger.info('Removing stale cache: ' + str(cacheitem))
|
|
self.fhdhr.db.set_cacheitem_value("cache_list", "offline_cache", [x for x in cache_list if x not in cache_to_kill], "origin")
|
|
|
|
def clear_cache(self):
|
|
cache_list = self.fhdhr.db.get_cacheitem_value("cache_list", "offline_cache", "origin") or []
|
|
for cacheitem in cache_list:
|
|
self.fhdhr.db.delete_cacheitem_value(cacheitem, "offline_cache", "origin")
|
|
self.fhdhr.logger.info('Removing cache: ' + str(cacheitem))
|
|
self.fhdhr.db.delete_cacheitem_value("cache_list", "offline_cache", "origin")
|