mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-07 20:15:24 +00:00
151 lines
5.2 KiB
Python
151 lines
5.2 KiB
Python
# 03.03.24
|
|
|
|
from rich.console import Console
|
|
from rich.table import Table
|
|
from rich.text import Text
|
|
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.
|
|
|
|
Args:
|
|
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.
|
|
|
|
Args:
|
|
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.
|
|
|
|
Args:
|
|
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.
|
|
|
|
Args:
|
|
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.
|
|
|
|
Args:
|
|
data_slice (List[Dict[str, Any]]): List of dictionaries containing TV show details to display.
|
|
"""
|
|
table = Table(title=Text("Show Details", justify="center", style="bold magenta"), 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:
|
|
row_data = [entry[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.
|
|
|
|
Args:
|
|
- force_int_input(bool): If True, only accept integer inputs from 0 to max_int_input
|
|
- max_int_input (int):
|
|
|
|
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]to restart, or [red]'q' [green]to quit.")
|
|
|
|
if not force_int_input:
|
|
key = Prompt.ask("[cyan]Insert media [red]index [yellow]or [red](*) [cyan]to download all media [yellow]or [red][1-2] [cyan]for a range of media")
|
|
else:
|
|
choices = [str(i) for i in range(0, max_int_input)]
|
|
choices.append("")
|
|
|
|
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.slice_end
|
|
self.slice_end += self.slice_end
|
|
if self.slice_end > total_items:
|
|
self.slice_end = total_items
|
|
|
|
else:
|
|
break
|
|
|
|
else:
|
|
self.console.print(f"\n\n[yellow][INFO] [red]You've reached the end. [green]Press [red]Enter [green]to restart, or [red]'q' [green]to quit.")
|
|
|
|
if not force_int_input:
|
|
key = Prompt.ask("[cyan]Insert media [red]index [yellow]or [red](*) [cyan]to download all media [yellow]or [red][1-2] [cyan]for a range of media")
|
|
else:
|
|
choices = [str(i) for i in range(0, max_int_input)]
|
|
choices.append("")
|
|
|
|
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
|