StreamingCommunity 2.5.7__py3-none-any.whl → 2.5.8__py3-none-any.whl

StreamingCommunity/TelegramHelp/telegram_bot.py

@@ -1,4 +1,4 @@
-# 04.02.25
+# 04.02.26
 # Made by: @GiuPic
 
 import os
@@ -9,16 +9,141 @@ import uuid
 import json
 import threading
 import subprocess
-
+import threading
+from typing import Optional
 
 # External libraries
 import telebot
 
+session_data = {}
+
+class TelegramSession:
+
+    def set_session(value):
+        session_data['script_id'] = value
+
+    def get_session():
+        return session_data.get('script_id', 'unknown')
+
+    def updateScriptId(screen_id, titolo):
+        json_file = "../../scripts.json"
+        try:
+            with open(json_file, 'r') as f:
+                scripts_data = json.load(f)
+        except FileNotFoundError:
+            scripts_data = []
+
+        # cerco lo script con lo screen_id
+        for script in scripts_data:
+            if script["screen_id"] == screen_id:
+                # se trovo il match, aggiorno il titolo
+                script["titolo"] = titolo
+
+                # aggiorno il file json
+                with open(json_file, 'w') as f:
+                    json.dump(scripts_data, f, indent=4)
+
+                return
+
+        print(f"Screen_id {screen_id} non trovato.")
+
+    def deleteScriptId(screen_id):
+        json_file = "../../scripts.json"
+        try:
+            with open(json_file, 'r') as f:
+                scripts_data = json.load(f)
+        except FileNotFoundError:
+            scripts_data = []
+
+        for script in scripts_data:
+            if script["screen_id"] == screen_id:
+                # se trovo il match, elimino lo script
+                scripts_data.remove(script)
+
+                # aggiorno il file json
+                with open(json_file, 'w') as f:
+                    json.dump(scripts_data, f, indent=4)
+
+                print(f"Script eliminato per screen_id {screen_id}")
+                return
+
+        print(f"Screen_id {screen_id} non trovato.")
+
+class TelegramRequestManager:
+    _instance = None
+
+    def __new__(cls, *args, **kwargs):
+        if not cls._instance:
+            cls._instance = super().__new__(cls)
+        return cls._instance
+
+    def __init__(self, json_file: str = "active_requests.json"):
+        if not hasattr(self, 'initialized'):
+            self.json_file = json_file
+            self.initialized = True
+            self.on_response_callback = None
+
+    def create_request(self, type: str) -> str:
+        request_data = {
+            "type": type,
+            "response": None,
+            "timestamp": time.time()
+        }
+
+        with open(self.json_file, "w") as f:
+            json.dump(request_data, f)
+
+        return "Ok"
+
+    def save_response(self, message_text: str) -> bool:
+        try:
+            # Carica il file JSON
+            with open(self.json_file, "r") as f:
+                data = json.load(f)
+
+            # Controlla se esiste la chiave 'type' e se la risposta è presente
+            if "type" in data and "response" in data:
+                data["response"] = message_text  # Aggiorna la risposta
 
-# Fix import
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
-from StreamingCommunity.TelegramHelp.request_manager import RequestManager
+                with open(self.json_file, "w") as f:
+                    json.dump(data, f, indent=4)
 
+                return True
+            else:
+                return False
+
+        except (FileNotFoundError, json.JSONDecodeError) as e:
+            print(f"⚠️ save_response - errore: {e}")
+            return False
+
+    def get_response(self) -> Optional[str]:
+        try:
+            with open(self.json_file, "r") as f:
+                data = json.load(f)
+
+                # Verifica se esiste la chiave "response"
+                if "response" in data:
+                    response = data["response"]  # Ottieni la risposta direttamente
+
+                    if response is not None and self.on_response_callback:
+                        self.on_response_callback(response)
+
+                    return response
+
+        except (FileNotFoundError, json.JSONDecodeError) as e:
+            print(f"get_response - errore: {e}")
+        return None
+
+    def clear_file(self) -> bool:
+        try:
+            with open(self.json_file, "w") as f:
+                json.dump({}, f)
+            print(f"File {self.json_file} è stato svuotato con successo.")
+            return True
+
+        except Exception as e:
+            print(f"⚠️ clear_file - errore: {e}")
+            return False
 
 # Funzione per caricare variabili da un file .env
 def load_env(file_path="../../.env"):
@@ -44,13 +169,17 @@ class TelegramBot:
             if os.path.exists(cls._config_file):
                 with open(cls._config_file, "r") as f:
                     config = json.load(f)
-                cls._instance = cls.init_bot(
-                    config["token"], config["authorized_user_id"]
-                )
+
+                # Assicura che authorized_user_id venga trattato come una lista
+                authorized_users = config.get('authorized_user_id', [])
+                if isinstance(authorized_users, str):
+                    authorized_users = [int(uid) for uid in authorized_users.split(",") if uid.strip().isdigit()]
+
+                cls._instance = cls.init_bot(config['token'], authorized_users)
+                #cls._instance = cls.init_bot(config['token'], config['authorized_user_id'])
+
             else:
-                raise Exception(
-                    "Bot non ancora inizializzato. Chiamare prima init_bot() con token e authorized_user_id"
-                )
+                raise Exception("Bot non ancora inizializzato. Chiamare prima init_bot() con token e authorized_user_id")
         return cls._instance
 
     @classmethod
@@ -63,7 +192,8 @@ class TelegramBot:
                 json.dump(config, f)
         return cls._instance
 
-    def __init__(self, token, authorized_user_id):
+    def __init__(self, token, authorized_users):
+
         def monitor_scripts():
             while True:
                 try:
@@ -132,10 +262,10 @@ class TelegramBot:
             )
 
         self.token = token
-        self.authorized_user_id = authorized_user_id
-        self.chat_id = authorized_user_id
+        self.authorized_users = authorized_users
+        self.chat_id = authorized_users
         self.bot = telebot.TeleBot(token)
-        self.request_manager = RequestManager()
+        self.request_manager = TelegramRequestManager()
 
         # Registra gli handler
         self.register_handlers()
@@ -171,7 +301,7 @@ class TelegramBot:
             self.handle_response(message)
 
     def is_authorized(self, user_id):
-        return user_id == self.authorized_user_id
+        return user_id in self.authorized_users
 
     def handle_get_id(self, message):
         if not self.is_authorized(message.from_user.id):
@@ -194,7 +324,6 @@ class TelegramBot:
 
         screen_id = str(uuid.uuid4())[:8]
         debug_mode = os.getenv("DEBUG")
-        verbose = debug_mode
 
         if debug_mode == "True":
             subprocess.Popen(["python3", "../../test_run.py", screen_id])
@@ -407,6 +536,13 @@ class TelegramBot:
         temp_file = f"/tmp/screen_output_{screen_id}.txt"
 
         try:
+            # Verifica se lo screen con l'ID specificato esiste
+            existing_screens = subprocess.check_output(["screen", "-list"]).decode('utf-8')
+            if screen_id not in existing_screens:
+                print(f"⚠️ La sessione screen con ID {screen_id} non esiste.")
+                self.bot.send_message(message.chat.id, f"⚠️ La sessione screen con ID {screen_id} non esiste.")
+                return
+
             # Cattura l'output della screen
             subprocess.run(
                 ["screen", "-X", "-S", screen_id, "hardcopy", "-h", temp_file],
@@ -440,51 +576,22 @@ class TelegramBot:
                 "\n\n", "\n"
             )  # Rimuovi newline multipli
 
