StreamingCommunity 2.5.2__py3-none-any.whl → 2.5.6__py3-none-any.whl

StreamingCommunity/Api/Site/guardaserie/__init__.py

@@ -1,51 +1,51 @@
-# 09.06.24
-
-from urllib.parse import quote_plus
-
-
-# Internal utilities
-from StreamingCommunity.Util.console import console, msg
-
-
-# Logic class
-from .site import title_search, run_get_select_title, media_search_manager
-from .series import download_series
-
-
-# Variable
-indice = 4
-_useFor = "serie"
-_deprecate = False
-_priority = 2
-_engineDownload = "hls"
-from .costant import SITE_NAME
-
-
-def search(string_to_search: str = None, get_onylDatabase: bool = False):
-    """
-    Main function of the application for film and series.
-    """
-
-    if string_to_search is None:
-        string_to_search = msg.ask(f"\n[purple]Insert word to search in [green]{SITE_NAME}").strip()
-    
-    # Search on database
-    len_database = title_search(quote_plus(string_to_search))
-
-    # Return list of elements
-    if get_onylDatabase:
-        return media_search_manager
-
-    if len_database > 0:
-
-        # Select title from list
-        select_title = run_get_select_title()
-
-        # Download only film
-        download_series(select_title)
-
-    else:
-        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
-
-        # Retry
+# 09.06.24
+
+from urllib.parse import quote_plus
+
+
+# Internal utilities
+from StreamingCommunity.Util.console import console, msg
+
+
+# Logic class
+from .site import title_search, run_get_select_title, media_search_manager
+from .series import download_series
+
+
+# Variable
+indice = 4
+_useFor = "serie"
+_deprecate = False
+_priority = 2
+_engineDownload = "hls"
+from .costant import SITE_NAME
+
+
+def search(string_to_search: str = None, get_onylDatabase: bool = False):
+    """
+    Main function of the application for film and series.
+    """
+
+    if string_to_search is None:
+        string_to_search = msg.ask(f"\n[purple]Insert word to search in [green]{SITE_NAME}").strip()
+    
+    # Search on database
+    len_database = title_search(quote_plus(string_to_search))
+
+    # Return list of elements
+    if get_onylDatabase:
+        return media_search_manager
+
+    if len_database > 0:
+
+        # Select title from list
+        select_title = run_get_select_title()
+
+        # Download only film
+        download_series(select_title)
+
+    else:
+        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
+
+        # Retry
         search()

StreamingCommunity/Api/Site/guardaserie/costant.py

@@ -1,19 +1,19 @@
-# 09.06.24
-
-import os
-
-
-# Internal utilities
-from StreamingCommunity.Util._jsonConfig import config_manager
-
-
-SITE_NAME = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
-ROOT_PATH = config_manager.get('DEFAULT', 'root_path')
-DOMAIN_NOW = config_manager.get_dict('SITE', SITE_NAME)['domain']
-
-SERIES_FOLDER = os.path.join(ROOT_PATH, config_manager.get('DEFAULT', 'serie_folder_name'))
-MOVIE_FOLDER = os.path.join(ROOT_PATH, config_manager.get('DEFAULT', 'movie_folder_name'))
-
-if config_manager.get_bool("DEFAULT", "add_siteName"):
-    SERIES_FOLDER = os.path.join(ROOT_PATH, SITE_NAME, config_manager.get('DEFAULT', 'serie_folder_name'))
-    MOVIE_FOLDER = os.path.join(ROOT_PATH, SITE_NAME, config_manager.get('DEFAULT', 'movie_folder_name'))   
+# 09.06.24
+
+import os
+
+
+# Internal utilities
+from StreamingCommunity.Util._jsonConfig import config_manager
+
+
+SITE_NAME = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
+ROOT_PATH = config_manager.get('DEFAULT', 'root_path')
+DOMAIN_NOW = config_manager.get_dict('SITE', SITE_NAME)['domain']
+
+SERIES_FOLDER = os.path.join(ROOT_PATH, config_manager.get('DEFAULT', 'serie_folder_name'))
+MOVIE_FOLDER = os.path.join(ROOT_PATH, config_manager.get('DEFAULT', 'movie_folder_name'))
+
+if config_manager.get_bool("DEFAULT", "add_siteName"):
+    SERIES_FOLDER = os.path.join(ROOT_PATH, SITE_NAME, config_manager.get('DEFAULT', 'serie_folder_name'))
+    MOVIE_FOLDER = os.path.join(ROOT_PATH, SITE_NAME, config_manager.get('DEFAULT', 'movie_folder_name'))   

StreamingCommunity/Api/Site/guardaserie/series.py

@@ -1,198 +1,200 @@
-# 13.06.24
-
-import os
-import sys
-
-
-# Internal utilities
-from StreamingCommunity.Util.console import console, msg
-from StreamingCommunity.Util.message import start_message
-from StreamingCommunity.Util.call_stack import get_call_stack
-from StreamingCommunity.Util.table import TVShowManager
-from StreamingCommunity.Lib.Downloader import HLS_Downloader
-
-
-# Logic class
-from StreamingCommunity.Api.Template.Util import manage_selection, map_episode_title, dynamic_format_number, validate_selection, validate_episode_selection, execute_search
-from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
-
-
-# Player
-from .util.ScrapeSerie import GetSerieInfo
-from StreamingCommunity.Api.Player.supervideo import VideoSource
-
-
-# Variable
-from .costant import SERIES_FOLDER
-
-
-
-def download_video(index_season_selected: int, index_episode_selected: int, scape_info_serie: GetSerieInfo) -> str:
-    """
-    Download a single episode video.
-
-    Parameters:
-        - tv_name (str): Name of the TV series.
-        - index_season_selected (int): Index of the selected season.
-        - index_episode_selected (int): Index of the selected episode.
-
-    Return:
-        - str: output path
-    """
-    start_message()
-    index_season_selected = dynamic_format_number(index_season_selected)
-
-    # Get info about episode
-    obj_episode = scape_info_serie.list_episodes[index_episode_selected - 1]
-    console.print(f"[yellow]Download: [red]{index_season_selected}:{index_episode_selected} {obj_episode.get('name')}")
-    print()
-
-    # Define filename and path for the downloaded video
-    mp4_name = f"{map_episode_title(scape_info_serie.tv_name, index_season_selected, index_episode_selected, obj_episode.get('name'))}.mp4"
-    mp4_path = os.path.join(SERIES_FOLDER, scape_info_serie.tv_name, f"S{index_season_selected}")
-
-    # Setup video source
-    video_source = VideoSource(obj_episode.get('url'))
-
-    # Get m3u8 master playlist
-    master_playlist = video_source.get_playlist()
-    
-    # Download the film using the m3u8 playlist, and output filename
-    r_proc = HLS_Downloader(
-        m3u8_playlist=master_playlist, 
-        output_filename=os.path.join(mp4_path, mp4_name)
-    ).start()
-
-    """if r_proc == 404:
-        time.sleep(2)
-
-        # Re call search function
-        if msg.ask("[green]Do you want to continue [white]([red]y[white])[green] or return at home[white]([red]n[white]) ", choices=['y', 'n'], default='y', show_choices=True) == "n":
-            frames = get_call_stack()
-            execute_search(frames[-4])"""
-
-    if r_proc != None:
-        console.print("[green]Result: ")
-        console.print(r_proc)
-
-    return os.path.join(mp4_path, mp4_name)
-
-
-def download_episode(scape_info_serie: GetSerieInfo, index_season_selected: int, download_all: bool = False) -> None:
-    """
-    Download all episodes of a season.
-
-    Parameters:
-        - tv_name (str): Name of the TV series.
-        - index_season_selected (int): Index of the selected season.
-        - download_all (bool): Download all seasons episodes
-    """
-
-    # Start message and collect information about episodes
-    start_message()
-    list_dict_episode = scape_info_serie.get_episode_number(index_season_selected)
-    episodes_count = len(list_dict_episode)
-
-    if download_all:
-
-        # Download all episodes without asking
-        for i_episode in range(1, episodes_count + 1):
-            download_video(index_season_selected, i_episode, scape_info_serie)
-        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(scape_info_serie.list_episodes)
-        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(index_season_selected, i_episode, scape_info_serie)
-
-
-def download_series(dict_serie: MediaItem) -> None:
-    """
-    Download all episodes of a TV series.
-
-    Parameters:
-        - dict_serie (MediaItem): obj with url name type and score
-    """
-
-    # Start message and set up video source
-    start_message()
-
-    # Init class
-    scape_info_serie = GetSerieInfo(dict_serie)
-
-    # Collect information about seasons
-    seasons_count = scape_info_serie.get_seasons_number()
-
-    # Prompt user for season selection and download episodes
-    console.print(f"\n[green]Seasons found: [red]{seasons_count}")
-    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(scape_info_serie, i_season, download_all=True)
-        else:
-
-            # Otherwise, let the user select specific episodes for the single season
-            download_episode(scape_info_serie, i_season, download_all=False)
-
-
-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 = TVShowManager()
-    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 media in obj_episode_manager:
-        table_show_manager.add_tv_show({
-            'Index': str(media.get('number')),
-            '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
+# 13.06.24
+
+import os
+import sys
+from typing import Tuple
+
+
+# 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
+
+
+# Logic class
+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 .util.ScrapeSerie import GetSerieInfo
+from StreamingCommunity.Api.Player.supervideo import VideoSource
+
+
+# Variable
+from .costant import SERIES_FOLDER
+
+
+
+def download_video(index_season_selected: int, index_episode_selected: int, scape_info_serie: GetSerieInfo) -> Tuple[str,bool]:
+    """
+    Download a single episode video.
+
+    Parameters:
+        - tv_name (str): Name of the TV series.
+        - 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 = scape_info_serie.list_episodes[index_episode_selected - 1]
+    console.print(f"[yellow]Download: [red]{index_season_selected}:{index_episode_selected} {obj_episode.get('name')}\n")
+    console.print(f"[cyan]You can safely stop the download with [bold]Ctrl+c[bold] [cyan] \n")
+
+    # Define filename and path for the downloaded video
+    mp4_name = f"{map_episode_title(scape_info_serie.tv_name, index_season_selected, index_episode_selected, obj_episode.get('name'))}.mp4"
+    mp4_path = os.path.join(SERIES_FOLDER, scape_info_serie.tv_name, f"S{index_season_selected}")
+
+    # Setup video source
+    video_source = VideoSource(obj_episode.get('url'))
+
+    # Get m3u8 master playlist
+    master_playlist = video_source.get_playlist()
+    
+    # Download the film using the m3u8 playlist, and output filename
+    r_proc = HLS_Downloader(
+        m3u8_url=master_playlist, 
+        output_path=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'], r_proc['stopped']
+
+
+def download_episode(scape_info_serie: GetSerieInfo, index_season_selected: int, download_all: bool = False) -> None:
+    """
+    Download all episodes of a season.
+
+    Parameters:
+        - tv_name (str): Name of the TV series.
+        - index_season_selected (int): Index of the selected season.
+        - download_all (bool): Download all seasons episodes
+    """
+
+    # Start message and collect information about episodes
+    start_message()
+    list_dict_episode = scape_info_serie.get_episode_number(index_season_selected)
+    episodes_count = len(list_dict_episode)
+
+    if download_all:
+
+        # Download all episodes without asking
+        for i_episode in range(1, episodes_count + 1):
+            path, stopped = download_video(index_season_selected, i_episode, scape_info_serie)
+
+            if stopped:
+                break
+
+        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(scape_info_serie.list_episodes)
+        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:
+            path, stopped = download_video(index_season_selected, i_episode, scape_info_serie)
+
+            if stopped:
+                break
+
+
+def download_series(dict_serie: MediaItem) -> None:
+    """
+    Download all episodes of a TV series.
+
+    Parameters:
+        - dict_serie (MediaItem): obj with url name type and score
+    """
+
+    # Start message and set up video source
+    start_message()
+
+    # Init class
+    scape_info_serie = GetSerieInfo(dict_serie)
+
+    # Collect information about seasons
+    seasons_count = scape_info_serie.get_seasons_number()
+
+    # Prompt user for season selection and download episodes
+    console.print(f"\n[green]Seasons found: [red]{seasons_count}")
+    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(scape_info_serie, i_season, download_all=True)
+        else:
+
+            # Otherwise, let the user select specific episodes for the single season
+            download_episode(scape_info_serie, i_season, download_all=False)
+
+
+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 = TVShowManager()
+    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 media in obj_episode_manager:
+        table_show_manager.add_tv_show({
+            'Index': str(media.get('number')),
+            '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