mirror of
https://github.com/Viren070/AIOStremio.git
synced 2025-12-01 23:24:19 +01:00
Viren070
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import base64
|
|
import os
|
|
from typing import Dict, List
|
|
|
|
import httpx
|
|
from fastapi import HTTPException
|
|
|
|
from utils.config import config
|
|
from utils.logger import logger
|
|
|
|
from .base import StreamingService
|
|
|
|
|
|
class DebridioService(StreamingService):
|
|
def __init__(self):
|
|
self.base_url = "https://debridio.adobotec.com"
|
|
self.debrid_api_key = config.get_addon_debrid_api_key("debridio")
|
|
self.debrid_service = config.get_addon_debrid_service("debridio")
|
|
self.options = f'{{"provider":"{self.debrid_service}","apiKey":"{self.debrid_api_key}","disableUncached":false,"qualityOrder":[],"excludeSize":"","maxReturnPerQuality":""}}'
|
|
self.options_encoded = base64.b64encode(self.options.encode()).decode("utf-8")
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return "Debridio"
|
|
|
|
async def _fetch_from_debridio(self, url: str) -> Dict:
|
|
async with httpx.AsyncClient() as client:
|
|
try:
|
|
response = await client.get(url)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
logger.debug(f"Debridio response: {data}")
|
|
return data
|
|
except httpx.HTTPError as e:
|
|
logger.error(f"Debridio request failed: {str(e)}")
|
|
raise HTTPException(status_code=502, detail="Upstream service error")
|
|
|
|
async def get_streams(self, meta_id: str) -> List[Dict]:
|
|
url = f"{self.base_url}/{self.options_encoded}/stream/{meta_id}"
|
|
logger.debug(f"Debridio stream url: {url}")
|
|
data = await self._fetch_from_debridio(url)
|
|
streams = data.get("streams", [])
|
|
for stream in streams:
|
|
stream["service"] = self.name
|
|
|
|
stream_name = stream.get("name", "")
|
|
if stream_name.startswith("["):
|
|
prefix = stream_name[1:stream_name.find("]")] if "]" in stream_name else ""
|
|
stream["is_cached"] = "+" in prefix
|
|
else:
|
|
stream["is_cached"] = True
|
|
|
|
return streams
|