mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-07 03:55:24 +00:00
252 lines
8.7 KiB
Python
252 lines
8.7 KiB
Python
# 3.12.23
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
# Internal utilities
|
|
from StreamingCommunity.Util.console import console, msg
|
|
from StreamingCommunity.Util.message import start_message
|
|
from StreamingCommunity.Util.table import TVShowManager
|
|
from StreamingCommunity.Lib.Downloader import HLS_Downloader
|
|
from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
|
|
from StreamingCommunity.TelegramHelp.session import get_session, updateScriptId, deleteScriptId
|
|
|
|
|
|
# Logic class
|
|
from .util.ScrapeSerie import ScrapeSerie
|
|
from StreamingCommunity.Api.Template.Util import manage_selection, map_episode_title, dynamic_format_number, validate_selection, validate_episode_selection
|
|
from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
|
|
|
|
|
|
# Player
|
|
from StreamingCommunity.Api.Player.vixcloud import VideoSource
|
|
|
|
|
|
# Variable
|
|
from .costant import SITE_NAME, SERIES_FOLDER, TELEGRAM_BOT
|
|
|
|
|
|
def download_video(index_season_selected: int, index_episode_selected: int, scrape_serie: ScrapeSerie, video_source: VideoSource) -> tuple[str,bool]:
|
|
"""
|
|
Download a single episode video.
|
|
|
|
Parameters:
|
|
- index_season_selected (int): Index of the selected season.
|
|
- index_episode_selected (int): Index of the selected episode.
|
|
|
|
Return:
|
|
- str: output path
|
|
- bool: kill handler status
|
|
"""
|
|
start_message()
|
|
index_season_selected = dynamic_format_number(index_season_selected)
|
|
|
|
# Get info about episode
|
|
obj_episode = scrape_serie.episode_manager.get(index_episode_selected - 1)
|
|
console.print(f"[yellow]Download: [red]{index_season_selected}:{index_episode_selected} {obj_episode.name}")
|
|
print()
|
|
|
|
if TELEGRAM_BOT:
|
|
bot = get_bot_instance()
|
|
|
|
# Invio a telegram
|
|
bot.send_message(
|
|
f"Download in corso\nSerie: {scrape_serie.series_name}\nStagione: {index_season_selected}\nEpisodio: {index_episode_selected}\nTitolo: {obj_episode.name}",
|
|
None
|
|
)
|
|
|
|
# Get script_id and update it
|
|
script_id = get_session()
|
|
if script_id != "unknown":
|
|
updateScriptId(script_id, f"{scrape_serie.series_name} - S{index_season_selected} - E{index_episode_selected} - {obj_episode.name}")
|
|
|
|
# Define filename and path for the downloaded video
|
|
mp4_name = f"{map_episode_title(scrape_serie.series_name, index_season_selected, index_episode_selected, obj_episode.name)}.mp4"
|
|
mp4_path = os.path.join(SERIES_FOLDER, scrape_serie.series_name, f"S{index_season_selected}")
|
|
|
|
# Retrieve scws and if available master playlist
|
|
video_source.get_iframe(obj_episode.id)
|
|
video_source.get_content()
|
|
master_playlist = video_source.get_playlist()
|
|
|
|
# Download the episode
|
|
r_proc = HLS_Downloader(
|
|
m3u8_playlist=master_playlist,
|
|
output_filename=os.path.join(mp4_path, mp4_name)
|
|
).start()
|
|
|
|
if "error" in r_proc.keys():
|
|
try:
|
|
os.remove(r_proc['path'])
|
|
except:
|
|
pass
|
|
|
|
return r_proc['path']
|
|
|
|
def download_episode(index_season_selected: int, scrape_serie: ScrapeSerie, video_source: VideoSource, download_all: bool = False) -> None:
|
|
"""
|
|
Download episodes of a selected season.
|
|
|
|
Parameters:
|
|
- index_season_selected (int): Index of the selected season.
|
|
- download_all (bool): Download all episodes in the season.
|
|
"""
|
|
|
|
# Clean memory of all episodes and get the number of the season
|
|
scrape_serie.episode_manager.clear()
|
|
|
|
# Start message and collect information about episodes
|
|
start_message()
|
|
scrape_serie.collect_info_season(index_season_selected)
|
|
episodes_count = scrape_serie.episode_manager.length()
|
|
|
|
if download_all:
|
|
|
|
# Download all episodes without asking
|
|
for i_episode in range(1, episodes_count + 1):
|
|
download_video(index_season_selected, i_episode, scrape_serie, video_source)
|
|
console.print(f"\n[red]End downloaded [yellow]season: [red]{index_season_selected}.")
|
|
|
|
else:
|
|
|
|
# Display episodes list and manage user selection
|
|
last_command = display_episodes_list(scrape_serie)
|
|
list_episode_select = manage_selection(last_command, episodes_count)
|
|
|
|
try:
|
|
list_episode_select = validate_episode_selection(list_episode_select, episodes_count)
|
|
except ValueError as e:
|
|
console.print(f"[red]{str(e)}")
|
|
return
|
|
|
|
# Download selected episodes if not stopped
|
|
for i_episode in list_episode_select:
|
|
download_video(index_season_selected, i_episode, scrape_serie, video_source)[1]
|
|
|
|
def download_series(select_season: MediaItem, version: str) -> None:
|
|
"""
|
|
Download episodes of a TV series based on user selection.
|
|
|
|
Parameters:
|
|
- select_season (MediaItem): Selected media item (TV series).
|
|
- domain (str): Domain from which to download.
|
|
- version (str): Version of the site.
|
|
"""
|
|
if TELEGRAM_BOT:
|
|
bot = get_bot_instance()
|
|
|
|
# Start message and set up video source
|
|
start_message()
|
|
|
|
# Init class
|
|
scrape_serie = ScrapeSerie(SITE_NAME)
|
|
video_source = VideoSource(SITE_NAME, True)
|
|
|
|
# Setup video source
|
|
scrape_serie.setup(version, select_season.id, select_season.slug)
|
|
video_source.setup(select_season.id)
|
|
|
|
# Collect information about seasons
|
|
scrape_serie.collect_info_title()
|
|
seasons_count = scrape_serie.season_manager.seasons_count
|
|
|
|
# Prompt user for season selection and download episodes
|
|
console.print(f"\n[green]Seasons found: [red]{seasons_count}")
|
|
|
|
if TELEGRAM_BOT:
|
|
console.print("\n[cyan]Insert season number [yellow](e.g., 1), [red]* [cyan]to download all seasons, "
|
|
"[yellow](e.g., 1-2) [cyan]for a range of seasons, or [yellow](e.g., 3-*) [cyan]to download from a specific season to the end")
|
|
|
|
bot.send_message(f"Stagioni trovate: {seasons_count}", None)
|
|
|
|
index_season_selected = bot.ask(
|
|
"select_title_episode",
|
|
"Inserisci il numero della stagione (es. 1), * per scaricare tutte le stagioni, (es. 1-2) per un intervallo di stagioni, o (es. 3-*) per scaricare dalla stagione specificata fino alla fine",
|
|
None
|
|
)
|
|
|
|
else:
|
|
index_season_selected = msg.ask(
|
|
"\n[cyan]Insert season number [yellow](e.g., 1), [red]* [cyan]to download all seasons, "
|
|
"[yellow](e.g., 1-2) [cyan]for a range of seasons, or [yellow](e.g., 3-*) [cyan]to download from a specific season to the end"
|
|
)
|
|
|
|
# Manage and validate the selection
|
|
list_season_select = manage_selection(index_season_selected, seasons_count)
|
|
|
|
try:
|
|
list_season_select = validate_selection(list_season_select, seasons_count)
|
|
except ValueError as e:
|
|
console.print(f"[red]{str(e)}")
|
|
return
|
|
|
|
# Loop through the selected seasons and download episodes
|
|
for i_season in list_season_select:
|
|
if len(list_season_select) > 1 or index_season_selected == "*":
|
|
|
|
# Download all episodes if multiple seasons are selected or if '*' is used
|
|
download_episode(i_season, scrape_serie, video_source, download_all=True)
|
|
else:
|
|
|
|
# Otherwise, let the user select specific episodes for the single season
|
|
download_episode(i_season, scrape_serie, video_source, download_all=False)
|
|
|
|
if TELEGRAM_BOT:
|
|
bot.send_message(f"Finito di scaricare tutte le serie e episodi", None)
|
|
|
|
# Get script_id
|
|
script_id = get_session()
|
|
if script_id != "unknown":
|
|
deleteScriptId(script_id)
|
|
|
|
|
|
def display_episodes_list(scrape_serie) -> str:
|
|
"""
|
|
Display episodes list and handle user input.
|
|
|
|
Returns:
|
|
last_command (str): Last command entered by the user.
|
|
"""
|
|
if TELEGRAM_BOT:
|
|
bot = get_bot_instance()
|
|
|
|
# Set up table for displaying episodes
|
|
table_show_manager = TVShowManager()
|
|
table_show_manager.set_slice_end(10)
|
|
|
|
# Add columns to the table
|
|
column_info = {
|
|
"Index": {'color': 'red'},
|
|
"Name": {'color': 'magenta'},
|
|
"Duration": {'color': 'green'}
|
|
}
|
|
table_show_manager.add_column(column_info)
|
|
|
|
# Populate the table with episodes information
|
|
if TELEGRAM_BOT:
|
|
choices = []
|
|
|
|
for i, media in enumerate(scrape_serie.episode_manager.episodes):
|
|
table_show_manager.add_tv_show({
|
|
'Index': str(media.number),
|
|
'Name': media.name,
|
|
'Duration': str(media.duration)
|
|
})
|
|
|
|
if TELEGRAM_BOT:
|
|
choice_text = f"{media.number} - {media.name} ({media.duration} min)"
|
|
choices.append(choice_text)
|
|
|
|
if TELEGRAM_BOT:
|
|
if choices:
|
|
bot.send_message(f"Lista episodi:", choices)
|
|
|
|
# Run the table and handle user input
|
|
last_command = table_show_manager.run()
|
|
|
|
if last_command == "q" or last_command == "quit":
|
|
console.print("\n[red]Quit [white]...")
|
|
sys.exit(0)
|
|
|
|
return last_command |