Update backend/adapters.py

backend/adapters.py

@@ -1,5 +1,9 @@
 import httpx
 from datetime import datetime, timezone
+import logging
+
+logger = logging.getLogger(__name__)
+
 class BaseAdapter:
     def __init__(self, url, api_key):
         self.base_url = url.rstrip("/")
@@ -15,61 +19,117 @@ class BaseAdapter:
     async def schedule_record(self, channel_id, start_ts, end_ts, name): raise NotImplementedError
     async def get_record_status(self, record_id): raise NotImplementedError
     async def get_video_url(self, record_id): raise NotImplementedError
+
 class EPGStationAdapter(BaseAdapter):
     async def get_epg(self, hours=24):
         now = int(datetime.now(timezone.utc).timestamp())
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.get(f"{self.base_url}/api/v3/programs", headers=self.headers, params={"startAt": now, "endAt": now + hours*3600, "type": 0}, verify=False)
-            items = res.json().get("programs", []) if res.status_code == 200 else []
-            return [{"id": str(p["id"]), "title": p.get("name","Unknown"), "start": p["startAt"], "end": p["endAt"], "channel_id": str(p["channelId"]), "channel_name": p.get("channelName","")} for p in items]
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.get(f"{self.base_url}/api/v3/programs", headers=self.headers, params={"startAt": now, "endAt": now + hours*3600, "type": 0})
+                if res.status_code != 200:
+                    logger.error(f"EPGStation API error: {res.status_code} - {res.text}")
+                    return []
+                items = res.json().get("programs", [])
+                return [{"id": str(p["id"]), "title": p.get("name","Unknown"), "start": p["startAt"], "end": p["endAt"], "channel_id": str(p["channelId"]), "channel_name": p.get("channelName","")} for p in items]
+        except httpx.ConnectError as e:
+            logger.error(f"Connection failed: {str(e)}")
+            return []
+        except Exception as e:
+            logger.error(f"EPG fetch error: {str(e)}")
+            return []
+    
     async def schedule_record(self, channel_id, start_ts, end_ts, name):
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.post(f"{self.base_url}/api/v3/reserves", json={"channelId": channel_id, "startAt": int(start_ts), "endAt": int(end_ts), "name": name, "isSaveOriginal": True}, headers=self.headers, verify=False)
-            return str(res.json().get("id","")) if res.status_code == 200 else ""
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.post(f"{self.base_url}/api/v3/reserves", json={"channelId": channel_id, "startAt": int(start_ts), "endAt": int(end_ts), "name": name, "isSaveOriginal": True}, headers=self.headers)
+                return str(res.json().get("id","")) if res.status_code == 200 else ""
+        except Exception as e:
+            logger.error(f"Schedule error: {str(e)}")
+            return ""
+    
     async def get_record_status(self, record_id):
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.get(f"{self.base_url}/api/v3/recorded/{record_id}", headers=self.headers, verify=False)
-            return res.json().get("state",{}) if res.status_code == 200 else {}
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.get(f"{self.base_url}/api/v3/recorded/{record_id}", headers=self.headers)
+                return res.json().get("state",{}) if res.status_code == 200 else {}
+        except:
+            return {}
+    
     async def get_video_url(self, record_id):
         return f"{self.base_url}/api/v3/videos/{record_id}/stream"
+
 class KonomiTVAdapter(BaseAdapter):
     async def get_epg(self, hours=24):
-        async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
-            res = await c.get(f"{self.base_url}/api/v1/programs", headers=self.headers)
-            items = res.json().get("programs", []) if res.status_code == 200 else []
-            return [{"id": str(p["id"]), "title": p.get("title","Unknown"), "start": p.get("start_time",0), "end": p.get("end_time",0), "channel_id": str(p.get("channel_id","")), "channel_name": p.get("channel_name","")} for p in items]
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.get(f"{self.base_url}/api/v1/programs", headers=self.headers)
+                if res.status_code != 200:
+                    return []
+                items = res.json().get("programs", [])
+                return [{"id": str(p["id"]), "title": p.get("title","Unknown"), "start": p.get("start_time",0), "end": p.get("end_time",0), "channel_id": str(p.get("channel_id","")), "channel_name": p.get("channel_name","")} for p in items]
+        except Exception as e:
+            logger.error(f"KonomiTV error: {str(e)}")
+            return []
+    
     async def schedule_record(self, channel_id, start_ts, end_ts, name):
