StreamingCommunity 3.2.9__py3-none-any.whl → 3.3.1__py3-none-any.whl

StreamingCommunity/Api/Site/guardaserie/__init__.py

@@ -1,5 +1,7 @@
 # 09.06.24
 
+import sys
+import subprocess
 from urllib.parse import quote_plus
 
 
@@ -12,6 +14,7 @@ from rich.prompt import Prompt
 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
@@ -30,6 +33,44 @@ msg = Prompt()
 console = Console()
 
 
+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 provided, it's returned directly (after stripping).
+    """
+    if string_to_search is not None:
+        return string_to_search.strip()
+
+    if site_constant.TELEGRAM_BOT:
+        bot = get_bot_instance()
+        user_response = bot.ask(
+            "key_search", # Request type
+            "Enter the search term\nor type 'back' to return to the menu: ",
+            None
+        )
+
+        if user_response is None:
+            bot.send_message("Timeout: No search term entered.", None)
+            return None
+
+        if user_response.lower() == 'back':
+            bot.send_message("Returning to the main menu...", None)
+            
+            try:
+                # Restart the script
+                subprocess.Popen([sys.executable] + sys.argv)
+                sys.exit()
+                
+            except Exception as e:
+                bot.send_message(f"Error during restart attempt: {e}", None)
+                return None # Return None if restart fails
+        
+        return user_response.strip()
+        
+    else:
+        return msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
+    
 def process_search_result(select_title, selections=None):
     """
     Handles the search result and initiates the download for either a film or series.
@@ -38,7 +79,18 @@ def process_search_result(select_title, selections=None):
         select_title (MediaItem): The selected media item
         selections (dict, optional): Dictionary containing selection inputs that bypass manual input
                                     {'season': season_selection, 'episode': episode_selection}
+
+    Returns:
+        bool: True if processing was successful, False otherwise
     """
+    if not select_title:
+        if site_constant.TELEGRAM_BOT:
+            bot = get_bot_instance()
+            bot.send_message("No title selected or selection cancelled.", None)
+        else:
+            console.print("[yellow]No title selected or selection cancelled.")
+        return False
+    
     season_selection = None
     episode_selection = None
     
@@ -47,6 +99,7 @@ def process_search_result(select_title, selections=None):
         episode_selection = selections.get('episode')
         
     download_series(select_title, season_selection, episode_selection)
+    return True
 
 def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_item: dict = None, selections: dict = None):
     """
@@ -59,27 +112,44 @@ def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_
         selections (dict, optional): Dictionary containing selection inputs that bypass manual input
                                     {'season': season_selection, 'episode': episode_selection}
     """
+    bot = None
+    if site_constant.TELEGRAM_BOT:
+        try:
+            from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
+            bot = get_bot_instance()
+        except Exception:
+            bot = None
+
     if direct_item:
         select_title = MediaItem(**direct_item)
         process_search_result(select_title, selections)
+        return True
+
+    # Get the user input for the search term
+    actual_search_query = get_user_input(string_to_search)
+
+    # Handle empty input
+    if not actual_search_query:
+        if bot:
+            if actual_search_query is None:
+                bot.send_message("Search term not provided or operation cancelled. Returning.", None)
         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()
-    
-    # Search on database
-    len_database = title_search(quote_plus(string_to_search))
+    # Search on database (preserve quote_plus usage)
+    len_database = title_search(quote_plus(actual_search_query))
 
     # 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,len_database)
+        select_title = get_select_title(table_show_manager, media_search_manager, len_database)
         process_search_result(select_title, selections)
+        return True
 
     else:
-
-        # If no results are found, ask again
-        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
-        search()
+        if bot:
+            bot.send_message(f"No results found for: '{actual_search_query}'", None)
+        else:
+            console.print(f"\n[red]Nothing matching was found for[white]: [purple]{actual_search_query}")
+        return

StreamingCommunity/Api/Site/mediasetinfinity/__init__.py

@@ -1,5 +1,8 @@
 # 21.05.24
 
+import sys
+import subprocess
+
 
 # External library
 from rich.console import Console
@@ -10,6 +13,7 @@ from rich.prompt import Prompt
 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
@@ -32,11 +36,40 @@ console = Console()
 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 provided, it's returned directly (after stripping).
     """
-    if string_to_search is None:
-        string_to_search = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
-
-    return msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
+    if string_to_search is not None:
+        return string_to_search.strip()
+
+    if site_constant.TELEGRAM_BOT:
+        bot = get_bot_instance()
+        user_response = bot.ask(
+            "key_search", # Request type
+            "Enter the search term\nor type 'back' to return to the menu: ",
+            None
+        )
+
+        if user_response is None:
+            bot.send_message("Timeout: No search term entered.", None)
+            return None
+
+        if user_response.lower() == 'back':
+            bot.send_message("Returning to the main menu...", None)
+            
+            try:
+                # Restart the script
+                subprocess.Popen([sys.executable] + sys.argv)
+                sys.exit()
+                
+            except Exception as e:
+                bot.send_message(f"Error during restart attempt: {e}", None)
+                return None # Return None if restart fails
+        
+        return user_response.strip()
+        
+    else:
+        return msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
 
 def process_search_result(select_title, selections=None):
     """
@@ -46,7 +79,18 @@ def process_search_result(select_title, selections=None):
         select_title (MediaItem): The selected media item
         selections (dict, optional): Dictionary containing selection inputs that bypass manual input
                                     {'season': season_selection, 'episode': episode_selection}
+
+    Returns:
+        bool: True if processing was successful, False otherwise
     """
+    if not select_title:
+        if site_constant.TELEGRAM_BOT:
+            bot = get_bot_instance()
+            bot.send_message("No title selected or selection cancelled.", None)
+        else:
+            console.print("[yellow]No title selected or selection cancelled.")
+        return False
+    
     if select_title.type == 'tv':
         season_selection = None
         episode_selection = None
@@ -56,9 +100,11 @@ def process_search_result(select_title, selections=None):
             episode_selection = selections.get('episode')
 
         download_series(select_title, season_selection, episode_selection)
+        return True
 
     else:
         download_film(select_title)
+        return True
 
 def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_item: dict = None, selections: dict = None):
     """
@@ -71,26 +117,43 @@ def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_
         selections (dict, optional): Dictionary containing selection inputs that bypass manual input
                                     {'season': season_selection, 'episode': episode_selection}
     """
+    bot = None
+    if site_constant.TELEGRAM_BOT:
+        bot = get_bot_instance()
+    
     if direct_item:
         select_title = MediaItem(**direct_item)
         process_search_result(select_title, selections)
+        return True
+    
+    # Get the user input for the search term
+    actual_search_query = get_user_input(string_to_search)
+
+    # Handle cases where user input is empty, or 'back' was handled (sys.exit or None return)
+    if not actual_search_query:
+        if bot:
+            if actual_search_query is None: # Specifically for timeout from bot.ask or failed restart
+                bot.send_message("Search term not provided or operation cancelled. Returning.", None)
         return
-
-    if string_to_search is None:
-        string_to_search = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
-
-    # Search on database
-    len_database = title_search(string_to_search)
+    
+    # Perform the database search
+    len_database = title_search(actual_search_query)
 
     # 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,len_database)
+        select_title = get_select_title(table_show_manager, media_search_manager, len_database)
         process_search_result(select_title, selections)
+        return True
     
     else:
-        # If no results are found, ask again
-        console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
-        search()
+        if bot:
+            bot.send_message(f"No results found for: '{actual_search_query}'", None)
+        else:
+            console.print(f"\n[red]Nothing matching was found for[white]: [purple]{actual_search_query}")
+
+        # Do not call search() recursively here to avoid infinite loops on no results.
+        # The flow should return to the caller (e.g., main menu in run.py).
+        return

StreamingCommunity/Api/Site/mediasetinfinity/film.py

@@ -60,7 +60,7 @@ def download_film(select_title: MediaItem) -> Tuple[str, bool]:
         cdm_device=get_wvd_path(),
         license_url=license_url,
         mpd_url=mpd_url,
-        output_path=mp4_path,
+        output_path=os.path.join(mp4_path, title_name),
     )
     dash_process.parse_manifest(custom_headers=get_headers())
 

StreamingCommunity/Api/Site/mediasetinfinity/site.py

@@ -1,6 +1,8 @@
 # 25.07.25
 
 import os
+import sys
+
 
 # External libraries
 import httpx
@@ -44,7 +46,14 @@ def title_search(query: str) -> int:
     cdm_device_path = get_wvd_path()
     if not cdm_device_path or not isinstance(cdm_device_path, (str, bytes, os.PathLike)) or not os.path.isfile(cdm_device_path):
         console.print(f"[bold red] CDM file not found or invalid path: {cdm_device_path}[/bold red]")
-        return None
+        sys.exit(0)
+
+    # Check if beToken is present
+    if (config_manager.get_dict("SITE_LOGIN", "mediasetinfinity")["beToken"] is None or config_manager.get_dict("SITE_LOGIN", "mediasetinfinity")["beToken"] == "") and \
+            (config_manager.get_dict("SITE_LOGIN", "mediasetinfinity").get("username") is None or config_manager.get_dict("SITE_LOGIN", "mediasetinfinity").get("username") == "" \
+            or config_manager.get_dict("SITE_LOGIN", "mediasetinfinity").get("password") is None or config_manager.get_dict("SITE_LOGIN", "mediasetinfinity").get("password") == ""):
+        console.print("[bold red] beToken or credentials are missing or empty.[/bold red]")
+        sys.exit(0)
 
     search_url = 'https://api-ott-prod-fe.mediaset.net/PROD/play/reco/account/v2.0'
     console.print(f"[cyan]Search url: [yellow]{search_url}")
@@ -114,7 +123,8 @@ def title_search(query: str) -> int:
             'name': item.get('title', ''),
             'type': media_type,
             'url': page_url,
-            'image': None,
+            'image': item.get('thumbnails', {}).get('image_header_poster-768x384', '').get('url', ''),
+            'date': item.get('year', ''),
         })
 
     return media_search_manager.get_length()

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

@@ -1,6 +1,5 @@
 # 16.03.25
 
-
 import logging
 from urllib.parse import urlparse
 
@@ -16,10 +15,6 @@ from StreamingCommunity.Util.config_json import config_manager
 from StreamingCommunity.Api.Player.Helper.Vixcloud.util import SeasonManager
 
 
-# Logic class
-from .get_license import get_bearer_token
-
-
 # Variable
 max_timeout = config_manager.get_int("REQUESTS", "timeout")
 
@@ -47,7 +42,10 @@ class GetSerieInfo:
         return self.serie_id
 
     def _get_public_id(self):
-        """Ottiene il public ID tramite l'API watchlist"""
+        self.public_id = "PR1GhC"
+        return self.public_id
+        
+        """
         bearer_token = get_bearer_token()
         headers = {
             'authorization': f'Bearer {bearer_token}',
@@ -73,6 +71,7 @@ class GetSerieInfo:
         else:
             logging.error(f"Failed to get public ID: {response.status_code}")
             return None
+        """
 
     def _get_series_data(self):
         """Ottiene i dati della serie tramite l'API"""
@@ -290,4 +289,4 @@ class GetSerieInfo:
             logging.error(f"Episode index {episode_index} is out of range for season {season_number}")
             return None
             
-        return episodes[episode_index]
+        return episodes[episode_index]

StreamingCommunity/Api/Site/mediasetinfinity/util/get_license.py

@@ -1,11 +1,15 @@
 # 16.03.25
 
+import json
+import time
 from urllib.parse import urlencode
 import xml.etree.ElementTree as ET
 
 
 # External library
 import httpx
+from rich.console import Console
+from seleniumbase import Driver
 
 
 # Internal utilities
@@ -14,9 +18,143 @@ from StreamingCommunity.Util.headers import get_headers, get_userAgent
 
 
 # Variable
+console = Console()
 MAX_TIMEOUT = config_manager.get_int("REQUESTS", "timeout")
