Update backend/main.py

backend/main.py

@@ -3,6 +3,7 @@ import time
 import asyncio
 import json
 import re
+import logging
 from datetime import datetime, timedelta
 from fastapi import FastAPI, HTTPException, Request, Depends, Query
 from fastapi.responses import Response, FileResponse
@@ -16,6 +17,9 @@ from backend.recorder import start_local_record, check_local_status, stream_file
 from backend.sync import upload_to_dataset
 from backend.config import load_sources, load_datasets, get_admin_credentials, generate_token, verify_token
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 app = FastAPI()
 app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])
 app.mount("/static", StaticFiles(directory="/app/frontend", html=True), name="static")
@@ -30,6 +34,7 @@ ADMIN_USER, ADMIN_PASS = get_admin_credentials()
 async def startup():
     await init_db()
     await sync_sources(load_sources())
+    logger.info("Application started")
 
 @app.get("/")
 async def root():
@@ -63,14 +68,34 @@ async def logout(credentials: HTTPAuthorizationCredentials = Depends(security)):
 async def get_sources_ep(current_user: dict = Depends(get_current_user)):
     return await get_sources()
 
+@app.get("/api/test-connection/{source_id}")
+async def test_connection(source_id: str, current_user: dict = Depends(get_current_user)):
+    src = next((s for s in await get_sources() if s["id"]==source_id), None)
+    if not src: 
+        raise HTTPException(404, "Source not found")
+    try:
+        async with httpx.AsyncClient(timeout=10.0, verify=False) as client:
+            res = await client.get(src["url"])
+            return {"status": "ok", "status_code": res.status_code, "url": src["url"]}
+    except Exception as e:
+        logger.error(f"Connection test failed: {str(e)}")
+        raise HTTPException(500, f"Connection failed: {str(e)}")
+
 @app.get("/api/epg/{source_id}")
 async def get_epg_ep(source_id: str, current_user: dict = Depends(get_current_user)):
     src = next((s for s in await get_sources() if s["id"]==source_id), None)
-    if not src: raise HTTPException(404, "Source not found")
+    if not src: 
+        raise HTTPException(404, "Source not found")
+    logger.info(f"Fetching EPG from {src['type']} at {src['url']}")
     adapter = get_adapter(src["type"], src["url"], src.get("api_key",""))
-    data = await adapter.get_epg()
-    await cache_epg(source_id, json.dumps(data), time.time())
-    return data
+    try:
+        data = await adapter.get_epg()
+        logger.info(f"EPG fetched: {len(data)} programs")
+        await cache_epg(source_id, json.dumps(data), time.time())
+        return data
+    except Exception as e:
+        logger.error(f"EPG fetch failed: {str(e)}")
+        raise HTTPException(500, f"Failed to fetch EPG: {str(e)}")
 
 @app.post("/api/record")
 async def schedule_record( data: dict, current_user: dict = Depends(get_current_user)):
@@ -172,4 +197,6 @@ async def monitor_remote(rec_id, source, adapter, rec_data):
                 token = DATASETS.get(rec_data["dataset_repo"])
                 asyncio.create_task(upload_to_dataset(local, rec_data["md5_name"], rec_data["original_name"], rec_data["dataset_repo"], token))
                 break
-        except: pass
+        except Exception as e:
+            logger.error(f"Monitor error: {str(e)}")
+            pass