e

EPG/fetch.py

@@ -32,6 +32,27 @@ client = httpx.AsyncClient(
     },
 )
 
+live_img = "https://i.gyazo.com/978f2eb4a199ca5b56b447aded0cb9e3.png"
+
+dummies = {
+    "Basketball.Dummy.us": live_img,
+    "Live.Event.us": live_img,
+    "MLB.Baseball.Dummy.us": None,
+    "MLS.Soccer.Dummy.us": None,
+    "NBA.Basketball.Dummy.us": None,
+    "NFL.Dummy.us": None,
+    "NHL.Hockey.Dummy.us": None,
+    "PPV.EVENTS.Dummy.us": live_img,
+    "Premier.League.Dummy.us": None,
+    "Racing.Dummy.us": None,
+    "Soccer.Dummy.us": live_img,
+    "Sports.Dummy.us": live_img,
+    "Tennis.Dummy.us": None,
+    "UEFA.Champions.League.Dummy.us": None,
+    "UFC.Fight.Pass.Dummy.us": live_img,
+    "WNBA.dummy.us": None,
+}
+
 
 async def fetch_xml(url: str) -> ET.Element:
     try:
@@ -50,9 +71,9 @@ async def fetch_xml(url: str) -> ET.Element:
 
 
 async def main() -> None:
-    data = json.loads(tvg_ids_file.read_text(encoding="utf-8"))
+    tvg_ids: dict[str, str] = json.loads(tvg_ids_file.read_text(encoding="utf-8"))
 
-    tvg_ids = {v["tvg-id"]: v["logo"] for v in data.values()}
+    tvg_ids |= dummies
 
     root = ET.Element("tv")
 
@@ -66,7 +87,8 @@ async def main() -> None:
         for channel in epg_data.findall("channel"):
             if (channel_id := channel.get("id")) in tvg_ids:
                 for icon_tag in channel.findall("icon"):
-                    icon_tag.set("src", tvg_ids[channel_id])
+                    if logo := tvg_ids.get(channel_id):
+                        icon_tag.set("src", logo)
 
                 if (url_tag := channel.find("url")) is not None:
                     channel.remove(url_tag)