mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-06 11:35:29 +00:00
2b4d355d17
Add separate cookie manager in config. Add api to emulate node js . Add 0 at the end of episode. Add except time vs real time in HLS.
112 lines
2.9 KiB
Python
112 lines
2.9 KiB
Python
# 11.03.24
|
|
|
|
import os
|
|
import logging
|
|
|
|
|
|
# Internal utilities
|
|
from Src.Util.console import console, msg
|
|
from Src.Lib.Downloader import HLS_Downloader
|
|
from Src.Util.message import start_message
|
|
from ..Template import manage_selection
|
|
|
|
|
|
# Logic class
|
|
from .Core.Player.vixcloud import VideoSource
|
|
|
|
|
|
# Variable
|
|
from .costant import ROOT_PATH, SITE_NAME, SERIES_FOLDER, MOVIE_FOLDER
|
|
video_source = VideoSource()
|
|
|
|
|
|
|
|
def download_episode(index_select: int):
|
|
"""
|
|
Downloads the selected episode.
|
|
|
|
Args:
|
|
- index_select (int): Index of the episode to download.
|
|
"""
|
|
|
|
# Get information about the selected episode
|
|
obj_episode = video_source.get_info_episode(index_select)
|
|
|
|
start_message()
|
|
console.print(f"[yellow]Download: [red]EP_{obj_episode.number} \n")
|
|
|
|
# Get the embed URL for the episode
|
|
embed_url = video_source.get_embed(obj_episode.id)
|
|
|
|
# Parse parameter in embed text
|
|
video_source.parse_script(embed_url)
|
|
|
|
# Create output path
|
|
mp4_path = None
|
|
mp4_name = f"{index_select + 1}.mp4"
|
|
if video_source.is_series:
|
|
mp4_path = os.path.join(ROOT_PATH, SITE_NAME, SERIES_FOLDER, video_source.series_name)
|
|
else:
|
|
mp4_path = os.path.join(ROOT_PATH, SITE_NAME, MOVIE_FOLDER, video_source.series_name)
|
|
|
|
# Start downloading
|
|
HLS_Downloader(
|
|
m3u8_playlist = video_source.get_playlist(),
|
|
output_filename = os.path.join(mp4_path, mp4_name)
|
|
).start()
|
|
|
|
|
|
def donwload_series(tv_id: int, tv_name: str):
|
|
"""
|
|
Function to download episodes of a TV series.
|
|
|
|
Args:
|
|
- tv_id (int): The ID of the TV series.
|
|
- tv_name (str): The name of the TV series.
|
|
"""
|
|
|
|
# Set up video source
|
|
video_source.setup(
|
|
media_id = tv_id,
|
|
series_name = tv_name
|
|
)
|
|
|
|
# Get the count of episodes for the TV series
|
|
episoded_count = video_source.get_count_episodes()
|
|
console.log(f"[cyan]Episodes find: [red]{episoded_count}")
|
|
|
|
# Prompt user to select an episode index
|
|
last_command = msg.ask("\n[cyan]Insert media [red]index [yellow]or [red](*) [cyan]to download all media [yellow]or [red][1-2] [cyan]for a range of media")
|
|
|
|
# Manage user selection
|
|
list_episode_select = manage_selection(last_command, episoded_count)
|
|
|
|
# Download selected episodes
|
|
if len(list_episode_select) == 1 and last_command != "*":
|
|
download_episode(list_episode_select[0]-1)
|
|
|
|
# Download all other episodes selecter
|
|
else:
|
|
for i_episode in list_episode_select:
|
|
download_episode(i_episode-1)
|
|
|
|
|
|
def donwload_film(id_film: int, title_name: str):
|
|
"""
|
|
Function to download a film.
|
|
|
|
Args:
|
|
- id_film (int): The ID of the film.
|
|
- title_name (str): The title of the film.
|
|
"""
|
|
|
|
# Set up video source
|
|
video_source.setup(
|
|
media_id = id_film,
|
|
series_name = title_name
|
|
)
|
|
video_source.is_series = False
|
|
|
|
# Start download
|
|
download_episode(0)
|