mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-05 02:55:25 +00:00
131 lines
3.9 KiB
Python
131 lines
3.9 KiB
Python
# 10.12.23
|
|
|
|
import sys
|
|
import os
|
|
import platform
|
|
import argparse
|
|
|
|
from typing import Callable
|
|
|
|
|
|
# Internal utilities
|
|
from Src.Util.message import start_message
|
|
from Src.Util.console import console, msg
|
|
from Src.Util._jsonConfig import config_manager
|
|
from Src.Upload.update import update as git_update
|
|
from Src.Util.os import get_system_summary
|
|
from Src.Util.logger import Logger
|
|
|
|
|
|
# Internal api
|
|
from Src.Api.Streamingcommunity import main_film_series as streamingcommunity_film_serie
|
|
from Src.Api.Animeunity import main_anime as streamingcommunity_anime
|
|
from Src.Api.Altadefinizione import main_film as altadefinizione_film
|
|
|
|
|
|
# Config
|
|
CLOSE_CONSOLE = config_manager.get_bool('DEFAULT', 'not_close')
|
|
|
|
|
|
def initialize():
|
|
"""
|
|
Initialize the application.
|
|
Checks Python version, removes temporary folder, and displays start message.
|
|
"""
|
|
|
|
# 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)
|
|
|
|
|
|
# Removing temporary folder
|
|
start_message()
|
|
|
|
|
|
# Attempting GitHub update
|
|
"""try:
|
|
git_update()
|
|
except Exception as e:
|
|
console.print(f"[blue]Req github [white]=> [red]Failed: {e}")"""
|
|
|
|
|
|
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.
|
|
"""
|
|
|
|
initialize()
|
|
|
|
if close_console:
|
|
while 1:
|
|
func()
|
|
else:
|
|
func()
|
|
|
|
|
|
def main():
|
|
|
|
log_not = Logger()
|
|
get_system_summary()
|
|
|
|
# Parse command line arguments
|
|
parser = argparse.ArgumentParser(description='Script to download film and series from the internet.')
|
|
parser.add_argument('-sa', '--streaming_anime', action='store_true', help='Check into anime category')
|
|
parser.add_argument('-sf', '--streaming_film', action='store_true', help='Check into film/tv series category')
|
|
parser.add_argument('-af', '--altadefinizione_film', action='store_true', help='Check into film/tv series category')
|
|
args = parser.parse_args()
|
|
|
|
# Mapping command-line arguments to functions
|
|
arg_to_function = {
|
|
'streaming_anime': streamingcommunity_anime,
|
|
'streaming_film': streamingcommunity_film_serie,
|
|
'altadefinizione_film': altadefinizione_film,
|
|
}
|
|
|
|
# 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, CLOSE_CONSOLE)
|
|
return
|
|
|
|
# Mapping user input to functions
|
|
input_to_function = {
|
|
'0': streamingcommunity_film_serie,
|
|
'1': streamingcommunity_anime,
|
|
'2': altadefinizione_film,
|
|
}
|
|
|
|
# Create dynamic prompt message and choices
|
|
choices = list(input_to_function.keys())
|
|
choice_labels = {
|
|
'0': "Film/Series",
|
|
'1': "Anime",
|
|
'2': "Altadefinizione"
|
|
}
|
|
prompt_message = "[cyan]Insert category [white](" + ", ".join(
|
|
f"[red]{key}[white]: [bold magenta]{label}[white]" for key, label in choice_labels.items()
|
|
) + ")[white]:[/cyan]"
|
|
|
|
# Ask the user for input
|
|
category = msg.ask(prompt_message, choices=choices, default="0")
|
|
|
|
# Run the corresponding function based on user input
|
|
if category in input_to_function:
|
|
run_function(input_to_function[category], CLOSE_CONSOLE)
|
|
else:
|
|
console.print("[red]Invalid category, you need to insert 0, 1, or 2.")
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |