StreamingCommunity 2.9.3__py3-none-any.whl → 2.9.5__py3-none-any.whl

StreamingCommunity/Api/Site/altadefinizione/series.py

@@ -0,0 +1,209 @@
+# 16.03.25
+
+import os
+from typing import Tuple
+
+
+# External library
+from rich.console import Console
+from rich.prompt import Prompt
+
+
+# Internal utilities
+from StreamingCommunity.Util.message import start_message
+from StreamingCommunity.Lib.Downloader import HLS_Downloader
+from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance, TelegramSession
+
+
+# Logic class
+from .util.ScrapeSerie import GetSerieInfo
+from StreamingCommunity.Api.Template.Util import (
+    manage_selection, 
+    map_episode_title, 
+    dynamic_format_number, 
+    validate_selection, 
+    validate_episode_selection, 
+    display_episodes_list
+)
+from StreamingCommunity.Api.Template.config_loader import site_constant
+from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
+
+
+# Player
+from StreamingCommunity.Api.Player.supervideo import VideoSource
+
+
+# Variable
+msg = Prompt()
+console = Console()
+
+
+def download_video(index_season_selected: int, index_episode_selected: int, scrape_serie: GetSerieInfo) -> 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(str(index_season_selected))
+
+    # Get info about episode
+    obj_episode = scrape_serie.seasons_manager.get_season_by_number(int(index_season_selected)).episodes.get(index_episode_selected-1)
+    console.print(f"[yellow]Download: [red]{index_season_selected}:{index_episode_selected} {obj_episode.name}")
+    print()
+
+    if site_constant.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 = TelegramSession.get_session()
+    if script_id != "unknown":
+        TelegramSession.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(site_constant.SERIES_FOLDER, scrape_serie.series_name, f"S{index_season_selected}")
+
+    # Retrieve scws and if available master playlist
+    video_source = VideoSource(obj_episode.url)
+    video_source.make_request(obj_episode.url)
+    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 r_proc['error'] is not None:
+        try: os.remove(r_proc['path'])
+        except: pass
+
+    return r_proc['path'], r_proc['stopped']
+    
+
+def download_episode(index_season_selected: int, scrape_serie: GetSerieInfo, 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.
+    """
+    start_message()
+    obj_episodes = scrape_serie.seasons_manager.get_season_by_number(index_season_selected).episodes
+    episodes_count = len(obj_episodes.episodes)
+
+    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)
+
+            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(obj_episodes.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 if not stopped
+        for i_episode in list_episode_select:
+            path, stopped = download_video(index_season_selected, i_episode, scrape_serie)
+
+            if stopped:
+                break
+
+def download_series(select_season: MediaItem) -> None:
+    """
+    Download episodes of a TV series based on user selection.
+
+    Parameters:
+        - select_season (MediaItem): Selected media item (TV series).
+    """
+    if site_constant.TELEGRAM_BOT:
+        bot = get_bot_instance()
+
+    start_message()
+
+    # Init class
+    scrape_serie = GetSerieInfo(select_season.url)
+
+    # Collect information about seasons
+    scrape_serie.collect_season()
+    seasons_count = len(scrape_serie.seasons_manager)
+
+    # Prompt user for season selection and download episodes
+    console.print(f"\n[green]Seasons found: [red]{seasons_count}")
+
+    if site_constant.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",
+            "Menu di selezione delle stagioni\n\n"
+            "- Inserisci il numero della stagione (ad esempio, 1)\n"
+            "- Inserisci * per scaricare tutte le stagioni\n"
+            "- Inserisci un intervallo di stagioni (ad esempio, 1-2) per scaricare da una stagione all'altra\n"
+            "- Inserisci (ad esempio, 3-*) per scaricare dalla stagione specificata fino alla fine della serie",
+            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, download_all=True)
+        else:
+
+            # Otherwise, let the user select specific episodes for the single season
+            download_episode(i_season, scrape_serie, download_all=False)
+
+    if site_constant.TELEGRAM_BOT:
+        bot.send_message(f"Finito di scaricare tutte le serie e episodi", None)
+
+        # Get script_id
+        script_id = TelegramSession.get_session()
+        if script_id != "unknown":
+            TelegramSession.deleteScriptId(script_id)

StreamingCommunity/Api/Site/altadefinizione/site.py

@@ -0,0 +1,93 @@
+# 16.03.25
+
+
+# External libraries
+import httpx
+from bs4 import BeautifulSoup
+from rich.console import Console
+
+
+# Internal utilities
+from StreamingCommunity.Util.config_json import config_manager
+from StreamingCommunity.Util.headers import get_userAgent
+from StreamingCommunity.Util.table import TVShowManager
+from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
+
+
+# Logic class
+from StreamingCommunity.Api.Template.config_loader import site_constant
+from StreamingCommunity.Api.Template.Class.SearchType import MediaManager
+
+
+# Variable
+console = Console()
+media_search_manager = MediaManager()
+table_show_manager = TVShowManager()
+max_timeout = config_manager.get_int("REQUESTS", "timeout")
+
+
+def title_search(title_search: str) -> int:
+    """
+    Search for titles based on a search query.
+      
+    Parameters:
+        - title_search (str): The title to search for.
+
+    Returns:
+        int: The number of titles found.
+    """
+    if site_constant.TELEGRAM_BOT:
+        bot = get_bot_instance()
+
+    media_search_manager.clear()
+    table_show_manager.clear()
+
+    search_url = f"{site_constant.FULL_URL}/?story={title_search}&do=search&subaction=search"
+    console.print(f"[cyan]Search url: [yellow]{search_url}")
+
+    try:
+        response = httpx.post(search_url, headers={'user-agent': get_userAgent()}, timeout=max_timeout, follow_redirects=True)
+        response.raise_for_status()
+
+    except Exception as e:
+        console.print(f"Site: {site_constant.SITE_NAME}, request search error: {e}")
+        if site_constant.TELEGRAM_BOT:
+            bot.send_message(f"ERRORE\n\nErrore nella richiesta di ricerca:\n\n{e}", None)
+        return 0
+
+    # Prepara le scelte per l'utente
+    if site_constant.TELEGRAM_BOT:
+        choices = []
+
+    # Create soup istance
+    soup = BeautifulSoup(response.text, "html.parser")
+
+    # Collect data from soup
+    for i, movie_div in enumerate(soup.find_all("div", class_="movie")):
+
+        title_tag = movie_div.find("h2", class_="movie-title")
+        title = title_tag.find("a").get_text(strip=True)
+        url = title_tag.find("a").get("href")
+
+        # Define typo
+        if "/serie-tv/" in url:
+            tipo = "tv"
+        else:
+            tipo = "film"
+
+        media_search_manager.add_media({
+            'url': url,
+            'name': title,
+            'type': tipo
+        })
+
+        if site_constant.TELEGRAM_BOT:
+            choice_text = f"{i} - {title} ({tipo})"
+            choices.append(choice_text)
+
+    if site_constant.TELEGRAM_BOT:
+        if choices:
+            bot.send_message(f"Lista dei risultati:", choices)
+	
+    # Return the number of titles found
+    return media_search_manager.get_length()

StreamingCommunity/Api/Site/altadefinizione/util/ScrapeSerie.py

@@ -0,0 +1,72 @@
+# 16.03.25
+
+# External libraries
+import httpx
+from bs4 import BeautifulSoup
+
+
+# Internal utilities
+from StreamingCommunity.Util.headers import get_userAgent
+from StreamingCommunity.Util.config_json import config_manager
+from StreamingCommunity.Api.Player.Helper.Vixcloud.util import SeasonManager
+
+
+# Variable
+max_timeout = config_manager.get_int("REQUESTS", "timeout")
+
+
+
+class GetSerieInfo:
+    def __init__(self, url):
+        """
+        Initialize the GetSerieInfo class for scraping TV series information.
+        
+        Args:
+            - url (str): The URL of the streaming site.
+        """
+        self.headers = {'user-agent': get_userAgent()}
+        self.url = url
+        self.seasons_manager = SeasonManager()
+
+    def collect_season(self) -> None:
+        """
+        Retrieve all episodes for all seasons
+        """
+        response = httpx.get(self.url, headers=self.headers)
+        soup = BeautifulSoup(response.text, "html.parser")
+        self.series_name = soup.find("title").get_text(strip=True).split(" - ")[0]
+
+        # Process all seasons
+        season_items = soup.find_all('div', class_='accordion-item')
+    
+        for season_idx, season_item in enumerate(season_items, 1):
+            season_header = season_item.find('div', class_='accordion-header')
+            if not season_header:
+                continue
+                
+            season_name = season_header.get_text(strip=True)
+            
+            # Create a new season and get a reference to it
+            current_season = self.seasons_manager.add_season({
+                'number': season_idx, 
+                'name': season_name
+            })
+            
+            # Find episodes for this season
+            episode_divs = season_item.find_all('div', class_='down-episode')
+            for ep_idx, ep_div in enumerate(episode_divs, 1):
+                episode_name_tag = ep_div.find('b')
+                if not episode_name_tag:
+                    continue
+                    
+                episode_name = episode_name_tag.get_text(strip=True)
+                link_tag = ep_div.find('a', string=lambda text: text and "Supervideo" in text)
+                episode_url = link_tag['href'] if link_tag else None
+                
+                # Add episode to the season
+                if current_season:
+                    current_season.episodes.add({
+                        'number': ep_idx,
+                        'name': episode_name,
+                        'url': episode_url
+                    })

StreamingCommunity/Api/Site/animeunity/__init__.py

@@ -11,11 +11,12 @@ from rich.prompt import Prompt
 
 # Internal utilities
 from StreamingCommunity.Api.Template import get_select_title
+from StreamingCommunity.Api.Template.config_loader import site_constant
+from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
 from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
 
 
 # Logic class
-from StreamingCommunity.Api.Template.config_loader import site_constant
 from .site import title_search, media_search_manager, table_show_manager
 from .film_serie import download_film, download_series
 
@@ -31,53 +32,73 @@ msg = Prompt()
 console = Console()
 
 
-def search(string_to_search: str = None, get_onylDatabase: bool = False):
-
-    if site_constant.TELEGRAM_BOT:
-        bot = get_bot_instance()
-
-        if string_to_search is None:
-
-            # Chiedi la scelta all'utente con il bot Telegram
+def get_user_input(string_to_search: str = None):
+    """
+    Asks the user to input a search term.
+    Handles both Telegram bot input and direct input.
+    """
+    if string_to_search is None:
+        if site_constant.TELEGRAM_BOT:
+            bot = get_bot_instance()
             string_to_search = bot.ask(
                 "key_search",
-                f"Inserisci la parola da cercare\noppure back per tornare alla scelta: ",
+                f"Enter the search term\nor type 'back' to return to the menu: ",
                 None
             )
 
             if string_to_search == 'back':
-                # Riavvia lo script
-                # Chiude il processo attuale e avvia una nuova istanza dello script
+
+                # Restart the script
                 subprocess.Popen([sys.executable] + sys.argv)
                 sys.exit()
-
-    else:
-        if string_to_search is None:
-            string_to_search = msg.ask(f"\n[purple]Insert word to search in [green]{site_constant.SITE_NAME}").strip() 
-
-    # Search on database
+        else:
+            string_to_search = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
+
+    return string_to_search
+
+def process_search_result(select_title):
+    """
+    Handles the search result and initiates the download for either a film or series.
+    """
+    download_series(select_title)
+
+def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_item: dict = None):
+    """
+    Main function of the application for search film, series and anime.
+
+    Parameters:
+        string_to_search (str, optional): String to search for
+        get_onlyDatabase (bool, optional): If True, return only the database object
+        direct_item (dict, optional): Direct item to process (bypass search)
+    """
+    if direct_item:
+        select_title = MediaItem(**direct_item)
+        process_search_result(select_title)
+        return
+
+    # Get the user input for the search term
+    string_to_search = get_user_input(string_to_search)
+
+    # Perform the database search
     len_database = title_search(string_to_search)
 
-    # Return list of elements
-    if get_onylDatabase:
+    ##If only the database is needed, return the manager
+    if get_onlyDatabase:
         return media_search_manager
+    
+    if site_constant.TELEGRAM_BOT:
+        bot = get_bot_instance()
 
     if len_database > 0:
-
-        # Select title from list (type: TV \ Movie \ OVA)
         select_title = get_select_title(table_show_manager, media_search_manager)
+        process_search_result(select_title)
 
-        if select_title.type == 'Movie' or select_title.type == 'OVA':
-            download_film(select_title)
-
-        else:
-            download_series(select_title)
-            
     else:
-        if site_constant.TELEGRAM_BOT:
-            bot.send_message(f"Nessun risultato trovato riprova", None)
-          
         console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
 
-        # Retry
+        if site_constant.TELEGRAM_BOT:
+            bot.send_message(f"No results found, please try again", None)
+
+        # If no results are found, ask again
+        string_to_search = get_user_input()
         search()

StreamingCommunity/Api/Site/animeunity/film_serie.py

@@ -123,7 +123,11 @@ def download_series(select_title: MediaItem):
 
         last_command = bot.ask(
             "select_title",
-            f"Inserisci l'indice del media o * per scaricare tutti i media, oppure 1-2 o 3-* per un intervallo di media.",
+            "Menu di selezione degli episodi: \n\n" 
+            "- Inserisci il numero dell'episodio (ad esempio, 1)\n" 
+            "- Inserisci * per scaricare tutti gli episodi\n" 
+            "- Inserisci un intervallo di episodi (ad esempio, 1-2) per scaricare da un episodio all'altro\n" 
+            "- Inserisci (ad esempio, 3-*) per scaricare dall'episodio specificato fino alla fine della serie",
             None
         )
 
@@ -137,7 +141,7 @@ def download_series(select_title: MediaItem):
 
     # Download selected episodes
     if len(list_episode_select) == 1 and last_command != "*":
-        path, _ = download_episode(list_episode_select[0]-1, scrape_serie, video_source)[0]
+        path, _ = download_episode(list_episode_select[0]-1, scrape_serie, video_source)
         return path
 
     # Download all other episodes selecter

StreamingCommunity/Api/Site/animeunity/site.py

@@ -19,7 +19,6 @@ from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
 
 # Logic class
 from StreamingCommunity.Api.Template.config_loader import site_constant
-from StreamingCommunity.Api.Template.Util import search_domain
 from StreamingCommunity.Api.Template.Class.SearchType import MediaManager
 
 
@@ -78,10 +77,8 @@ def get_real_title(record):
     """
     if record['title_eng'] is not None:
         return record['title_eng']
-    
     elif record['title'] is not None:
         return record['title']
-    
     else:
         return record['title_it']
 
@@ -102,14 +99,6 @@ def title_search(title: str) -> int:
     media_search_manager.clear()
     table_show_manager.clear()
 
-    # Check if domain is working
-    domain_to_use, base_url = search_domain(site_constant.FULL_URL)
-
-    if domain_to_use is None or base_url is None:
-        console.print("[bold red]Error: Unable to determine valid domain or base URL.[/bold red]")
-        console.print("[yellow]The service might be temporarily unavailable or the domain may have changed.[/yellow]")
-        sys.exit(1)
-
     # Create parameter for request
     data = get_token()
     cookies = {'animeunity_session': data.get('animeunity_session')}
@@ -132,6 +121,7 @@ def title_search(title: str) -> int:
 
     except Exception as e:
         console.print(f"Site: {site_constant.SITE_NAME}, request search error: {e}")
+        return 0
 
     # Inizializza la lista delle scelte
     if site_constant.TELEGRAM_BOT:

StreamingCommunity/Api/Site/cb01new/__init__.py

@@ -10,10 +10,11 @@ from rich.prompt import Prompt
 
 # Internal utilities
 from StreamingCommunity.Api.Template import get_select_title
+from StreamingCommunity.Api.Template.config_loader import site_constant
+from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
 
 
 # Logic class
-from StreamingCommunity.Api.Template.config_loader import site_constant
 from .site import title_search, media_search_manager, table_show_manager
 from .film import download_film
 
@@ -29,10 +30,26 @@ msg = Prompt()
 console = Console()
 
 
-def search(string_to_search: str = None, get_onylDatabase: bool = False):
+def process_search_result(select_title):
+    """
+    Handles the search result and initiates the download for either a film or series.
+    """
+    # !!! ADD TYPE DONT WORK FOR SERIE
+    download_film(select_title)
+
+def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_item: dict = None):
     """
-    Main function of the application for film and series.
+    Main function of the application for search film, series and anime.
+
+    Parameters:
+        string_to_search (str, optional): String to search for
+        get_onylDatabase (bool, optional): If True, return only the database object
+        direct_item (dict, optional): Direct item to process (bypass search)
     """
+    if direct_item:
+        select_title = MediaItem(**direct_item)
+        process_search_result(select_title)
+        return
 
     if string_to_search is None:
         string_to_search = msg.ask(f"\n[purple]Insert word to search in [green]{site_constant.SITE_NAME}").strip()
@@ -40,21 +57,16 @@ def search(string_to_search: str = None, get_onylDatabase: bool = False):
     # Search on database
     len_database = title_search(quote_plus(string_to_search))
 
-    # Return list of elements
-    if get_onylDatabase:
+    ## If only the database is needed, return the manager
+    if get_onlyDatabase:
         return media_search_manager
     
     if len_database > 0:
-
-        # Select title from list
         select_title = get_select_title(table_show_manager, media_search_manager)
-
-        # !!! ADD TYPE DONT WORK FOR SERIE
-        download_film(select_title)
-
-
+        process_search_result(select_title)
+        
     else:
-        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
 
-        # Retry
+        # If no results are found, ask again
+        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
         search()

StreamingCommunity/Api/Site/cb01new/site.py

@@ -17,7 +17,6 @@ from StreamingCommunity.Util.table import TVShowManager
 
 # Logic class
 from StreamingCommunity.Api.Template.config_loader import site_constant
-from StreamingCommunity.Api.Template.Util import search_domain
 from StreamingCommunity.Api.Template.Class.SearchType import MediaManager
 
 
