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

StreamingCommunity/Api/Site/streamingcommunity/series.py

@@ -1,207 +1,259 @@
-# 3.12.23
-
-import os
-import sys
-import time
-
-
-# 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 .util.ScrapeSerie import ScrapeSerie
-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 StreamingCommunity.Api.Player.vixcloud import VideoSource
-
-
-# Variable
-from .costant import SITE_NAME, SERIES_FOLDER
-
-
-
-def download_video(index_season_selected: int, index_episode_selected: int, scrape_serie: ScrapeSerie, video_source: VideoSource) -> str:
-    """
-    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
-    """
-    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()
-
-    # 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 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(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
-        for i_episode in list_episode_select:
-            download_video(index_season_selected, i_episode, scrape_serie, video_source)
-
-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.
-    """
-
-    # 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}")
-    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)
-
-
-def display_episodes_list(scrape_serie) -> 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'},
-        "Duration": {'color': 'green'}
-    }
-    table_show_manager.add_column(column_info)
-
-    # Populate the table with episodes information
-    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)
-        })
-
-    # 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
+# 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_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(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):
+            path, stopped = download_video(index_season_selected, i_episode, scrape_serie, video_source)
+
+            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(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:
+            path, stopped = download_video(index_season_selected, i_episode, scrape_serie, video_source)
+
+            if stopped:
+                break
+
+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