mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-03 10:00:10 +00:00
1c89398054
* Add ENABLE_VIDEO * Fix proxy * Add error proxy * Update config.json * Fix telegram_bot (#312) * Update config.json * Fix telegram_bot * fix bug * Fix StreamingCommunity site * Delete console.log * fix doppio string_to_search * Update __init__.py * Update site.py * Update config.json * Update site.py * Update config.json * Update __init__.py * Update __init__.py * Fix proxy (#319) * Add ENABLE_VIDEO * Fix proxy * Add error proxy * Update config.json * Refactor user input handling and improve messaging in __init__.py --------- Co-authored-by: None <62809003+Arrowar@users.noreply.github.com> Co-authored-by: l1n00 <> * Fix proxy __init__ * Update os.py --------- Co-authored-by: l1n00 <delmolinonicola@gmail.com>
102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
# 29.04.25
|
|
|
|
# External library
|
|
from rich.console import Console
|
|
from rich.prompt import Prompt
|
|
|
|
|
|
# Internal utilities
|
|
from StreamingCommunity.Api.Template import get_select_title
|
|
from StreamingCommunity.Lib.Proxies.proxy import ProxyFinder
|
|
from StreamingCommunity.Api.Template.config_loader import site_constant
|
|
from StreamingCommunity.Api.Template.Class.SearchType import MediaItem
|
|
|
|
|
|
# Logic class
|
|
from .site import title_search, table_show_manager, media_search_manager
|
|
from .film import download_film
|
|
from .series import download_series
|
|
|
|
|
|
# Variable
|
|
indice = 7
|
|
_useFor = "Film_&_Serie"
|
|
_priority = 0
|
|
_engineDownload = "hls"
|
|
_deprecate = False
|
|
|
|
msg = Prompt()
|
|
console = Console()
|
|
proxy = None
|
|
|
|
|
|
def get_user_input(string_to_search: str = None):
|
|
"""
|
|
Asks the user to input a search term.
|
|
Handles both Telegram bot input and direct input.
|
|
"""
|
|
string_to_search = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
|
|
return string_to_search
|
|
|
|
def process_search_result(select_title, selections=None, proxy=None):
|
|
"""
|
|
Handles the search result and initiates the download for either a film or series.
|
|
|
|
Parameters:
|
|
select_title (MediaItem): The selected media item
|
|
selections (dict, optional): Dictionary containing selection inputs that bypass manual input
|
|
{'season': season_selection, 'episode': episode_selection}
|
|
"""
|
|
if select_title.type == 'tv':
|
|
season_selection = None
|
|
episode_selection = None
|
|
|
|
if selections:
|
|
season_selection = selections.get('season')
|
|
episode_selection = selections.get('episode')
|
|
|
|
download_series(select_title, season_selection, episode_selection, proxy)
|
|
|
|
else:
|
|
download_film(select_title, proxy)
|
|
|
|
def search(string_to_search: str = None, get_onlyDatabase: bool = False, direct_item: dict = None, selections: dict = None):
|
|
"""
|
|
Main function of the application for search.
|
|
|
|
Parameters:
|
|
string_to_search (str, optional): String to search for
|
|
get_onlyDatabase (bool, optional): If True, return only the database object
|
|
direct_item (dict, optional): Direct item to process (bypass search)
|
|
selections (dict, optional): Dictionary containing selection inputs that bypass manual input
|
|
{'season': season_selection, 'episode': episode_selection}
|
|
"""
|
|
if direct_item:
|
|
select_title = MediaItem(**direct_item)
|
|
process_search_result(select_title, selections) # DONT SUPPORT PROXY FOR NOW
|
|
return
|
|
|
|
# Check proxy if not already set
|
|
finder = ProxyFinder(site_constant.FULL_URL)
|
|
proxy = finder.find_fast_proxy()
|
|
|
|
if string_to_search is None:
|
|
string_to_search = msg.ask(f"\n[purple]Insert a word to search in [green]{site_constant.SITE_NAME}").strip()
|
|
|
|
# Perform search on the database using the obtained query
|
|
finder = ProxyFinder(url=f"{site_constant.FULL_URL}/serie/euphoria/")
|
|
proxy = finder.find_fast_proxy()
|
|
len_database = title_search(string_to_search, proxy)
|
|
|
|
# If only the database is needed, return the manager
|
|
if get_onlyDatabase:
|
|
return media_search_manager
|
|
|
|
if len_database > 0:
|
|
select_title = get_select_title(table_show_manager, media_search_manager,len_database)
|
|
process_search_result(select_title, selections, proxy)
|
|
|
|
else:
|
|
# If no results are found, ask again
|
|
console.print(f"\n[red]Nothing matching was found for[white]: [purple]{string_to_search}")
|
|
search() |