mirror of
https://github.com/Viren070/MediaFusion.git
synced 2025-12-01 23:21:11 +01:00
Mohamed Zumair
48b3f5e85f
* Migrate streaming providers functions to async & improve performance * Fix streaming provider file selection & validate file extension as video type * Improve RD torrent handling * Find file based on filename or the largest file or episode based and remove index base file finding due to not trustable * Add support for updating torrent matadata with offcloud * Fix RD background task to update torrent metadata * Fix OC file size callback function * Fix AD, DL & PM * Fix PikPak httpx client leak & use medialink for speed & set 5 min for token cache * Migrate to AIOSeedrcc & fix season pack file selecting & refactorings * Qbittorrent: Add support for season pack file finding & disable dht, pex, lsd on private indexer torrent * Migrate to aiohttp for stability on resource fetching & bugfixes * Migrated to aioseedrcc & update libs * pass indexer type & remove Upper info hash finding
107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
import asyncio
|
|
from typing import List, Optional
|
|
|
|
from fastapi import BackgroundTasks
|
|
|
|
from db.models import TorrentStreams
|
|
from db.schemas import UserData
|
|
from streaming_providers.exceptions import ProviderException
|
|
from streaming_providers.offcloud.client import OffCloud
|
|
|
|
|
|
async def get_video_url_from_offcloud(
|
|
info_hash: str,
|
|
magnet_link: str,
|
|
user_data: UserData,
|
|
stream: TorrentStreams,
|
|
background_tasks: BackgroundTasks,
|
|
filename: Optional[str] = None,
|
|
season: Optional[int] = None,
|
|
episode: Optional[int] = None,
|
|
max_retries: int = 5,
|
|
retry_interval: int = 5,
|
|
**kwargs,
|
|
) -> str:
|
|
async with OffCloud(token=user_data.streaming_provider.token) as oc_client:
|
|
# Check if the torrent already exists
|
|
torrent_info = await oc_client.get_available_torrent(info_hash)
|
|
if torrent_info:
|
|
request_id = torrent_info.get("requestId")
|
|
torrent_info = await oc_client.get_torrent_info(request_id)
|
|
if torrent_info["status"] == "downloaded":
|
|
return await oc_client.create_download_link(
|
|
request_id,
|
|
torrent_info,
|
|
stream,
|
|
filename,
|
|
season,
|
|
episode,
|
|
background_tasks,
|
|
)
|
|
if torrent_info["status"] == "error":
|
|
raise ProviderException(
|
|
f"Error transferring magnet link to OffCloud. {torrent_info['errorMessage']}",
|
|
"transfer_error.mp4",
|
|
)
|
|
else:
|
|
# If torrent doesn't exist, add it
|
|
response_data = await oc_client.add_magnet_link(magnet_link)
|
|
request_id = response_data["requestId"]
|
|
|
|
# Wait for download completion and get the direct link
|
|
torrent_info = await oc_client.wait_for_status(
|
|
request_id, "downloaded", max_retries, retry_interval
|
|
)
|
|
return await oc_client.create_download_link(
|
|
request_id,
|
|
torrent_info,
|
|
stream,
|
|
filename,
|
|
season,
|
|
episode,
|
|
background_tasks,
|
|
)
|
|
|
|
|
|
async def update_oc_cache_status(
|
|
streams: List[TorrentStreams], user_data: UserData, **kwargs
|
|
):
|
|
"""Updates the cache status of streams based on OffCloud's instant availability."""
|
|
try:
|
|
async with OffCloud(token=user_data.streaming_provider.token) as oc_client:
|
|
instant_availability_data = (
|
|
await oc_client.get_torrent_instant_availability(
|
|
[stream.id for stream in streams]
|
|
)
|
|
)
|
|
for stream in streams:
|
|
stream.cached = stream.id in instant_availability_data
|
|
except ProviderException:
|
|
pass
|
|
|
|
|
|
async def fetch_downloaded_info_hashes_from_oc(
|
|
user_data: UserData, **kwargs
|
|
) -> List[str]:
|
|
"""Fetches the info_hashes of all torrents downloaded in the OffCloud account."""
|
|
try:
|
|
async with OffCloud(token=user_data.streaming_provider.token) as oc_client:
|
|
available_torrents = await oc_client.get_user_torrent_list()
|
|
return [
|
|
torrent["originalLink"].split("btih:")[1].split("&")[0]
|
|
for torrent in available_torrents
|
|
if "btih:" in torrent["originalLink"]
|
|
]
|
|
except ProviderException:
|
|
return []
|
|
|
|
|
|
async def delete_all_torrents_from_oc(user_data: UserData, **kwargs):
|
|
"""Deletes all torrents from the Offcloud account."""
|
|
async with OffCloud(token=user_data.streaming_provider.token) as oc_client:
|
|
torrents = await oc_client.get_user_torrent_list()
|
|
await asyncio.gather(
|
|
*[oc_client.delete_torrent(torrent["requestId"]) for torrent in torrents],
|
|
return_exceptions=True,
|
|
)
|