StreamingCommunity 1.7.6__py3-none-any.whl → 1.9.1__py3-none-any.whl

StreamingCommunity/run.py

@@ -1,196 +1,202 @@
-# 10.12.23
-
-import os
-import sys
-import time
-import glob
-import logging
-import platform
-import argparse
-import importlib
-from typing import Callable
-
-
-# Internal utilities
-from StreamingCommunity.Src.Util.message import start_message
-from StreamingCommunity.Src.Util.console import console, msg
-from StreamingCommunity.Src.Util._jsonConfig import config_manager
-from StreamingCommunity.Src.Upload.update import update as git_update
-from StreamingCommunity.Src.Util.os import os_summary
-from StreamingCommunity.Src.Lib.TMBD import tmdb
-from StreamingCommunity.Src.Util.logger import Logger
-
-
-# Config
-CLOSE_CONSOLE = config_manager.get_bool('DEFAULT', 'not_close')
-SHOW_TRENDING = config_manager.get_bool('DEFAULT', 'show_trending')
-
-
-def run_function(func: Callable[..., None], close_console: bool = False) -> None:
-    """
-    Run a given function indefinitely or once, depending on the value of close_console.
-
-    Parameters:
-        func (Callable[..., None]): The function to run.
-        close_console (bool, optional): Whether to close the console after running the function once. Defaults to False.
-    """
-    if close_console:
-        while 1:
-            func()
-    else:
-        func()
-
-
-def load_search_functions():
-    modules = []
-    loaded_functions = {}
-
-    # Traverse the Api directory
-    api_dir = os.path.join(os.path.dirname(__file__), 'Src', 'Api', 'Site')
-    init_files = glob.glob(os.path.join(api_dir, '*', '__init__.py'))
-    
-    # Retrieve modules and their indices
-    for init_file in init_files:
-
-        # Get folder name as module name
-        module_name = os.path.basename(os.path.dirname(init_file))
-        logging.info(f"Load module name: {module_name}")
-
-        try:
-            # Dynamically import the module
-            mod = importlib.import_module(f'StreamingCommunity.Src.Api.Site.{module_name}')
-
-            # Get 'indice' from the module
-            indice = getattr(mod, 'indice', 0)
-            is_deprecate = bool(getattr(mod, '_deprecate', True))
-            use_for = getattr(mod, '_useFor', 'other')
-
-            if not is_deprecate:
-                modules.append((module_name, indice, use_for))
-
-        except Exception as e:
-            console.print(f"[red]Failed to import module {module_name}: {str(e)}")
-
-    # Sort modules by 'indice'
-    modules.sort(key=lambda x: x[1])
-
-    # Load search functions in the sorted order
-    for module_name, _, use_for in modules:
-
-        # Construct a unique alias for the module
-        module_alias = f'{module_name}_search'
-
-        try:
-
-            # Dynamically import the module
-            mod = importlib.import_module(f'StreamingCommunity.Src.Api.Site.{module_name}')
-
-            # Get the search function from the module (assuming the function is named 'search' and defined in __init__.py)
-            search_function = getattr(mod, 'search')
-
-            # Add the function to the loaded functions dictionary
-            loaded_functions[module_alias] = (search_function, use_for)
-
-        except Exception as e:
-            console.print(f"[red]Failed to load search function from module {module_name}: {str(e)}")
-
-    return loaded_functions
-
-
-def initialize():
-
-    # Get start message
-    start_message()
-
-    # Get system info
-    os_summary.get_system_summary()
-
-    # Set terminal size for win 7
-    if platform.system() == "Windows" and "7" in platform.version():
-        os.system('mode 120, 40')
-
-    # Check python version
-    if sys.version_info < (3, 7):
-        console.log("[red]Install python version > 3.7.16")
-        sys.exit(0)
-
-    # Attempting GitHub update
-    try:
-        git_update()
-        print()
-    except:
-        console.log("[red]Error with loading github.")
-
-    # Show trending film and series
-    if SHOW_TRENDING:
-        tmdb.display_trending_films()
-        print()
-        tmdb.display_trending_tv_shows()
-        print()
-        
-
-def main():
-
-    start = time.time()
-
-    # Create logger
-    log_not = Logger()
-    initialize()
-
-    # Load search functions
-    search_functions = load_search_functions()
-    logging.info(f"Load module in: {time.time() - start} s")
-
-    # Create dynamic argument parser
-    parser = argparse.ArgumentParser(description='Script to download film and series from the internet.')
-
-    color_map = {
-        "anime": "red",
-        "film_serie": "yellow",
-        "film": "blue",
-        "serie": "green",
-        "other": "white"
-    }
-
-    # Add dynamic arguments based on loaded search modules
-    for alias, (_, use_for) in search_functions.items():
-        short_option = alias[:3].upper()
-        long_option = alias
-        parser.add_argument(f'-{short_option}', f'--{long_option}', action='store_true', help=f'Search for {alias.split("_")[0]} on streaming platforms.')
-
-    # Parse command line arguments
-    args = parser.parse_args()
-
-    # Mapping command-line arguments to functions
-    arg_to_function = {alias: func for alias, (func, _) in search_functions.items()}
-
-    # Check which argument is provided and run the corresponding function
-    for arg, func in arg_to_function.items():
-        if getattr(args, arg):
-            run_function(func)
-            return
-
-    # Mapping user input to functions
-    input_to_function = {str(i): func for i, (alias, (func, _)) in enumerate(search_functions.items())}
-
-    # Create dynamic prompt message and choices
-    choice_labels = {str(i): (alias.split("_")[0].capitalize(), use_for) for i, (alias, (_, use_for)) in enumerate(search_functions.items())}
-
-    # Display the category legend in a single line
-    legend_text = " | ".join([f"[{color}]{category.capitalize()}[/{color}]" for category, color in color_map.items()])
-    console.print(f"[bold green]Category Legend:[/bold green] {legend_text}")
-
-    # Construct the prompt message with color-coded site names
-    prompt_message = "[green]Insert category [white](" + ", ".join(
-        [f"{key}: [{color_map[label[1]]}]{label[0]}[/{color_map[label[1]]}]" for key, label in choice_labels.items()]
-    ) + "[white])"
-
-    # Ask the user for input
-    category = msg.ask(prompt_message, choices=list(choice_labels.keys()), default="0", show_choices=False, show_default=False)
-
-    # Run the corresponding function based on user input
-    if category in input_to_function:
-        run_function(input_to_function[category])
-    else:
-        console.print("[red]Invalid category.")
-        sys.exit(0)
+# 10.12.23
+
+import os
+import sys
+import time
+import glob
+import logging
+import platform
+import argparse
+import importlib
+from typing import Callable
+
+
+# Internal utilities
+from StreamingCommunity.Util.message import start_message
+from StreamingCommunity.Util.console import console, msg
+from StreamingCommunity.Util._jsonConfig import config_manager
+from StreamingCommunity.Upload.update import update as git_update
+from StreamingCommunity.Util.os import OsSummary
+from StreamingCommunity.Lib.TMBD import tmdb
+from StreamingCommunity.Util.logger import Logger
+
+
+# Config
+CLOSE_CONSOLE = config_manager.get_bool('DEFAULT', 'not_close')
+SHOW_TRENDING = config_manager.get_bool('DEFAULT', 'show_trending')
+
+
+def run_function(func: Callable[..., None], close_console: bool = False) -> None:
+    """
+    Run a given function indefinitely or once, depending on the value of close_console.
+
+    Parameters:
+        func (Callable[..., None]): The function to run.
+        close_console (bool, optional): Whether to close the console after running the function once. Defaults to False.
+    """
+    if close_console:
+        while 1:
+            func()
+    else:
+        func()
+
+
+def load_search_functions():
+    modules = []
+    loaded_functions = {}
+
+    # Find api home directory
+    if getattr(sys, 'frozen', False):  # Modalità PyInstaller
+        base_path = os.path.join(sys._MEIPASS, "StreamingCommunity")
+    else:
+        base_path = os.path.dirname(__file__)
+
+    api_dir = os.path.join(base_path, 'Api', 'Site')
+    init_files = glob.glob(os.path.join(api_dir, '*', '__init__.py'))
+    
+    # Retrieve modules and their indices
+    for init_file in init_files:
+
+        # Get folder name as module name
+        module_name = os.path.basename(os.path.dirname(init_file))
+        logging.info(f"Load module name: {module_name}")
+
+        try:
+            # Dynamically import the module
+            mod = importlib.import_module(f'StreamingCommunity.Api.Site.{module_name}')
+
+            # Get 'indice' from the module
+            indice = getattr(mod, 'indice', 0)
+            is_deprecate = bool(getattr(mod, '_deprecate', True))
+            use_for = getattr(mod, '_useFor', 'other')
+
+            if not is_deprecate:
+                modules.append((module_name, indice, use_for))
+
+        except Exception as e:
+            console.print(f"[red]Failed to import module {module_name}: {str(e)}")
+
+    # Sort modules by 'indice'
+    modules.sort(key=lambda x: x[1])
+
+    # Load search functions in the sorted order
+    for module_name, _, use_for in modules:
+
+        # Construct a unique alias for the module
+        module_alias = f'{module_name}_search'
+
+        try:
+
+            # Dynamically import the module
+            mod = importlib.import_module(f'StreamingCommunity.Api.Site.{module_name}')
+
+            # Get the search function from the module (assuming the function is named 'search' and defined in __init__.py)
+            search_function = getattr(mod, 'search')
+
+            # Add the function to the loaded functions dictionary
+            loaded_functions[module_alias] = (search_function, use_for)
+
+        except Exception as e:
+            console.print(f"[red]Failed to load search function from module {module_name}: {str(e)}")
+
+    return loaded_functions
+
+
+def initialize():
+
+    # Get start message
+    start_message()
+
+    # Get system info
+    os_summary = OsSummary()
+    os_summary.get_system_summary()
+
+    # Set terminal size for win 7
+    if platform.system() == "Windows" and "7" in platform.version():
+        os.system('mode 120, 40')
+
+    # Check python version
+    if sys.version_info < (3, 7):
+        console.log("[red]Install python version > 3.7.16")
+        sys.exit(0)
+
+    # Attempting GitHub update
+    try:
+        git_update()
+        print()
+    except:
+        console.log("[red]Error with loading github.")
+
+    # Show trending film and series
+    if SHOW_TRENDING:
+        tmdb.display_trending_films()
+        print()
+        tmdb.display_trending_tv_shows()
+        print()
+        
+
+def main():
+
+    start = time.time()
+
+    # Create logger
+    log_not = Logger()
+    initialize()
+
+    # Load search functions
+    search_functions = load_search_functions()
+    logging.info(f"Load module in: {time.time() - start} s")
+
+    # Create dynamic argument parser
+    parser = argparse.ArgumentParser(description='Script to download film and series from the internet.')
+
+    color_map = {
+        "anime": "red",
+        "film_serie": "yellow",
+        "film": "blue",
+        "serie": "green",
+        "other": "white"
+    }
+
+    # Add dynamic arguments based on loaded search modules
+    for alias, (_, use_for) in search_functions.items():
+        short_option = alias[:3].upper()
+        long_option = alias
+        parser.add_argument(f'-{short_option}', f'--{long_option}', action='store_true', help=f'Search for {alias.split("_")[0]} on streaming platforms.')
+
+    # Parse command line arguments
+    args = parser.parse_args()
+
+    # Mapping command-line arguments to functions
+    arg_to_function = {alias: func for alias, (func, _) in search_functions.items()}
+
+    # Check which argument is provided and run the corresponding function
+    for arg, func in arg_to_function.items():
+        if getattr(args, arg):
+            run_function(func)
+            return
+
+    # Mapping user input to functions
+    input_to_function = {str(i): func for i, (alias, (func, _)) in enumerate(search_functions.items())}
+
+    # Create dynamic prompt message and choices
+    choice_labels = {str(i): (alias.split("_")[0].capitalize(), use_for) for i, (alias, (_, use_for)) in enumerate(search_functions.items())}
+
+    # Display the category legend in a single line
+    legend_text = " | ".join([f"[{color}]{category.capitalize()}[/{color}]" for category, color in color_map.items()])
+    console.print(f"\n[bold green]Category Legend:[/bold green] {legend_text}")
+
+    # Construct the prompt message with color-coded site names
+    prompt_message = "[green]Insert category [white](" + ", ".join(
+        [f"{key}: [{color_map[label[1]]}]{label[0]}[/{color_map[label[1]]}]" for key, label in choice_labels.items()]
+    ) + "[white])"
+
+    # Ask the user for input
+    category = msg.ask(prompt_message, choices=list(choice_labels.keys()), default="0", show_choices=False, show_default=False)
+
+    # Run the corresponding function based on user input
+    if category in input_to_function:
+        run_function(input_to_function[category])
+    else:
+        console.print("[red]Invalid category.")
+        sys.exit(0)