From dbce0b6745152a1594f34ae3aed808315481ee5d Mon Sep 17 00:00:00 2001
From: deathbybandaid <github@deathbybandaid.net>
Date: Tue, 27 Oct 2020 11:03:29 -0400
Subject: [PATCH] Add EPG debug

---
 fHDHR/api/hub/device/__init__.py |  2 +-
 fHDHR/api/hub/device/epg.py      | 14 ++++++++++++++
 fHDHR/api/hub/device/tuners.py   |  9 ++++++---
 3 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/fHDHR/api/hub/device/__init__.py b/fHDHR/api/hub/device/__init__.py
index 7b337f3..a01c8dd 100644
--- a/fHDHR/api/hub/device/__init__.py
+++ b/fHDHR/api/hub/device/__init__.py
@@ -14,7 +14,7 @@ class fHDHR_Device():
 
         self.epg = epg.EPG(settings, self.channels)
 
-        self.tuners = Tuners(settings)
+        self.tuners = Tuners(settings, self.epg)
 
         self.watch = WatchStream(settings, self.channels, self.tuners)
 
diff --git a/fHDHR/api/hub/device/epg.py b/fHDHR/api/hub/device/epg.py
index 69921a7..ae1c1a1 100644
--- a/fHDHR/api/hub/device/epg.py
+++ b/fHDHR/api/hub/device/epg.py
@@ -1,6 +1,7 @@
 import os
 import json
 import time
+import datetime
 from collections import OrderedDict
 from multiprocessing import Process
 
@@ -31,6 +32,19 @@ class EPG():
             self.epgscan = Process(target=self.epgServerProcess)
             self.epgscan.start()
 
+    def whats_on_now(self, channel):
+        epgdict = self.get_epg()
+        listings = epgdict[channel]["listing"]
+        for listing in listings:
+            nowtime = datetime.datetime.utcnow()
+            start_time = datetime.datetime.strptime(listing["time_start"], '%Y%m%d%H%M%S +0000')
+            end_time = datetime.datetime.strptime(listing["time_end"], '%Y%m%d%H%M%S +0000')
+            if start_time <= nowtime <= end_time:
+                epgitem = epgdict[channel].copy()
+                epgitem["listing"] = [listing]
+                return epgitem
+        return None
+
     def get_epg(self):
         epgdict = None
         if os.path.isfile(self.epg_cache_file):
diff --git a/fHDHR/api/hub/device/tuners.py b/fHDHR/api/hub/device/tuners.py
index addf035..660497b 100644
--- a/fHDHR/api/hub/device/tuners.py
+++ b/fHDHR/api/hub/device/tuners.py
@@ -6,8 +6,9 @@ from fHDHR.tools import humanized_time
 
 
 class Tuner():
-    def __init__(self, inum):
+    def __init__(self, inum, epg):
         self.number = inum
+        self.epg = epg
         self.tuner_lock = threading.Lock()
         self.set_off_status()
 
@@ -37,6 +38,7 @@ class Tuner():
                 humanized_time(
                     int((datetime.datetime.utcnow() - current_status["time_start"]).total_seconds())))
             current_status["time_start"] = str(current_status["time_start"])
+            current_status["epg"] = self.epg.whats_on_now(current_status["accessed"].split("v")[-1])
         return current_status
 
     def set_off_status(self):
@@ -45,12 +47,13 @@ class Tuner():
 
 class Tuners():
 
-    def __init__(self, settings):
+    def __init__(self, settings, epg):
         self.config = settings
+        self.epg = epg
         self.max_tuners = int(self.config.dict["fhdhr"]["tuner_count"])
 
         for i in range(1, self.max_tuners + 1):
-            exec("%s = %s" % ("self.tuner_" + str(i), "Tuner(i)"))
+            exec("%s = %s" % ("self.tuner_" + str(i), "Tuner(i, epg)"))
 
     def tuner_grab(self, stream_args, tunernum=None):
         tunerselected = None