@@ -41,14 +40,6 @@ def title_search(word_to_search: str) -> int:
     media_search_manager.clear()
     table_show_manager.clear()
 
-    # Check if domain is working
-    domain_to_use, base_url = search_domain(site_constant.FULL_URL)
-
-    if domain_to_use is None or base_url is None:
-        console.print("[bold red]Error: Unable to determine valid domain or base URL.[/bold red]")
-        console.print("[yellow]The service might be temporarily unavailable or the domain may have changed.[/yellow]")
-        sys.exit(1)
-
     search_url = f"{site_constant.FULL_URL}/?s={word_to_search}"
     console.print(f"[cyan]Search url: [yellow]{search_url}")
 
@@ -58,21 +49,21 @@ def title_search(word_to_search: str) -> int:
 
     except Exception as e:
         console.print(f"Site: {site_constant.SITE_NAME}, request search error: {e}")
+        return 0
 
     # Create soup and find table
     soup = BeautifulSoup(response.text, "html.parser")
 
-    for div in soup.find_all("div", class_ = "card-content"):
+    for card in soup.find_all("div", class_=["card", "mp-post", "horizontal"]):
         try:
-
-            url = div.find("h3").find("a").get("href")
-            title = div.find("h3").find("a").get_text(strip=True)
-            desc = div.find("p").find("strong").text
+            title_tag = card.find("h3", class_="card-title").find("a")
+            url = title_tag.get("href")
+            title = title_tag.get_text(strip=True)
 
             title_info = {
                 'name': title,
-                'desc': desc,
-                'url': url
+                'url': url,
+                'type': 'film'
             }
 
             media_search_manager.add_media(title_info)