mirror of
https://github.com/Viren070/AIOStremio.git
synced 2025-12-01 23:24:19 +01:00
Viren070
84 lines
4.3 KiB
Python
84 lines
4.3 KiB
Python
import json
|
|
import os
|
|
import time
|
|
import asyncio
|
|
import aiohttp
|
|
from utils.logger import logger
|
|
from utils.config import config
|
|
from utils.service_manager import ServiceManager
|
|
from services.base import StreamingService
|
|
from typing import List
|
|
from utils.cache import cache
|
|
|
|
_caching_seasons = set()
|
|
|
|
async def cache_season(meta_id: str, services: List[StreamingService]):
|
|
try:
|
|
# Extract series ID and season from meta_id (tt1234567:1:2 for S1E2)
|
|
parts = meta_id.split(":")
|
|
if len(parts) == 3:
|
|
series_id, season, _ = parts
|
|
|
|
season_key = f"{series_id}:{season}"
|
|
if season_key in _caching_seasons:
|
|
logger.debug(f"Season {season_key} is already being cached")
|
|
return
|
|
|
|
_caching_seasons.add(season_key)
|
|
logger.info(f"Starting to cache season {season} for series {series_id}")
|
|
|
|
try:
|
|
async with aiohttp.ClientSession() as session:
|
|
cinemeta_url = f"https://v3-cinemeta.strem.io/meta/{series_id.split(':')[0]}.json"
|
|
async with session.get(cinemeta_url) as response:
|
|
if response.status == 200:
|
|
data = await response.json()
|
|
if "meta" in data and "videos" in data["meta"]:
|
|
# Filter episodes for the current season
|
|
season_episodes = [
|
|
video for video in data["meta"]["videos"]
|
|
if video.get("season") == int(season)
|
|
]
|
|
|
|
name = data["meta"]["name"]
|
|
|
|
logger.info(f"Found {len(season_episodes)} episodes in season {season} for {name}")
|
|
|
|
start_time = time.time()
|
|
|
|
# Create service manager instance
|
|
service_manager = ServiceManager(services)
|
|
|
|
for episode in season_episodes:
|
|
ep_num = episode.get("episode", 0)
|
|
if meta_id.endswith('.json'):
|
|
ep_meta_id = f"{series_id.split(':')[0]}:{season}:{ep_num}.json"
|
|
else:
|
|
ep_meta_id = f"{series_id.split(':')[0]}:{season}:{ep_num}"
|
|
ep_name = f"S{season}E{ep_num} - {episode.get('name', 'Unknown')}"
|
|
|
|
logger.info(f"Caching episode {ep_name} of {name}")
|
|
|
|
# Fetch streams using ServiceManager
|
|
try:
|
|
streams = await service_manager.fetch_all_streams(ep_meta_id)
|
|
if streams:
|
|
# Store streams in Redis cache
|
|
cache_key = f"raw_streams:{ep_meta_id}"
|
|
await cache.set(cache_key, {"streams": streams}, ttl=config.cache_ttl_seconds)
|
|
logger.info(f"Successfully cached streams for {ep_name}")
|
|
else:
|
|
logger.warning(f"No streams found for {ep_name}")
|
|
except Exception as e:
|
|
logger.error(f"Error fetching streams for {ep_name}: {str(e)}")
|
|
|
|
# Wait before next episode to avoid rate limits
|
|
await asyncio.sleep(60)
|
|
|
|
logger.info(f"Completed caching for all episodes in season {season} of {name} in {time.time() - start_time} seconds")
|
|
finally:
|
|
_caching_seasons.remove(season_key)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error in season caching: {str(e)}", exc_info=True)
|