i&%ddlZddlZddlmZddlmZddlmZddlm Z ddl m Z eZ eZ d dedefd Zd Zd Zd Zy)N)Console)Prompt)Table) start_message)load_search_functions search_termsselected_sitesc H t}i}|#tjdj}i}|j D]'\}\}}||vrg||<||j ||f)|t jdt jdt jdt jdtjdgdd }|d k(rt|j}n|d k(rt jd t|jd D].\} }t jd| d|j0tjdd } | jdD cgc]8} t|jt| jd z :} } g}| D]-}|j|gD]\}} |j |/nt jdt|j d D]E\} \}} |jddj}t jd| d|Gtjdd }|jdD cgc]} t| jd z  }} |D cgc]-} | t|kst|j| /}} t jd|t jdt|d|D]}|jddj}t jd|||\}} ||d}|rt!|drt|j"dkDrg||<|j"D]I}t!|dr|j$j'}ni}||d <||d!<||j |Kt jd"t|j"d#|rYg}|j D]\}}|D]}|j |t-||t/|}|r t1|||St jd&|tjd'd(d)gd( d(k(r t3|Scc} wcc} wcc} w#t($r/}t jd$|d%t+|Yd}~d}~wwxYw)*au Perform a search across multiple sites based on selection. Parameters: search_terms (str, optional): The terms to search for. If None, will prompt the user. selected_sites (list, optional): List of site aliases to search. If None, will search all sites. Returns: dict: Consolidated search results from all searched sites. Nz- [purple]Enter search terms for global searchz [green]Select sites to search:z[cyan]1. Search all siteszyan]2. Search by categoryz[cyan]3. Select specific sitesz[green]Enter your choice (1-3))123r )choicesdefaultr z$ [green]Select categories to search:z[cyan]z. z1[green]Enter category numbers separated by commas)r,z( [green]Select specific sites to search:_r.z-[green]Enter site numbers separated by commasz [green]Searching for: [yellow]z[green]Searching across: z sites z[cyan]Search url in: T)get_onlyDatabase media_list__dict__source source_aliasz[green]Found result:  z[red]Error searching : z$ [red]No results found for: [yellow]z[green]Search again? (y/n)yn)rmsgaskstripitemsappendconsoleprintlistkeys enumerate capitalizesplitintgetlenhasattrrrcopy Exceptionstrdisplay_consolidated_results select_from_consolidated_resultsprocess_selected_item global_search)rr search_functions all_resultssites_by_categoryaliasfunccategorychoiceicategory_choicescselected_categoriesr site_name site_choicesselected_indicesdatabaseelement item_dicteall_media_itemsresultsitem selected_items a/home/nidran/Documenti/script/streaming-community/StreamingCommunity/cli/command/global_search.pyr3r3s-.KwwOPVVX #3#9#9#;:h , ,*, h '(#**E4=9:  89 12 34 679?\_` S=!"2"7"7"9:N s] MMA B():)?)?)A1E E 8 qcH,?,?,A+BCD E #ww'Zdgwh ]m]s]stw]x"yXY4(9(>(>(@#A#aggi.QRBR#S"y "yN/ 1 1 5 5h C1HE1"))%01 1 MME F!*+;+A+A+CQ!G 8 :E1!KK,Q/::<  qc9+67 877#R\_7`L:F:L:LS:QRQAGGIq 0R RHXv1\]`cdt`u\ud#3#8#8#:;A>vNv MM4\NCD MM-c..A-B)LM IKK$Q'224  -i[9:"5)a IL4@HGHl;HDWDW@X[\@\%' E"'22 9Gw 3$+$4$4$9$9$; $& +4Ih'05In-&--i8 9 5c(:M:M6N5OrRS3I>)//1 -NE7 -&&t, - - %_lC9I  !-1A B   =l^LM 77/#sS7 QUX X O [#z SvD I MM1)Bs1vhG H H Is1=S##S S$ S$B9S)) T!2$TT!c tjdttj d|dt dd}|j ddd |j d d |j ddtd|D}|r|j dd|j ddt|dD]\}}|jd|jdd}|jd|jdd}|jdd}|rt|jddnd} |r:|jt|t|t|| t||jt|t|t|t|tj |y)z Display consolidated search results from multiple sites. Parameters: all_media_items (list): List of media items from all searched sites. search_terms (str): The search terms used. rz$ [green]Search results for: [yellow]z Tcyan) show_header header_style#dim)stylewidthTitle) min_widthType)rSc32K|]}d|vxr|dyw)yearN).0rHs rJ z/display_consolidated_results..s Ot6T>2d6l2OsYearSourcetitlenameUnknowntype media_typerrZN) timesleeprr"r#r add_columnanyr&r*r/add_row) rFrtablehas_yearr;rHrbrfrrZs rJr0r0s| JJqMO MM9,sKL d 8E SQ/ W+ V2&OOOH q) XR(_a0L4$((69"=>XXfdhh|Y&GH (I.,4s488FB'($  MM#a&#e*c*otS[ Q MM#a&#e*c*os6{ KL MM%c |syt|}tjd|dt|dzDcgc] }t |c}dd}|dk(ry|t |dz Scc}w) z Allow user to select an item from consolidated results. Parameters: all_media_items (list): List of media items from all searched sites. Returns: dict: The selected media item or None if no selection was made. Nz[green]Select item # (1-z) or 0 to cancelrr F)rr show_choices0)r+rrranger/r))rF max_indexr;r:s rJr1r1sy O$I WW "9+-=>!&y1}!56AQ6 F} 3v;? ++7sA& c |jd}|r||vrtjdy||\}}tjd|jdd}dD]}|j|}|sn|jd|jdd }|jd |jd d }|r*tjd |d|d|d ||ytjd|jD] } tjd| d|| "y#t$r@} tjdt | t j dYd} ~ yd} ~ wwxYw)z Process the selected item - download the media using the appropriate site API. Parameters: selected_item (dict): The selected media item. search_functions (dict): Dictionary of search functions by alias. rzB[red]Error: Cannot process this item - source information missing.Nz# [green]Processing selection from: r)idmedia_idIDitem_idurlrerfunknownrbrcrdz[green]Selected item: z (ID: z, Type: )) direct_itemz [red]Error processing download: zDownload processing errorz0[red]Error: Item ID not found. Available fields:z [yellow]- r)r*r"r#r.r/logging exceptionr%) rIr4rr8rryid_field item_type item_titlerEkeys rJr2r2s!$$^4L ##H-   !!&-*;*;L)*TUI""7M,=,=fi,PQJ .zl& R[Q\\]^_ ; ] +  HI %%' DC MMJse2mC.@-AB C D  ; MMrsY !CC ) hyyDyvB,4)Dro