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

StreamingCommunity/Util/os.py

@@ -1,514 +0,0 @@
-# 24.01.24
-
-import io
-import os
-import sys
-import time
-import shutil
-import hashlib
-import logging
-import platform
-import unidecode
-import subprocess
-import contextlib
-import pathvalidate
-import urllib.request
-import importlib.metadata
-
-
-# External library
-import httpx
-
-
-# Internal utilities
-from .ffmpeg_installer import check_ffmpeg
-from StreamingCommunity.Util.console import console, msg
-
-
-# Variable
-OS_CONFIGURATIONS = {
-        'windows': {
-            'max_length': 255,
-            'invalid_chars': '<>:"/\\|?*',
-            'reserved_names': [
-                "CON", "PRN", "AUX", "NUL",
-                "COM1", "COM2", "COM3", "COM4", "COM5", "COM6", "COM7", "COM8", "COM9",
-                "LPT1", "LPT2", "LPT3", "LPT4", "LPT5", "LPT6", "LPT7", "LPT8", "LPT9"
-            ],
-            'max_path': 255
-        },
-        'darwin': {
-            'max_length': 4096,
-            'invalid_chars': '/:',
-            'reserved_names': [],
-            'hidden_file_restriction': True
-        },
-        'linux': {
-            'max_length': 4096,
-            'invalid_chars': '/\0',
-            'reserved_names': []
-        }
-    }
-
-
-
-class OsManager:
-    def __init__(self):
-        self.system = self._detect_system()
-        self.config = OS_CONFIGURATIONS.get(self.system, {})
-
-    def _detect_system(self) -> str:
-        """Detect and normalize operating system name."""
-        system = platform.system().lower()
-
-        if system in OS_CONFIGURATIONS:
-            return system
-        
-        raise ValueError(f"Unsupported operating system: {system}")
-
-    def _process_filename(self, filename: str) -> str:
-        """
-        Comprehensively process filename with cross-platform considerations.
-        
-        Args:
-            filename (str): Original filename.
-        
-        Returns:
-            str: Processed filename.
-        """
-        # Preserve file extension
-        logging.info("_process_filename: ", filename)
-        name, ext = os.path.splitext(filename)
-        
-        # Handle length restrictions
-        if len(name) > self.config['max_length']:
-            name = self._truncate_filename(name)
-        
-        # Reconstruct filename
-        processed_filename = name + ext
-        
-        return processed_filename
-
-    def _truncate_filename(self, name: str) -> str:
-        """
-        Truncate filename based on OS-specific rules.
-        
-        Args:
-            name (str): Original filename.
-        
-        Returns:
-            str: Truncated filename.
-        """
-        logging.info("_truncate_filename: ", name)
-
-        if self.system == 'windows':
-            return name[:self.config['max_length'] - 3] + '___'
-        elif self.system == 'darwin':
-            return name[:self.config['max_length']]
-        elif self.system == 'linux':
-            return name[:self.config['max_length'] - 2] + '___'
-
-    def get_sanitize_file(self, filename: str) -> str:
-        """
-        Sanitize filename using pathvalidate with unidecode.
-        
-        Args:
-            filename (str): Original filename.
-        
-        Returns:
-            str: Sanitized filename.
-        """
-        logging.info("get_sanitize_file: ", filename)
-
-        # Decode unicode characters and sanitize
-        decoded_filename = unidecode.unidecode(filename)
-        sanitized_filename = pathvalidate.sanitize_filename(decoded_filename)
-        
-        # Truncate if necessary based on OS configuration
-        name, ext = os.path.splitext(sanitized_filename)
-        if len(name) > self.config['max_length']:
-            name = self._truncate_filename(name)
-        
-        logging.info("return :", name + ext)
-        return name + ext
-
-    def get_sanitize_path(self, path: str) -> str:
-        """
-        Sanitize folder path using pathvalidate with unidecode.
-        
-        Args:
-            path (str): Original folder path.
-        
-        Returns:
-            str: Sanitized folder path.
-        """
-        logging.info("get_sanitize_file: ", path)
-
-        # Decode unicode characters and sanitize
-        decoded_path = unidecode.unidecode(path)
-        sanitized_path = pathvalidate.sanitize_filepath(decoded_path)
-        
-        # Split path and process each component
-        path_components = os.path.normpath(sanitized_path).split(os.sep)
-        processed_components = []
-        
-        for component in path_components:
-            # Truncate component if necessary
-            if len(component) > self.config['max_length']:
-                component = self._truncate_filename(component)
-            processed_components.append(component)
-
-        logging.info("return :", os.path.join(*processed_components))
-        return os.path.join(*processed_components)
-
-    def create_path(self, path: str, mode: int = 0o755) -> bool:
-        """
-        Create directory path with specified permissions.
-        
-        Args:
-            path (str): Path to create.
-            mode (int, optional): Directory permissions. Defaults to 0o755.
-        
-        Returns:
-            bool: True if path created successfully, False otherwise.
-        """
-        try:
-            # Sanitize path first
-            sanitized_path = self.get_sanitize_path(path)
-            
-            # Create directory with recursive option
-            os.makedirs(sanitized_path, mode=mode, exist_ok=True)
-            return True
-        
-        except Exception as e:
-            logging.error(f"Path creation error: {e}")
-            return False
-
-    def remove_folder(self, folder_path: str) -> bool:
-        """
-        Safely remove a folder.
-        
-        Args:
-            folder_path (str): Path of directory to remove.
-        
-        Returns:
-            bool: Removal status.
-        """
-        try:
-            shutil.rmtree(folder_path)
-            return True
-        except OSError as e:
-            logging.error(f"Folder removal error: {e}")
-            return False
-    
-    def remove_files_except_one(self, folder_path: str, keep_file: str) -> None:
-        """
-        Delete all files in a folder except for one specified file.
-
-        Parameters:
-            - folder_path (str): The path to the folder containing the files.
-            - keep_file (str): The filename to keep in the folder.
-        """
-
-        try:
-            # List all files in the folder
-            files_in_folder = os.listdir(folder_path)
-            
-            # Iterate over each file in the folder
-            for file_name in files_in_folder:
-                file_path = os.path.join(folder_path, file_name)
-                
-                # Check if the file is not the one to keep and is a regular file
-                if file_name != keep_file and os.path.isfile(file_path):
-                    os.remove(file_path)  # Delete the file
-
-        except Exception as e:
-            logging.error(f"An error occurred: {e}")
-            raise
-    
-    def check_file(self, file_path: str) -> bool:
-        """
-        Check if a file exists at the given file path.
-
-        Parameters:
-            file_path (str): The path to the file.
-
-        Returns:
-            bool: True if the file exists, False otherwise.
-        """
-        try:
-            logging.info(f"Check if file exists: {file_path}")
-            if os.path.exists(file_path):
-                logging.info(f"The file '{file_path}' exists.")
-                return True
-            
-            else:
-                return False
-            
-        except Exception as e:
-            logging.error(f"An error occurred while checking file existence: {e}")
-            return False
-
-
-class InternManager():
-
-    def format_file_size(self, size_bytes: float) -> str:
-        """
-        Formats a file size from bytes into a human-readable string representation.
-
-        Parameters:
-            size_bytes (float): Size in bytes to be formatted.
-
-        Returns:
-            str: Formatted string representing the file size with appropriate unit (B, KB, MB, GB, TB).
-        """
-        if size_bytes <= 0:
-            return "0B"
-
-        units = ['B', 'KB', 'MB', 'GB', 'TB']
-        unit_index = 0
-
-        while size_bytes >= 1024 and unit_index < len(units) - 1:
-            size_bytes /= 1024
-            unit_index += 1
-
-        return f"{size_bytes:.2f} {units[unit_index]}"
-
-    def format_transfer_speed(self, bytes: float) -> str:
-        """
-        Formats a transfer speed from bytes per second into a human-readable string representation.
-
-        Parameters:
-            bytes (float): Speed in bytes per second to be formatted.
-
-        Returns:
-            str: Formatted string representing the transfer speed with appropriate unit (Bytes/s, KB/s, MB/s).
-        """
-        if bytes < 1024:
-            return f"{bytes:.2f} Bytes/s"
-        elif bytes < 1024 * 1024:
-            return f"{bytes / 1024:.2f} KB/s"
-        else:
-            return f"{bytes / (1024 * 1024):.2f} MB/s"
-
-    @staticmethod
-    def check_internet():
-        while True:
-            try:
-                httpx.get("https://www.google.com")
-                console.log("[bold green]Internet is available![/bold green]")
-                break
-
-            except urllib.error.URLError:
-                console.log("[bold red]Internet is not available. Waiting...[/bold red]")
-                time.sleep(5)
-
-        print()
-
-
-class OsSummary():
-
-    def get_executable_version(self, command: list):
-        """
-        Get the version of a given command-line executable.
-
-        Args:
-            command (list): The command to run, e.g., `['ffmpeg', '-version']`.
-
-        Returns:
-            str: The version string of the executable.
-        """
-        try:
-            version_output = subprocess.check_output(command, stderr=subprocess.STDOUT).decode().split('\n')[0]
-            return version_output.split(" ")[2]
-        
-        except (FileNotFoundError, subprocess.CalledProcessError):
-            console.print(f"{command[0]} not found", style="bold red")
-            sys.exit(0)
-
-    def check_ffmpeg_location(self, command: list):
-        """
-        Run 'where ffmpeg' command to check FFmpeg's location.
-
-        Returns:
-            str: Location of FFmpeg executable or None if not found
-        """
-        try:
-            result = subprocess.check_output(command, stderr=subprocess.STDOUT, text=True).strip()
-            return result
-        
-        except subprocess.CalledProcessError:
-            console.print("FFmpeg not found in system PATH", style="bold red")
-            sys.exit(0)
-
-    def get_library_version(self, lib_name: str):
-        """
-        Retrieve the version of a Python library.
-        
-        Args:
-            lib_name (str): The name of the Python library.
-        
-        Returns:
-            str: The library name followed by its version, or `-not installed` if not found.
-        """
-        try:
-            version = importlib.metadata.version(lib_name)
-            return f"{lib_name}-{version}"
-        
-        except importlib.metadata.PackageNotFoundError:
-            return f"{lib_name}-not installed"
-
-    def download_requirements(self, url: str, filename: str):
-        """
-        Download the requirements.txt file from the specified URL if not found locally using requests.
-        
-        Args:
-            url (str): The URL to download the requirements file from.
-            filename (str): The local filename to save the requirements file as.
-        """
-        try:
-            import requests
-            
-            console.print(f"{filename} not found locally. Downloading from {url}...", style="bold yellow")
-            response = requests.get(url)
-            
-            if response.status_code == 200:
-                with open(filename, 'wb') as f:
-                    f.write(response.content)
-                console.print(f"{filename} successfully downloaded.", style="bold green")
-            
-            else:
-                console.print(f"Failed to download {filename}. HTTP Status code: {response.status_code}", style="bold red")
-                sys.exit(1)
-        
-        except Exception as e:
-            console.print(f"Failed to download {filename}: {e}", style="bold red")
-            sys.exit(1)
-
-    def install_library(self, lib_name: str):
-        """
-        Install a Python library using pip.
-
-        Args:
-            lib_name (str): The name of the library to install.
-        """
-        try:
-            console.print(f"Installing {lib_name}...", style="bold yellow")
-            subprocess.check_call([sys.executable, "-m", "pip", "install", lib_name])
-            console.print(f"{lib_name} installed successfully!", style="bold green")
-            
-        except subprocess.CalledProcessError as e:
-            console.print(f"Failed to install {lib_name}: {e}", style="bold red")
-            sys.exit(1)
-
-    def check_python_version(self):
-        """
-        Check if the installed Python is the official CPython distribution.
-        Exits with a message if not the official version.
-        """
-        python_implementation = platform.python_implementation()
-
-        if python_implementation != "CPython":
-            console.print(f"[bold red]Warning: You are using a non-official Python distribution: {python_implementation}.[/bold red]")
-            console.print("Please install the official Python from [bold blue]https://www.python.org[/bold blue] and try again.", style="bold yellow")
-            sys.exit(0)
-
-    async def get_system_summary(self):
-        """
-        Generate a summary of the system environment.
-
-        Includes:
-            - Python version and implementation details.
-            - Operating system and architecture.
-            - Versions of `ffmpeg` and `ffprobe` executables.
-            - Installed Python libraries as listed in `requirements.txt`.
-        """
-        
-        # Check if Python is the official CPython
-        self.check_python_version()
-
-        # Check internet connectivity
-        InternManager().check_internet()
-        console.print("[bold blue]System Summary[/bold blue][white]:")
-
-        # Python version and platform
-        python_version = sys.version.split()[0]
-        python_implementation = platform.python_implementation()
-        arch = platform.machine()
-        os_info = platform.platform()
-        glibc_version = 'glibc ' + '.'.join(map(str, platform.libc_ver()[1]))
-        
-        console.print(f"[cyan]Python[white]: [bold red]{python_version} ({python_implementation} {arch}) - {os_info} ({glibc_version})[/bold red]")
-        logging.info(f"Python: {python_version} ({python_implementation} {arch}) - {os_info} ({glibc_version})")
-        
-        # ffmpeg and ffprobe versions
-        ffmpeg_path, ffprobe_path = check_ffmpeg()
-        
-        # Locate ffmpeg and ffprobe
-        if "binary" not in ffmpeg_path:        
-            ffmpeg_path = self.check_ffmpeg_location(['where', 'ffmpeg'])
-        if "binary" not in ffprobe_path:
-            ffprobe_path = self.check_ffmpeg_location(['where', 'ffprobe'])
-
-        ffmpeg_version = self.get_executable_version([ffprobe_path, '-version'])
-        ffprobe_version = self.get_executable_version([ffprobe_path, '-version'])
-
-        console.print(f"[cyan]Path[white]: [red]ffmpeg [bold yellow]'{ffmpeg_path}'[/bold yellow][white], [red]ffprobe '[bold yellow]{ffprobe_path}'[/bold yellow]")
-        console.print(f"[cyan]Exe versions[white]: [bold red]ffmpeg {ffmpeg_version}, ffprobe {ffprobe_version}[/bold red]")
-
-        # Check if requirements.txt exists, if not on pyinstaller
-        if not getattr(sys, 'frozen', False):
-            requirements_file = 'requirements.txt'
-            
-            if not os.path.exists(requirements_file):
-                self.download_requirements(
-                    'https://raw.githubusercontent.com/Lovi-0/StreamingCommunity/refs/heads/main/requirements.txt',
-                    requirements_file
-                )
-            
-            # Read the optional libraries from the requirements file, get only name without version if "library==1.0.0"
-            optional_libraries = [line.strip().split("=")[0] for line in open(requirements_file, 'r', encoding='utf-8-sig')]
-            
-            # Check if libraries are installed and prompt to install missing ones
-            for lib in optional_libraries:
-                installed_version = self.get_library_version(lib)
-
-                if 'not installed' in installed_version:
-                    user_response = msg.ask(f"{lib} is not installed. Do you want to install it? (yes/no)", default="y")
-                    
-                    if user_response.lower().strip() in ["yes", "y"]:
-                        self.install_library(lib)
-                
-                else:
-                    logging.info(f"Library: {installed_version}")
-            
-            console.print(f"[cyan]Libraries[white]: [bold red]{', '.join([self.get_library_version(lib) for lib in optional_libraries])}[/bold red]\n")
-            logging.info(f"Libraries: {', '.join([self.get_library_version(lib) for lib in optional_libraries])}")
-
-
-
-# OTHER
-os_manager = OsManager()
-internet_manager = InternManager()
-os_summary = OsSummary()
-
-@contextlib.contextmanager
-def suppress_output():
-    with contextlib.redirect_stdout(io.StringIO()):
-        yield
-
-def compute_sha1_hash(input_string: str) -> str:
-    """
-    Computes the SHA-1 hash of the input string.
-
-    Parameters:
-        - input_string (str): The string to be hashed.
-
-    Returns:
-        str: The SHA-1 hash of the input string.
-    """
-    # Compute the SHA-1 hash
-    hashed_string = hashlib.sha1(input_string.encode()).hexdigest()
-    
-    # Return the hashed string
-    return hashed_string

