mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-06 19:45:24 +00:00
353 lines
12 KiB
Python
353 lines
12 KiB
Python
# 10.12.23
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
import glob
|
||
import logging
|
||
import platform
|
||
import argparse
|
||
import importlib
|
||
import threading, asyncio
|
||
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 os_summary
|
||
from StreamingCommunity.Util.logger import Logger
|
||
|
||
|
||
# Telegram util
|
||
from StreamingCommunity.TelegramHelp.session import get_session, deleteScriptId
|
||
from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
|
||
|
||
|
||
# Config
|
||
CLOSE_CONSOLE = config_manager.get_bool('DEFAULT', 'not_close')
|
||
TELEGRAM_BOT = config_manager.get_bool('DEFAULT', 'telegram_bot')
|
||
|
||
|
||
|
||
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 = {}
|
||
|
||
# Lista dei siti da escludere se TELEGRAM_BOT è attivo
|
||
excluded_sites = {"cb01new", "ddlstreamitaly", "guardaserie", "ilcorsaronero", "mostraguarda"} if TELEGRAM_BOT else set()
|
||
|
||
# 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))
|
||
|
||
# Se il modulo è nella lista da escludere, saltalo
|
||
if module_name in excluded_sites:
|
||
continue
|
||
|
||
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.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()
|
||
except:
|
||
console.log("[red]Error with loading github.")
|
||
|
||
|
||
def restart_script():
|
||
"""Riavvia lo script con gli stessi argomenti della riga di comando."""
|
||
print("\n🔄 Riavvio dello script...\n")
|
||
python = sys.executable
|
||
os.execv(python, [python] + sys.argv)
|
||
|
||
|
||
def force_exit():
|
||
"""Forza la chiusura dello script in qualsiasi contesto."""
|
||
|
||
print("\n🛑 Chiusura dello script in corso...")
|
||
|
||
# 1️⃣ Chiudi tutti i thread tranne il principale
|
||
for t in threading.enumerate():
|
||
if t is not threading.main_thread():
|
||
print(f"🔄 Chiusura thread: {t.name}")
|
||
t.join(timeout=1)
|
||
|
||
# 2️⃣ Ferma asyncio, se attivo
|
||
try:
|
||
loop = asyncio.get_event_loop()
|
||
if loop.is_running():
|
||
print("⚡ Arresto del loop asyncio...")
|
||
loop.stop()
|
||
except RuntimeError:
|
||
pass
|
||
|
||
# 3️⃣ Esce con sys.exit(), se fallisce usa os._exit()
|
||
try:
|
||
print("✅ Uscita con sys.exit(0)")
|
||
sys.exit(0)
|
||
except SystemExit:
|
||
pass
|
||
|
||
print("🚨 Uscita forzata con os._exit(0)")
|
||
os._exit(0)
|
||
|
||
|
||
def main(script_id):
|
||
|
||
if TELEGRAM_BOT:
|
||
bot = get_bot_instance()
|
||
bot.send_message(f"🏁 Avviato script {script_id}", None)
|
||
|
||
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 argument parser
|
||
parser = argparse.ArgumentParser(
|
||
description='Script to download movies and series from the internet. Use these commands to configure the script and control its behavior.'
|
||
)
|
||
|
||
parser.add_argument("script_id", nargs="?", default="unknown", help="ID dello script")
|
||
|
||
# Add arguments for the main configuration parameters
|
||
parser.add_argument(
|
||
'--add_siteName', type=bool, help='Enable or disable adding the site name to the file name (e.g., true/false).'
|
||
)
|
||
parser.add_argument(
|
||
'--disable_searchDomain', type=bool, help='Enable or disable searching in configured domains (e.g., true/false).'
|
||
)
|
||
parser.add_argument(
|
||
'--not_close', type=bool, help='If set to true, the script will not close the console after execution (e.g., true/false).'
|
||
)
|
||
|
||
# Add arguments for M3U8 configuration
|
||
parser.add_argument(
|
||
'--default_video_worker', type=int, help='Number of workers for video during M3U8 download (default: 12).'
|
||
)
|
||
parser.add_argument(
|
||
'--default_audio_worker', type=int, help='Number of workers for audio during M3U8 download (default: 12).'
|
||
)
|
||
|
||
# Add options for audio and subtitles
|
||
parser.add_argument(
|
||
'--specific_list_audio', type=str, help='Comma-separated list of specific audio languages to download (e.g., ita,eng).'
|
||
)
|
||
parser.add_argument(
|
||
'--specific_list_subtitles', type=str, help='Comma-separated list of specific subtitle languages to download (e.g., eng,spa).'
|
||
)
|
||
|
||
# Add arguments for search functions
|
||
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()
|
||
|
||
# Map command-line arguments to the config values
|
||
config_updates = {}
|
||
|
||
if args.add_siteName is not None:
|
||
config_updates['DEFAULT.add_siteName'] = args.add_siteName
|
||
if args.disable_searchDomain is not None:
|
||
config_updates['DEFAULT.disable_searchDomain'] = args.disable_searchDomain
|
||
if args.not_close is not None:
|
||
config_updates['DEFAULT.not_close'] = args.not_close
|
||
if args.default_video_worker is not None:
|
||
config_updates['M3U8_DOWNLOAD.default_video_worker'] = args.default_video_worker
|
||
if args.default_audio_worker is not None:
|
||
config_updates['M3U8_DOWNLOAD.default_audio_worker'] = args.default_audio_worker
|
||
if args.specific_list_audio is not None:
|
||
config_updates['M3U8_DOWNLOAD.specific_list_audio'] = args.specific_list_audio.split(',')
|
||
if args.specific_list_subtitles is not None:
|
||
config_updates['M3U8_DOWNLOAD.specific_list_subtitles'] = args.specific_list_subtitles.split(',')
|
||
|
||
# Apply the updates to the config file
|
||
for key, value in config_updates.items():
|
||
section, option = key.split('.')
|
||
config_manager.set_key(section, option, value)
|
||
|
||
config_manager.write_config()
|
||
|
||
# Map 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])"
|
||
|
||
if TELEGRAM_BOT:
|
||
|
||
# Mappa delle emoji per i colori
|
||
emoji_map = {
|
||
"yellow": "🟡", # Giallo
|
||
"red": "🔴", # Rosso
|
||
"blue": "🔵", # Blu
|
||
"green": "🟢" # Verde
|
||
}
|
||
|
||
# Display the category legend in a single line
|
||
category_legend_str = "Categorie: \n" + " | ".join([
|
||
f"{emoji_map.get(color, '⚪')} {category.capitalize()}"
|
||
for category, color in color_map.items()
|
||
])
|
||
|
||
# Costruisci il messaggio con le emoji al posto dei colori
|
||
prompt_message = "Inserisci il sito:\n" + "\n".join(
|
||
[f"{key}: {emoji_map[color_map[label[1]]]} {label[0]}" for key, label in choice_labels.items()]
|
||
)
|
||
|
||
console.print(f"\n{prompt_message}")
|
||
|
||
# Chiedi la scelta all'utente con il bot Telegram
|
||
category = bot.ask(
|
||
"select_provider",
|
||
f"{category_legend_str}\n\n{prompt_message}",
|
||
None # Passiamo la lista delle chiavi come scelte
|
||
)
|
||
|
||
else:
|
||
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:
|
||
|
||
if TELEGRAM_BOT:
|
||
bot.send_message(f"Categoria non valida", None)
|
||
|
||
console.print("[red]Invalid category.")
|
||
|
||
if CLOSE_CONSOLE:
|
||
restart_script() # Riavvia lo script invece di uscire
|
||
else:
|
||
force_exit() # Usa la funzione per chiudere sempre
|
||
|
||
if TELEGRAM_BOT:
|
||
bot.send_message(f"Chiusura in corso", None)
|
||
|
||
# Delete script_id
|
||
script_id = get_session()
|
||
if script_id != "unknown":
|
||
deleteScriptId(script_id)
|