-        async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
-            res = await c.post(f"{self.base_url}/api/v1/reserve", json={"channel_id": channel_id, "start_time": int(start_ts), "end_time": int(end_ts)}, headers=self.headers)
-            return str(res.json().get("id","")) if res.status_code == 200 else ""
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.post(f"{self.base_url}/api/v1/reserve", json={"channel_id": channel_id, "start_time": int(start_ts), "end_time": int(end_ts)}, headers=self.headers)
+                return str(res.json().get("id","")) if res.status_code == 200 else ""
+        except:
+            return ""
+    
     async def get_record_status(self, record_id): return {"status": "completed"}
     async def get_video_url(self, record_id): return f"{self.base_url}/api/v1/stream/{record_id}"
+
 class MirakurunAdapter(BaseAdapter):
     async def get_epg(self, hours=24):
         now = int(datetime.now(timezone.utc).timestamp())
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.get(f"{self.base_url}/api/programs", headers=self.headers, params={"startAt": now, "endAt": now + hours*3600}, verify=False)
-            items = res.json() if res.status_code == 200 else []
-            return [{"id": str(p["id"]), "title": p.get("name","Unknown"), "start": p.get("startAt",0), "end": p.get("endAt",0), "channel_id": str(p.get("channelId","")), "channel_name": p.get("channelName","")} for p in items]
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.get(f"{self.base_url}/api/programs", headers=self.headers, params={"startAt": now, "endAt": now + hours*3600})
+                if res.status_code != 200:
+                    return []
+                items = res.json()
+                return [{"id": str(p["id"]), "title": p.get("name","Unknown"), "start": p.get("startAt",0), "end": p.get("endAt",0), "channel_id": str(p.get("channelId","")), "channel_name": p.get("channelName","")} for p in items]
+        except:
+            return []
+    
     async def schedule_record(self, channel_id, start_ts, end_ts, name):
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.post(f"{self.base_url}/api/reserves", json={"channelType": "GR", "channel": channel_id, "startAt": int(start_ts), "endAt": int(end_ts)}, headers=self.headers, verify=False)
-            return str(res.json().get("id","")) if res.status_code == 200 else ""
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.post(f"{self.base_url}/api/reserves", json={"channelType": "GR", "channel": channel_id, "startAt": int(start_ts), "endAt": int(end_ts)}, headers=self.headers)
+                return str(res.json().get("id","")) if res.status_code == 200 else ""
+        except:
+            return ""
+    
     async def get_record_status(self, record_id): return {"status": "completed"}
     async def get_video_url(self, record_id): return f"{self.base_url}/api/recorded/{record_id}/stream.mp4"
+
 class EDCBAdapter(BaseAdapter):
     async def get_epg(self, hours=24):
         now = int(datetime.now(timezone.utc).timestamp())
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.get(f"{self.base_url}/api/programs", headers=self.headers, params={"startAt": now, "endAt": now + hours*3600}, verify=False)
-            items = res.json() if res.status_code == 200 else []
-            return [{"id": str(p["id"]), "title": p.get("title","Unknown"), "start": p.get("startTime",0), "end": p.get("endTime",0), "channel_id": str(p.get("channelID","")), "channel_name": p.get("channelName","")} for p in items]
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.get(f"{self.base_url}/api/programs", headers=self.headers, params={"startAt": now, "endAt": now + hours*3600})
+                if res.status_code != 200:
+                    return []
+                items = res.json()
+                return [{"id": str(p["id"]), "title": p.get("title","Unknown"), "start": p.get("startTime",0), "end": p.get("endTime",0), "channel_id": str(p.get("channelID","")), "channel_name": p.get("channelName","")} for p in items]
+        except:
+            return []
+    
     async def schedule_record(self, channel_id, start_ts, end_ts, name):
-        async with httpx.AsyncClient(timeout=30.0) as c:
-            res = await c.post(f"{self.base_url}/api/reserve", json={"channelID": channel_id, "startTime": int(start_ts), "endTime": int(end_ts), "recMode": 4}, headers=self.headers, verify=False)
-            return str(res.json().get("id","")) if res.status_code == 200 else ""
+        try:
+            async with httpx.AsyncClient(timeout=30.0, verify=False) as c:
+                res = await c.post(f"{self.base_url}/api/reserve", json={"channelID": channel_id, "startTime": int(start_ts), "endTime": int(end_ts), "recMode": 4}, headers=self.headers)
+                return str(res.json().get("id","")) if res.status_code == 200 else ""
+        except:
+            return ""
+    
     async def get_record_status(self, record_id): return {"status": "completed"}
     async def get_video_url(self, record_id): return f"{self.base_url}/api/recfile/{record_id}"
+
 def get_adapter(source_type, url, api_key):
     mapping = {"epgstation": EPGStationAdapter, "mirakurun": MirakurunAdapter, "konomitv": KonomiTVAdapter, "edcb": EDCBAdapter}
     cls = mapping.get(source_type.lower(), EPGStationAdapter)