StreamingCommunity/Util/table.py

@@ -1,163 +0,0 @@
-# 03.03.24
-
-from rich.console import Console
-from rich.table import Table
-from rich.prompt import Prompt
-from rich.style import Style
-from typing import Dict, List, Any
-
-
-# Internal utilities
-from .message import start_message
-
-
-class TVShowManager:
-    def __init__(self):
-        """
-        Initialize TVShowManager with provided column information.
-
-        Parameters:
-            - column_info (Dict[str, Dict[str, str]]): Dictionary containing column names, their colors, and justification.
-        """
-        self.console = Console()
-        self.tv_shows: List[Dict[str, Any]] = []  # List to store TV show data as dictionaries
-        self.slice_start: int = 0
-        self.slice_end: int = 5
-        self.step: int = self.slice_end
-        self.column_info = []
-
-    def set_slice_end(self, new_slice: int) -> None:
-        """
-        Set the end of the slice for displaying TV shows.
-
-        Parameters:
-            - new_slice (int): The new value for the slice end.
-        """
-        self.slice_end = new_slice
-        self.step = new_slice
-
-    def add_column(self, column_info: Dict[str, Dict[str, str]]) -> None:
-        """
-        Add column information.
-    
-        Parameters:
-            - column_info (Dict[str, Dict[str, str]]): Dictionary containing column names, their colors, and justification.
-        """
-        self.column_info = column_info
-
-    def add_tv_show(self, tv_show: Dict[str, Any]):
-        """
-        Add a TV show to the list of TV shows.
-
-        Parameters:
-            - tv_show (Dict[str, Any]): Dictionary containing TV show details.
-        """
-        self.tv_shows.append(tv_show)
-
-    def display_data(self, data_slice: List[Dict[str, Any]]):
-        """
-        Display TV show data in a tabular format.
-
-        Parameters:
-            - data_slice (List[Dict[str, Any]]): List of dictionaries containing TV show details to display.
-        """
-        table = Table(border_style="white")
-
-        # Add columns dynamically based on provided column information
-        for col_name, col_style in self.column_info.items():
-            color = col_style.get("color", None)
-            if color:
-                style = Style(color=color)
-            else:
-                style = None
-            table.add_column(col_name, style=style, justify='center')
-
-        # Add rows dynamically based on available TV show data
-        for entry in data_slice:
-            # Create row data while handling missing keys
-            row_data = [entry.get(col_name, '') for col_name in self.column_info.keys()]
-            table.add_row(*row_data)
-
-        self.console.print(table)  # Use self.console.print instead of print
-
-
-    def run(self, force_int_input: bool = False, max_int_input: int = 0) -> str:
-        """
-        Run the TV show manager application.
-
-        Parameters:
-            - force_int_input(bool): If True, only accept integer inputs from 0 to max_int_input
-            - max_int_input (int): range of row to show
-        
-        Returns:
-            str: Last command executed before breaking out of the loop.
-        """
-        total_items = len(self.tv_shows)
-        last_command = ""  # Variable to store the last command executed
-
-        while True:
-            start_message()
-
-            # Display table
-            self.display_data(self.tv_shows[self.slice_start:self.slice_end])
-
-            # Handling user input for loading more items or quitting
-            if self.slice_end < total_items:
-                self.console.print(f"\n\n[yellow][INFO] [green]Press [red]Enter [green]for next page, or [red]'q' [green]to quit.")
-
-                if not force_int_input:
-                    key = Prompt.ask(
-                        "\n[cyan]Insert media index [yellow](e.g., 1), [red]* [cyan]to download all media, "
-                        "[yellow](e.g., 1-2) [cyan]for a range of media, or [yellow](e.g., 3-*) [cyan]to download from a specific index to the end"
-                    )
-                    
-                else:
-                    choices = [str(i) for i in range(0, max_int_input)]
-                    choices.extend(["q", ""])
-
-                    key = Prompt.ask("[cyan]Insert media [red]index", choices=choices, show_choices=False)
-                last_command = key
-
-                if key.lower() == "q":
-                    break
-
-                elif key == "":
-                    self.slice_start += self.step
-                    self.slice_end += self.step
-                    if self.slice_end > total_items:
-                        self.slice_end = total_items
-
-                else:
-                    break
-
-            else:
-                # Last slice, ensure all remaining items are shown
-                self.console.print(f"\n\n[yellow][INFO] [red]You've reached the end. [green]Press [red]Enter [green]for next page, or [red]'q' [green]to quit.")
-                if not force_int_input:
-                    key = Prompt.ask(
-                        "\n[cyan]Insert media index [yellow](e.g., 1), [red]* [cyan]to download all media, "
-                        "[yellow](e.g., 1-2) [cyan]for a range of media, or [yellow](e.g., 3-*) [cyan]to download from a specific index to the end"
-                    )
-
-
-                else:
-                    choices = [str(i) for i in range(0, max_int_input)]
-                    choices.extend(["q", ""])
-
-                    key = Prompt.ask("[cyan]Insert media [red]index", choices=choices, show_choices=False)
-                last_command = key
-
-                if key.lower() == "q":
-                    break
-
-                elif key == "":
-                    self.slice_start = 0
-                    self.slice_end = self.step
-
-                else:
-                    break
-            
-        return last_command
-
-    def clear(self):
-        self.tv_shows = []