iqxddlZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZGd d Zy) N)DictListAny)Console)Table)Prompt)box) start_messageceZdZdZdeeeeeffddfdZdeeefddfdZde eeefddfd Z dd e d e defd Z dd Zy) TVShowManagerc|t|_g|_d|_d|_|j|_g|_y)z? Initialize TVShowManager with default values. r N)rconsoletv_shows slice_start slice_endstep column_infoselfs [/home/nidran/Documenti/script/streaming-community/StreamingCommunity/utils/console/table.py__init__zTVShowManager.__init__s6y .0 NN rreturnNc||_y)z Add column information. Parameters: - column_info (Dict[str, Dict[str, str]]): Dictionary containing column names, their colors, and justification. N)r)rrs r add_columnzTVShowManager.add_column s 'rtv_showc@|r|jj|yy)z Add a TV show to the list of TV shows. Parameters: - tv_show (Dict[str, Any]): Dictionary containing TV show details. N)rappend)rrs r add_tv_showzTVShowManager.add_tv_show)s  MM  ) r data_slicec |stjdy|jstjdyttj dddd}|jj D]P\}}|jd d }|jd d }|jd d}|j||||Rt|D]f\}} | s |jjDcgc]}t| j|d} }|dzdk(rdnd } |j| d| ih|jj|y cc}w)z Display TV show data in a tabular format. Parameters: - data_slice (List[Dict[str, Any]]): List of dictionaries containing TV show details to display. zNothing to display.z)Error: Column information not configured.Tcyanblue)rr )r show_header header_style border_stylepaddingcolorwhitewidthNjustifycenter)styler.r-r dimr0)loggingerrorrrr ROUNDEDitemsgetr enumeratekeysstradd_rowrprint) rr"tablecol_name col_styler+r-r.idxentryrow_datar0s r display_datazTVShowManager.display_data3sP MM/ 0 MME F   $(#3#3#9#9#;  HiMM'73EMM'40EmmIx8G      $J/ 6JCIMIYIYI^I^I`aXC (B 78aa!$qA4 x5u5  6 5! bs6"Eforce_int_input max_int_inputc|js*tjdtjd|j stjdyt |j}d} t|j|j|j}|sJ|dkDrEd|_t|j||_ |j|j|j}|j|}|dk(rtjd|j|kr|jjd|sd}tj |}nHdgt#|dzD cgc] } t%| c} zd d gz} d }tj || d }|}|j'dvr |S|dk(rU|xj|jz c_|xj|jz c_ |j|kDr||_ n |S|jjd|sd}tj |}nHdgt#|dzD cgc] } t%| c} zd d gz} d }tj || d }|}|j'dvr |S|dk(rd|_|j|_ n |S[cc} wcc} w)aF Run the TV show manager application. Parameters: - force_int_input(bool): If True, only accept integer inputs from 0 to max_int_input - max_int_input (int): range of row to show Returns: str: Last command executed before breaking out of the loop. z%Error: No data available for display.rzError: Columns not configured.r1r$r zG [green]Press [red]Enter [green]for next page, [red]'q' [green]to quit.z [cyan]Insert media index [yellow](e.g., 1), [red]* [cyan]to download all media, [yellow](e.g., 1-2) [cyan]for a range of media, or [yellow](e.g., 3-*) [cyan]to download from a specific index to the endqquitz[cyan]Insert media [red]indexF)choices show_choices)rHrIzZ [green]You've reached the end. [red]Enter [green]for first page, [red]'q' [green]to quit.)rr4r5sysexitrlenr rrminrrDrr=raskranger;lower) rrErF total_items last_command current_slice result_func prompt_msgkeyirJs rrunzTVShowManager.runas}} MMA B HHQK MM: ;$--(   O!MM$*:*:4>>JM [1_#$ !$TYY !< $ d.>.>t~~ N ++M:Kc! ~~ + ""#mn&#pJ **Z0C "deMA&?'D'*4S>*d*,"tDcN';,","\V4VVCVp#rr )rLr4typingrrr rich.consoler rich.tabler rich.promptrrichr messager r rbrrris2 ""!#k#k#r