mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-06 19:45:24 +00:00
136 lines
3.6 KiB
Python
136 lines
3.6 KiB
Python
# 13.06.24
|
|
|
|
import os
|
|
import sys
|
|
import logging
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
# Internal utilities
|
|
from Src.Util.console import console
|
|
from Src.Util.message import start_message
|
|
from Src.Util.os import os_manager
|
|
from Src.Util.table import TVShowManager
|
|
from Src.Lib.Downloader import MP4_downloader
|
|
from ..Template import manage_selection, map_episode_title, validate_episode_selection
|
|
|
|
|
|
# Logic class
|
|
from ..Template.Class.SearchType import MediaItem
|
|
from .Player.ScrapeSerie import GetSerieInfo
|
|
from .Player.ddl import VideoSource
|
|
|
|
|
|
# Variable
|
|
from .costant import ROOT_PATH, SITE_NAME, SERIES_FOLDER
|
|
table_show_manager = TVShowManager()
|
|
video_source = VideoSource()
|
|
|
|
|
|
|
|
def download_video(scape_info_serie: GetSerieInfo, index_episode_selected: int) -> None:
|
|
"""
|
|
Download a single episode video.
|
|
|
|
Parameters:
|
|
- tv_name (str): Name of the TV series.
|
|
- index_episode_selected (int): Index of the selected episode.
|
|
"""
|
|
|
|
start_message()
|
|
|
|
# Get info about episode
|
|
obj_episode = scape_info_serie.list_episodes[index_episode_selected - 1]
|
|
console.print(f"[yellow]Download: [red]{obj_episode.get('name')}")
|
|
print()
|
|
|
|
# Define filename and path for the downloaded video
|
|
title_name = os_manager.get_sanitize_file(
|
|
f"{map_episode_title(scape_info_serie.tv_name, None, index_episode_selected, obj_episode.get('name'))}.mp4"
|
|
)
|
|
mp4_path = os.path.join(ROOT_PATH, SITE_NAME, SERIES_FOLDER, scape_info_serie.tv_name)
|
|
|
|
# Create output folder
|
|
os_manager.create_path(mp4_path)
|
|
|
|
# Setup video source
|
|
video_source.setup(obj_episode.get('url'))
|
|
|
|
# Get m3u8 master playlist
|
|
master_playlist = video_source.get_playlist()
|
|
|
|
# Parse start page url
|
|
parsed_url = urlparse(obj_episode.get('url'))
|
|
|
|
MP4_downloader(
|
|
url = master_playlist,
|
|
path = os.path.join(mp4_path, title_name),
|
|
referer = f"{parsed_url.scheme}://{parsed_url.netloc}/",
|
|
)
|
|
|
|
|
|
def download_thread(dict_serie: MediaItem):
|
|
"""
|
|
Download all episode of a thread
|
|
"""
|
|
|
|
# Start message and set up video source
|
|
start_message()
|
|
|
|
# Init class
|
|
scape_info_serie = GetSerieInfo(dict_serie)
|
|
|
|
# Collect information about thread
|
|
list_dict_episode = scape_info_serie.get_episode_number()
|
|
episodes_count = len(list_dict_episode)
|
|
|
|
# Display episodes list and manage user selection
|
|
last_command = display_episodes_list()
|
|
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
|
|
for i_episode in list_episode_select:
|
|
download_video(scape_info_serie, i_episode)
|
|
|
|
|
|
|
|
def display_episodes_list(obj_episode_manager) -> str:
|
|
"""
|
|
Display episodes list and handle user input.
|
|
|
|
Returns:
|
|
last_command (str): Last command entered by the user.
|
|
"""
|
|
|
|
# Set up table for displaying episodes
|
|
table_show_manager.set_slice_end(10)
|
|
|
|
# Add columns to the table
|
|
column_info = {
|
|
"Index": {'color': 'red'},
|
|
"Name": {'color': 'magenta'},
|
|
}
|
|
table_show_manager.add_column(column_info)
|
|
|
|
# Populate the table with episodes information
|
|
for i, media in enumerate(obj_episode_manager):
|
|
table_show_manager.add_tv_show({
|
|
'Index': str(i+1),
|
|
'Name': media.get('name'),
|
|
})
|
|
|
|
# Run the table and handle user input
|
|
last_command = table_show_manager.run()
|
|
|
|
if last_command == "q":
|
|
console.print("\n[red]Quit [white]...")
|
|
sys.exit(0)
|
|
|
|
return last_command
|