-            # Estrarre tutte le parti da "Download:" fino a "Video" o "Subtitle", senza includerli
-            download_matches = re.findall(
-                r"Download: (.*?)(?:Video|Subtitle)", cleaned_output
-            )
-            if download_matches:
-                # Serie TV e Film StreamingCommunity
-
-                proc_matches = re.findall(r"Proc: ([\d\.]+%)", cleaned_output)
-
-                # Creare una stringa unica con tutti i risultati
-                result_string = "\n".join(
-                    [
-                        f"Download: {download_matches[i].strip()}\nDownload al {proc_matches[i]}"
-                        for i in range(len(download_matches))
-                        if i < len(proc_matches)
-                    ]
-                )
-
-                if result_string != "":
-                    cleaned_output = result_string
-                else:
-                    print(f"❌ La parola 'Download:' non è stata trovata nella stringa.")
-            else:
-
-                download_list = []
-
-                # Estrai tutte le righe che iniziano con "Download:" fino al prossimo "Download" o alla fine della riga
-                matches = re.findall(r"Download:\s*(.*?)(?=Download|$)", cleaned_output)
+            # Dentro cleaned_output c'è una stringa recupero quello che si trova tra ## ##
+            download_section = re.search(r"##(.*?)##", cleaned_output, re.DOTALL)
+            if download_section:
+                cleaned_output_0 = "Download: " + download_section.group(1).strip()
 
-                # Se sono stati trovati download, stampali
-                if matches:
-                    for i, match in enumerate(matches, 1):
-                        # rimuovo solo la parte "downloader.py:57Result:400" se esiste
-                        match = re.sub(r"downloader.py:\d+Result:400", "", match)
-                        match = match.strip()  # Rimuovo gli spazi bianchi in eccesso
-                        if match:  # Assicurati che la stringa non sia vuota
-                            print(f"Download {i}: {match}")
+            # Recupero tutto quello che viene dopo con ####
+            download_section_bottom = re.search(r"####(.*)", cleaned_output, re.DOTALL)
+            if download_section_bottom:
+                cleaned_output_1 = download_section_bottom.group(1).strip()
 
-                        # Aggiungi il risultato modificato alla lista
-                        download_list.append(f"Download {i}: {match}")
-
-                    # Creare una stringa unica con tutti i risultati
-                    cleaned_output = "\n".join(download_list)
-                else:
-                    print("❌ Nessun download trovato")
+            # Unico i due risultati se esistono
+            if cleaned_output_0 and cleaned_output_1:
+                cleaned_output = f"{cleaned_output_0}\n{cleaned_output_1}"
+                # Rimuovo 'segments.py:302' e 'downloader.py:385' se presente
+                cleaned_output = re.sub(r'downloader\.py:\d+', '', cleaned_output)
+                cleaned_output = re.sub(r'segments\.py:\d+', '', cleaned_output)
 
             # Invia l'output pulito
             print(f"📄 Output della screen {screen_id}:\n{cleaned_output}")
@@ -505,7 +612,16 @@ class TelegramBot:
         os.remove(temp_file)
 
     def send_message(self, message, choices):
-        if choices is None:
+
+        formatted_message = message
+        if choices:
+            formatted_choices = "\n".join(choices)
+            formatted_message = f"{message}\n\n{formatted_choices}"
+
+        for chat_id in self.authorized_users:
+            self.bot.send_message(chat_id, formatted_message)
+
+        """ if choices is None:
             if self.chat_id:
                 print(f"{message}")
                 self.bot.send_message(self.chat_id, message)
@@ -514,7 +630,7 @@ class TelegramBot:
             message = f"{message}\n\n{formatted_choices}"
             if self.chat_id:
                 print(f"{message}")
-                self.bot.send_message(self.chat_id, message)
+                self.bot.send_message(self.chat_id, message) """
 
     def _send_long_message(self, chat_id, text, chunk_size=4096):
         """Suddivide e invia un messaggio troppo lungo in più parti."""
@@ -527,16 +643,20 @@ class TelegramBot:
 
         if choices is None:
             print(f"{prompt_message}")