+beToken = None
+network_data = []
 
 
+def save_network_data(data):
+    """Save network data and check for beToken"""
+    global network_data
+    
+    # Filter only for login API responses
+    if data.get('method') == 'Network.responseReceived':
+        params = data.get('params', {})
+        response = params.get('response', {})
+        url = response.get('url', '')
+        
+        if "persona/login/v2.0" in url and params.get('type') != 'Preflight':
+            network_data.append(data)
+            console.print(f"[green]Found login API request to: {url}")
+
+
+def generate_betoken(username: str, password: str, sleep_action: float = 1.0) -> str:
+    driver = Driver(uc=True, uc_cdp_events=True, incognito=True, headless=True)
+    
+    try:
+        console.print("[cyan]Launching browser...")
+        be_token_holder = {"token": None}
+        global network_data
+        network_data = []  # Reset network data
+
+        # Load home page
+        console.print("[cyan]Navigating to Mediaset Play...")
+        driver.uc_open_with_reconnect("https://www.mediasetplay.mediaset.it", sleep_action)
+        
+        # Add CDP listeners after opening the page
+        driver.add_cdp_listener(
+            "Network.responseReceived",
+            lambda data: save_network_data(data)
+        )
+        driver.sleep(sleep_action)
+
+        # Accept privacy policy if present
+        try:
+            driver.click("#rti-privacy-accept-btn-screen1-id", timeout=3)
+        except Exception:
+            pass
+
+        # Click Login using the specific div structure
+        console.print("[cyan]Clicking login button...")
+        try:
+            driver.click('div.dwv_v span:contains("Login")', timeout=5)
+        except Exception as e:
+            print(f"Error clicking login: {e}")
+            try:
+                driver.click('div.dwv_v', timeout=3)
+            except Exception:
+                pass
+
+        driver.sleep(sleep_action)
+
+        # Click "Accedi con email e password" using the specific input
+        console.print("[cyan]Clicking email/password login...")
+        try:
+            driver.click('input.gigya-input-submit[value="Accedi con email e password"]', timeout=5)
+        except Exception as e:
+            print(f"Error clicking email login: {e}")
+            return None
+
+        driver.sleep(sleep_action)
+
+        # Fill login credentials using specific IDs
+        console.print("[cyan]Filling login credentials...")
+        try:
+            email_input = 'input[name="username"].gigya-input-text'
+            driver.wait_for_element(email_input, timeout=5)
+            driver.type(email_input, username)
+            
+            password_input = 'input[name="password"].gigya-input-password'
+            driver.wait_for_element(password_input, timeout=5)
+            driver.type(password_input, password)
+            
+        except Exception as e:
+            print(f"Error filling credentials: {e}")
+            return None
+
+        driver.sleep(sleep_action)
+
+        # Click Continue/Procedi using the submit button
+        console.print("[cyan]Clicking continue button...")
+        try:
+            driver.click('input.gigya-input-submit[type="submit"][value="Continua"]', timeout=5)
+        except Exception as e:
+            print(f"Error clicking continue: {e}")
+            return None
+
+        # Wait for login response and parse network data
+        console.print("[cyan]Waiting for login response...")
+        for attempt in range(30):
+            driver.sleep(0.3)
+            
+            # Check network data for beToken - skip preflight requests
+            for data in network_data:
+                if data.get('method') == 'Network.responseReceived':
+                    params = data.get('params', {})
+                    response = params.get('response', {})
+                    url = response.get('url', '')
+                    request_type = params.get('type', '')
+                    
+                    if "persona/login/v2.0" in url and request_type != 'Preflight':
+                        request_id = params.get('requestId')
+                        
+                        if request_id:
+                            try:
+                                response_body = driver.execute_cdp_cmd(
+                                    'Network.getResponseBody', 
+                                    {'requestId': request_id}
+                                )
+                                body = response_body.get('body', '')
+
+                                if body:
+                                    response_data = json.loads(body)
+                                    be_token = response_data.get("response", {}).get("beToken")
+
+                                    if be_token:
+                                        be_token_holder["token"] = be_token
+                                        console.print("[green]Login successful! BeToken found!")
+                                        return be_token
+                                    
+                            except Exception:
+                                continue
+
+        console.print(f"[yellow]Login completed. Total network events captured: {len(network_data)}")
+        return be_token_holder["token"]
+        
+    finally:
+        driver.quit()
+
 def get_bearer_token():
     """
     Gets the BEARER_TOKEN for authentication.
@@ -24,6 +162,29 @@ def get_bearer_token():
     Returns:
         str: The bearer token string.
     """
+    global beToken
+
+    # Read beToken from config if already present
+    beToken = config_manager.get_dict("SITE_LOGIN", "mediasetinfinity")["beToken"]
+    if beToken is not None and len(beToken) != 0:
+        return beToken
+        
+    username = config_manager.get_dict("SITE_LOGIN", "mediasetinfinity").get("username", "")
+    password = config_manager.get_dict("SITE_LOGIN", "mediasetinfinity").get("password", "")
+    
+    if username and password:
+        beToken = generate_betoken(username, password)
+        
+        if beToken is not None:
+
+            # Save current beToken
+            current_value = config_manager.get("SITE_LOGIN", "mediasetinfinity", dict)
+            current_value["beToken"] = beToken
+            config_manager.set_key("SITE_LOGIN", "mediasetinfinity", current_value)
+            config_manager.save_config()
+
+            return beToken
+            
     return config_manager.get_dict("SITE_LOGIN", "mediasetinfinity")["beToken"]
 
 def get_playback_url(BEARER_TOKEN, CONTENT_ID):
@@ -158,6 +319,7 @@ def get_tracking_info(BEARER_TOKEN, PLAYBACK_JSON):
         response.raise_for_status()
 
         smil_xml = response.text
+        time.sleep(0.2)
         results = parse_smil_for_tracking_and_video(smil_xml)
         return results