StreamingCommunity 2.9.4__py3-none-any.whl → 2.9.6__py3-none-any.whl

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

@@ -55,8 +55,7 @@ def download_episode(index_select: int, scrape_serie: ScrapeSerieAnime, video_so
     if obj_episode is not None:
 
         start_message()
-        console.print(f"[yellow]Download:  [red]EP_{obj_episode.number} \n")
-        console.print("[cyan]You can safely stop the download with [bold]Ctrl+c[bold] [cyan] \n")
+        console.print(f"[bold yellow]Download:[/bold yellow] [red]{site_constant.SITE_NAME}[/red] ([cyan]E{obj_episode.number}[/cyan]) \n")
 
         if site_constant.TELEGRAM_BOT:
             bot.send_message(f"Download in corso:\nTitolo:{scrape_serie.series_name}\nEpisodio: {obj_episode.number}", None)
@@ -70,7 +69,7 @@ def download_episode(index_select: int, scrape_serie: ScrapeSerieAnime, video_so
         video_source.get_embed(obj_episode.id)
 
         # Create output path
-        title_name = f"{scrape_serie.series_name}_EP_{dynamic_format_number(str(obj_episode.number))}.mp4"
+        mp4_name = f"{scrape_serie.series_name}_EP_{dynamic_format_number(str(obj_episode.number))}.mp4"
 
         if scrape_serie.is_series:
             mp4_path = os_manager.get_sanitize_path(os.path.join(site_constant.ANIME_FOLDER, scrape_serie.series_name))
@@ -84,7 +83,7 @@ def download_episode(index_select: int, scrape_serie: ScrapeSerieAnime, video_so
         # Start downloading
         path, kill_handler = MP4_downloader(
             url=str(video_source.src_mp4).strip(),
-            path=os.path.join(mp4_path, title_name)
+            path=os.path.join(mp4_path, mp4_name)
         )
 
         return path, kill_handler
@@ -123,7 +122,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
         )
 

StreamingCommunity/Api/Site/animeunity/site.py

@@ -77,20 +77,18 @@ 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']
 
 
-def title_search(title: str) -> int:
+def title_search(query: str) -> int:
     """
-    Function to perform an anime search using a provided title.
+    Function to perform an anime search using a provided query.
 
     Parameters:
-        - title_search (str): The title to search for.
+        - query (str): The query to search for.
 
     Returns:
         - int: A number containing the length of media search manager.
@@ -108,7 +106,7 @@ def title_search(title: str) -> int:
         'user-agent': get_userAgent(),
         'x-csrf-token': data.get('csrf_token')
     }
-    json_data =  {'title': title}
+    json_data =  {'title': query}
 
     # Send a POST request to the API endpoint for live search
     try:

StreamingCommunity/Api/Site/animeworld/__init__.py

@@ -0,0 +1,71 @@
+# 21.03.25
+
+# External library
+from rich.console import Console
+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 .site import title_search, media_search_manager, table_show_manager
+from .serie import download_series
+
+
+# Variable
+indice = 8
+_useFor = "anime"
+_deprecate = False
+_priority = 2
+_engineDownload = "mp4"
+
+msg = Prompt()
+console = Console()
+
+
+
+def process_search_result(select_title):
+    """
+    Handles the search result and initiates the download for either a film or series.
+    """
+    if select_title.type == "TV":
+        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 = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
+
+    # Perform the database search
+    len_database = title_search(string_to_search)
+
+    ##If only the database is needed, return the manager
+    if get_onlyDatabase:
+        return media_search_manager
+
+    if len_database > 0:
+        select_title = get_select_title(table_show_manager, media_search_manager)
+        process_search_result(select_title)
+
+    else:
+        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
+
+        # If no results are found, ask again
+        string_to_search = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
+        search()

StreamingCommunity/Api/Site/animeworld/serie.py

@@ -0,0 +1,107 @@
+# 11.03.24
+
+import os
+import logging
+from typing import Tuple
+
+
+# External library
+from rich.console import Console
+from rich.prompt import Prompt
+
+
+# Internal utilities
+from StreamingCommunity.Util.os import os_manager
+from StreamingCommunity.Util.message import start_message
+from StreamingCommunity.Lib.Downloader import MP4_downloader
+
+
+# Logic class
+from .util.ScrapeSerie import ScrapSerie
+from StreamingCommunity.Api.Template.config_loader import site_constant
+from StreamingCommunity.Api.Template.Util import manage_selection, dynamic_format_number, map_episode_title
+from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
+
+
+# Player
+from StreamingCommunity.Api.Player.sweetpixel import AnimeWorldPlayer
+
+
+# Variable
+console = Console()
+msg = Prompt()
+KILL_HANDLER = bool(False)
+
+
+
+def download_episode(index_select: int, scrape_serie: ScrapSerie, episodes) -> Tuple[str,bool]:
+    """
+    Downloads the selected episode.
+
+    Parameters:
+        - index_select (int): Index of the episode to download.
+
+    Return:
+        - str: output path
+        - bool: kill handler status
+    """
+    start_message()
+
+    # Get information about the selected episode
+    console.print(f"[bold yellow]Download:[/bold yellow] [red]{site_constant.SITE_NAME}[/red] ([cyan]E{index_select+1}[/cyan]) \n")
+
+    # Define filename and path for the downloaded video
+    mp4_name = f"{scrape_serie.get_name()}_EP_{dynamic_format_number(str(index_select+1))}.mp4"
+    mp4_path = os.path.join(site_constant.ANIME_FOLDER, scrape_serie.get_name())
+
+    # Create output folder
+    os_manager.create_path(mp4_path)
+
+    # Collect mp4 link
+    video_source = AnimeWorldPlayer(site_constant.FULL_URL, episodes[index_select], scrape_serie.session_id, scrape_serie.csrf_token)
+    mp4_link = video_source.get_download_link()
+
+    # Start downloading
+    path, kill_handler = MP4_downloader(
+        url=str(mp4_link).strip(),
+        path=os.path.join(mp4_path, mp4_name)
+    )
+
+    return path, kill_handler
+
+
+def download_series(select_title: MediaItem):
+    """
+    Function to download episodes of a TV series.
+
+    Parameters:
+        - tv_id (int): The ID of the TV series.
+        - tv_name (str): The name of the TV series.
+    """
+    start_message()
+
+    scrape_serie = ScrapSerie(select_title.url, site_constant.FULL_URL)
+
+    # Get the count of episodes for the TV series
+    episodes = scrape_serie.get_episodes()
+    episoded_count = len(episodes)
+    console.print(f"[cyan]Episodes find: [red]{episoded_count}")
+
+    # Prompt user to select an episode index
+    last_command = msg.ask("\n[cyan]Insert media [red]index [yellow]or [red]* [cyan]to download all media [yellow]or [red]1-2 [cyan]or [red]3-* [cyan]for a range of media")
+
+    # Manage user selection
+    list_episode_select = manage_selection(last_command, episoded_count)
+
+    # Download selected episodes
+    if len(list_episode_select) == 1 and last_command != "*":
+        path, _ = download_episode(list_episode_select[0]-1, scrape_serie, episodes)
+        return path
+
+    # Download all other episodes selecter
+    else:
+        kill_handler = False
+        for i_episode in list_episode_select:
+            if kill_handler:
+                break
+            _, kill_handler = download_episode(i_episode-1, scrape_serie, episodes)

StreamingCommunity/Api/Site/animeworld/site.py

@@ -0,0 +1,111 @@
+# 21.03.25
+
+import logging
+
+# 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, get_headers
+from StreamingCommunity.Util.table import TVShowManager
+
+
+# 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 get_session_and_csrf() -> dict:
+    """
+    Get the session ID and CSRF token from the website's cookies and HTML meta data.
+    """
+    # Send an initial GET request to the website
+    response = httpx.get(site_constant.FULL_URL, headers=get_headers())
+
+    # Extract the sessionId from the cookies
+    session_id = response.cookies.get('sessionId')
+    logging.info(f"Session ID: {session_id}")
+
+    # Use BeautifulSoup to parse the HTML and extract the CSRF-Token
+    soup = BeautifulSoup(response.text, 'html.parser')
+
+    # Try to find the CSRF token in a meta tag or hidden input
+    csrf_token = None
+    meta_tag = soup.find('meta', {'name': 'csrf-token'})
+    if meta_tag:
+        csrf_token = meta_tag.get('content')
+
+    # If it's not in the meta tag, check for hidden input fields
+    if not csrf_token:
+        input_tag = soup.find('input', {'name': '_csrf'})
+        if input_tag:
+            csrf_token = input_tag.get('value')
+
+    logging.info(f"CSRF Token: {csrf_token}")
+    return session_id, csrf_token
+
+def title_search(query: str) -> int:
+    """
+    Function to perform an anime search using a provided title.
+
+    Parameters:
+        - query (str): The query to search for.
+
+    Returns:
+        - int: A number containing the length of media search manager.
+    """
+    search_url = f"{site_constant.FULL_URL}/search?keyword={query}"
+    console.print(f"[cyan]Search url: [yellow]{search_url}")
+
+    # Make the GET request
+    try:
+        response = httpx.get(search_url, headers={'User-Agent': get_userAgent()})
+
+    except Exception as e:
+        console.print(f"Site: {site_constant.SITE_NAME}, request search error: {e}")
+        return 0
+
+    # Create soup istance
+    soup = BeautifulSoup(response.text, 'html.parser')
+
+    # Collect data from soup
+    for element in soup.find_all('a', class_='poster'):
+        try:
+            title = element.find('img').get('alt')
+            url = f"{site_constant.FULL_URL}{element.get('href')}"
+            status_div = element.find('div', class_='status')
+            is_dubbed = False
+            anime_type = 'TV'
+
+            if status_div:
+                if status_div.find('div', class_='dub'):
+                    is_dubbed = True
+                
+                if status_div.find('div', class_='movie'):
+                    anime_type = 'Movie'
+                elif status_div.find('div', class_='ona'):
+                    anime_type = 'ONA'
+
+                media_search_manager.add_media({
+                    'name': title,
+                    'type': anime_type,
+                    'DUB': is_dubbed,
+                    'url': url
+                })
+
+        except Exception as e:
+            print(f"Error parsing a film entry: {e}")
+
+    # Return the length of media search manager
+    return media_search_manager.get_length()

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

@@ -0,0 +1,79 @@
+# 21.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.Util.os import os_manager
+
+
+# Player
+from ..site import get_session_and_csrf
+from StreamingCommunity.Api.Player.sweetpixel import AnimeWorldPlayer
+
+
+# Variable
+max_timeout = config_manager.get_int("REQUESTS", "timeout")
+
+
+class ScrapSerie:
+    def __init__(self, url, full_url):
+        """Initialize the ScrapSerie object with the provided URL and setup the HTTP client."""
+        self.url = url
+        self.link = httpx.URL(url).path 
+        self.session_id, self.csrf_token = get_session_and_csrf()
+        self.client = httpx.Client(
+            cookies={"sessionId": self.session_id},
+            headers={"User-Agent": get_userAgent(), "csrf-token": self.csrf_token},
+            base_url=full_url
+        )
+
+        try:
+            self.response = self.client.get(self.link, timeout=max_timeout, follow_redirects=True)
+            self.response.raise_for_status()
+
+        except:
+            raise Exception(f"Failed to retrieve anime page.")
+
+
+    def get_name(self):
+        """Extract and return the name of the anime series."""
+        soup = BeautifulSoup(self.response.content, "html.parser")
+        return os_manager.get_sanitize_file(soup.find("h1", {"id": "anime-title"}).get_text(strip=True))
+    
+    def get_episodes(self, nums=None):
+        """Fetch and return the list of episodes, optionally filtering by specific episode numbers."""
+        soup = BeautifulSoup(self.response.content, "html.parser")
+
+        raw_eps = {}
+        for data in soup.select('li.episode > a'):
+            epNum = data.get('data-episode-num')
+            epID = data.get('data-episode-id')
+
+            if nums and epNum not in nums:
+                continue
+
+            if epID not in raw_eps:
+                raw_eps[epID] = {
+                    'number': epNum,
+                    'link': f"/api/download/{epID}"
+                }
+
+        episodes = [episode_data for episode_data in raw_eps.values()]
+        return episodes
+
+    def get_episode(self, index):
+        """Fetch a specific episode based on the index, and return an AnimeWorldPlayer instance."""
+        episodes = self.get_episodes()
+        
+        if 0 <= index < len(episodes):
+            episode_data = episodes[index]
+            return AnimeWorldPlayer(episode_data, self.session_id, self.csrf_token)
+        
+        else:
+            raise IndexError("Episode index out of range")

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/film.py

@@ -37,7 +37,7 @@ def download_film(select_title: MediaItem) -> str:
         - str: output path
     """
     start_message()
-    console.print(f"[yellow]Download:  [red]{select_title.name} \n")
+    console.print(f"[bold yellow]Download:[/bold yellow] [red]{site_constant.SITE_NAME}[/red] → [cyan]{select_title.name}[/cyan] \n")
 
     # Setup api manger
     video_source = VideoSource(select_title.url)