-            self.bot.send_message(
+            """ self.bot.send_message(
                 self.chat_id,
                 f"{prompt_message}",
-            )
+            ) """
+            for chat_id in self.authorized_users:  # Manda a tutti gli ID autorizzati
+                self.bot.send_message(chat_id, f"{prompt_message}")
         else:
             print(f"{prompt_message}\n\nOpzioni: {', '.join(choices)}")
-            self.bot.send_message(
+            """ self.bot.send_message(
                 self.chat_id,
                 f"{prompt_message}\n\nOpzioni: {', '.join(choices)}",
-            )
+            ) """
+            for chat_id in self.authorized_users:  # Manda a tutti gli ID autorizzati
+                self.bot.send_message(chat_id, f"{prompt_message}\n\nOpzioni: {', '.join(choices)}")
 
         start_time = time.time()
         while time.time() - start_time < timeout:
@@ -546,7 +666,8 @@ class TelegramBot:
             time.sleep(1)
 
         print(f"⚠️ Timeout: nessuna risposta ricevuta.")
-        self.bot.send_message(self.chat_id, "⚠️ Timeout: nessuna risposta ricevuta.")
+        for chat_id in self.authorized_users:  # Manda a tutti gli ID autorizzati
+            self.bot.send_message(chat_id, "⚠️ Timeout: nessuna risposta ricevuta.")
         self.request_manager.clear_file()
         return None
 
@@ -558,4 +679,37 @@ class TelegramBot:
 
 
 def get_bot_instance():
-    return TelegramBot.get_instance()
+    return TelegramBot.get_instance()
+
+# Esempio di utilizzo
+if __name__ == "__main__":
+
+    # Usa le variabili
+    token = os.getenv("TOKEN_TELEGRAM")
+    authorized_users = os.getenv("AUTHORIZED_USER_ID")
+
+    # Controlla se le variabili sono presenti
+    if not token:
+        print("Errore: TOKEN_TELEGRAM non è definito nel file .env.")
+        sys.exit(1)
+
+    if not authorized_users:
+        print("Errore: AUTHORIZED_USER_ID non è definito nel file .env.")
+        sys.exit(1)
+
+    try:
+        TOKEN = token  # Inserisci il token del tuo bot Telegram sul file .env
+        AUTHORIZED_USER_ID = list(map(int, authorized_users.split(",")))  # Inserisci il tuo ID utente Telegram sul file .env
+    except ValueError as e:
+        print(f"Errore nella conversione degli ID autorizzati: {e}. Controlla il file .env e assicurati che gli ID siano numeri interi separati da virgole.")
+        sys.exit(1)
+
+    # Inizializza il bot
+    bot = TelegramBot.init_bot(TOKEN, AUTHORIZED_USER_ID)
+    bot.run()
+
+"""
+start - Avvia lo script
+list - Lista script attivi
+get - Mostra ID utente Telegram
+"""

StreamingCommunity/Util/_jsonConfig.py

@@ -4,17 +4,18 @@ import os
 import sys
 import json
 import logging
+from pathlib import Path
 from typing import Any, List
 
 
 class ConfigManager:
-    def __init__(self, file_path: str = 'config.json') -> None:
+    def __init__(self, file_name: str = 'config.json') -> None:
         """Initialize the ConfigManager.
 
         Parameters:
             - file_path (str, optional): The path to the configuration file. Default is 'config.json'.
         """
-        self.file_path = file_path
+        self.file_path = Path(__file__).parent.parent.parent / file_name
         self.config = {}
         self.cache = {}
 
@@ -50,17 +51,17 @@ class ConfigManager:
     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
-            
+
             logging.info(f"{filename} not found locally. Downloading from {url}...")
             response = requests.get(url)
-            
+
             if response.status_code == 200:
                 with open(filename, 'wb') as f:
                     f.write(response.content)
@@ -68,7 +69,7 @@ class ConfigManager:
             else:
                 logging.error(f"Failed to download {filename}. HTTP Status code: {response.status_code}")
                 sys.exit(0)
-        
+
         except Exception as e:
             logging.error(f"Failed to download {filename}: {e}")
             sys.exit(0)
@@ -89,15 +90,15 @@ class ConfigManager:
 
         if cache_key in self.cache:
             return self.cache[cache_key]
-        
+
         if section in self.config and key in self.config[section]:
             value = self.config[section][key]
         else:
             raise ValueError(f"Key '{key}' not found in section '{section}'")
-        
+
         value = self._convert_to_data_type(value, data_type)
         self.cache[cache_key] = value
-        
+
         return value
 
     def _convert_to_data_type(self, value: str, data_type: type) -> Any:
@@ -120,7 +121,7 @@ class ConfigManager:
             return None
         else:
             return value
-        
+
     def get(self, section: str, key: str) -> Any:
         """Read a value from the configuration file.
 
@@ -144,7 +145,7 @@ class ConfigManager:
             int: The integer value.
         """
         return self.read_key(section, key, int)
-    
+
     def get_float(self, section: str, key: str) -> int:
         """Read an float value from the configuration file.
 
@@ -180,7 +181,7 @@ class ConfigManager:
             list: The list value.
         """
         return self.read_key(section, key, list)
-    
+
     def get_dict(self, section: str, key: str) -> dict:
         """Read a dictionary value from the configuration file.
 
@@ -220,7 +221,7 @@ class ConfigManager:
                 json.dump(self.config, f, indent=4)
         except Exception as e:
             print(f"Error writing configuration file: {e}")
-            
+
 
 # Initialize
 config_manager = ConfigManager()

StreamingCommunity/Util/ffmpeg_installer.py

@@ -4,8 +4,6 @@ import os
 import glob
 import gzip
 import shutil
-import tarfile
-import zipfile
 import logging
 import platform
 import subprocess
@@ -179,64 +177,59 @@ class FFMPEGDownloader:
                     size = file.write(chunk)
                     progress.update(download_task, advance=size)
             return True
+        
         except Exception as e:
             logging.error(f"Download error: {e}")
             return False
 
-    def _extract_and_copy_binaries(self, archive_path: str) -> Tuple[Optional[str], Optional[str], Optional[str]]:
+    def _extract_file(self, gz_path: str, final_path: str) -> bool:
         """
-        Extract FFmpeg binaries from the downloaded archive and copy them to the base directory.
+        Extract a gzipped file and set proper permissions.
 
         Parameters:
-            archive_path (str): Path to the downloaded archive file
+            gz_path (str): Path to the gzipped file
+            final_path (str): Path where the extracted file should be saved
 
         Returns:
-            Tuple[Optional[str], Optional[str], Optional[str]]: Paths to the extracted ffmpeg, 
-            ffprobe, and ffplay executables. Returns None for each executable that couldn't be extracted.
+            bool: True if extraction was successful, False otherwise
         """
         try:
-            extraction_path = os.path.join(self.base_dir, 'temp_extract')
-            os.makedirs(extraction_path, exist_ok=True)
-
-            if archive_path.endswith('.zip'):
-                with zipfile.ZipFile(archive_path, 'r') as zip_ref:
-                    zip_ref.extractall(extraction_path)
-            elif archive_path.endswith('.tar.xz'):
-                with tarfile.open(archive_path) as tar_ref:
-                    tar_ref.extractall(extraction_path)
-            elif archive_path.endswith('.gz'):
-                file_name = os.path.basename(archive_path)[:-3]  # Remove extension .gz
-                output_path = os.path.join(extraction_path, file_name)
-                with gzip.open(archive_path, 'rb') as f_in, open(output_path, 'wb') as f_out:
+            logging.info(f"Attempting to extract {gz_path} to {final_path}")
+            
+            # Check if source file exists and is readable
+            if not os.path.exists(gz_path):
+                logging.error(f"Source file {gz_path} does not exist")
+                return False
+                
+            if not os.access(gz_path, os.R_OK):
+                logging.error(f"Source file {gz_path} is not readable")
+                return False
+
+            # Extract the file
+            with gzip.open(gz_path, 'rb') as f_in:
+                # Test if the gzip file is valid
+                try:
+                    f_in.read(1)
+                    f_in.seek(0)
+                except Exception as e:
+                    logging.error(f"Invalid gzip file {gz_path}: {e}")
+                    return False
+
+                # Extract the file
+                with open(final_path, 'wb') as f_out:
                     shutil.copyfileobj(f_in, f_out)
-            else:
-                raise ValueError("Unsupported archive format")
-
-            config = FFMPEG_CONFIGURATION[self.os_name]
-            executables = config['executables']
-            found_paths = []
-
-            for executable in executables:
-                exe_paths = glob.glob(os.path.join(extraction_path, '**', executable), recursive=True)
-                if exe_paths:
-                    dest_path = os.path.join(self.base_dir, executable)
-                    shutil.copy2(exe_paths[0], dest_path)
-                    
-                    if self.os_name != 'windows':
-                        os.chmod(dest_path, 0o755)
-                    
-                    found_paths.append(dest_path)
-                else:
-                    found_paths.append(None)
 
-            shutil.rmtree(extraction_path, ignore_errors=True)
-            os.remove(archive_path)
-
-            return tuple(found_paths) if len(found_paths) == 3 else (None, None, None)
+            # Set executable permissions
+            os.chmod(final_path, 0o755)
+            logging.info(f"Successfully extracted {gz_path} to {final_path}")
+            
+            # Remove the gzip file
+            os.remove(gz_path)
+            return True
 
         except Exception as e:
-            logging.error(f"Extraction/copy error: {e}")
-            return None, None, None
+            logging.error(f"Extraction error for {gz_path}: {e}")
+            return False
 
     def download(self) -> Tuple[Optional[str], Optional[str], Optional[str]]:
         """
@@ -244,30 +237,43 @@ class FFMPEGDownloader:
 
         Returns:
             Tuple[Optional[str], Optional[str], Optional[str]]: Paths to ffmpeg, ffprobe, and ffplay executables.
-            Returns None for each executable that couldn't be downloaded or set up.
         """
         config = FFMPEG_CONFIGURATION[self.os_name]
         executables = [exe.format(arch=self.arch) for exe in config['executables']]
-        
+        successful_extractions = []
+
         for executable in executables:
-            download_url = f"https://github.com/eugeneware/ffmpeg-static/releases/latest/download/{executable}.gz"
-            download_path = os.path.join(self.base_dir, f"{executable}.gz")
-            final_path = os.path.join(self.base_dir, executable)
-            
-            console.print(f"[bold blue]Downloading {executable} from GitHub[/]")
-            if not self._download_file(download_url, download_path):
-                return None, None, None
-            
-            with gzip.open(download_path, 'rb') as f_in, open(final_path, 'wb') as f_out:
-                shutil.copyfileobj(f_in, f_out)
-            
-            os.chmod(final_path, 0o755)
-            os.remove(download_path)
-        
+            try:
+                download_url = f"https://github.com/eugeneware/ffmpeg-static/releases/latest/download/{executable}.gz"
+                download_path = os.path.join(self.base_dir, f"{executable}.gz")
+                final_path = os.path.join(self.base_dir, executable)
+                
+                # Log the current operation
+                logging.info(f"Processing {executable}")
+                console.print(f"[bold blue]Downloading {executable} from GitHub[/]")
+                
+                # Download the file
+                if not self._download_file(download_url, download_path):
+                    console.print(f"[bold red]Failed to download {executable}[/]")
+                    continue
+
+                # Extract the file
+                if self._extract_file(download_path, final_path):
+                    successful_extractions.append(final_path)
+                    console.print(f"[bold green]Successfully installed {executable}[/]")
+                else:
+                    console.print(f"[bold red]Failed to extract {executable}[/]")
+
+            except Exception as e:
+                logging.error(f"Error processing {executable}: {e}")
+                console.print(f"[bold red]Error processing {executable}: {str(e)}[/]")
+                continue
+
+        # Return the results based on successful extractions
         return (
-            os.path.join(self.base_dir, executables[0]),
-            os.path.join(self.base_dir, executables[1]),
-            None
+            successful_extractions[0] if len(successful_extractions) > 0 else None,
+            successful_extractions[1] if len(successful_extractions) > 1 else None,
+            None  # ffplay is not included in the current implementation
         )
 
 def check_ffmpeg() -> Tuple[Optional[str], Optional[str], Optional[str]]: