iXddlZddlZddlmZddlmZmZddlmZm Z m Z GddZ y)N) BeautifulSoup) create_client get_headers) SeasonManagerEpisodeSeasonc neZdZddedededefdZddZd eddfd Zdefd Zd ede fd Z d edede fdZ y) GetSerieInfoNmedia_id series_nameyearprovider_languagecd|_t|_||_||_||_t |_||_|d|_||_ yy)a/ Initialize the GetSerieInfo class for scraping TV series information. Args: - url (str): The URL of the streaming site. - media_id (int, optional): Unique identifier for the media - series_name (str, optional): Name of the TV series FNT) is_seriesrheadersurlr r rseasons_managerrr )selfrr r r rs l/home/nidran/Documenti/script/streaming-community/StreamingCommunity/services/streamingcommunity/scrapper.py__init__zGetSerieInfo.__init__sU"}    ,!2  "!DN*D  #returnc  t|jj|jd|jd|j }|j t|jd}tj|jdddijd}|d |_ |jd ijd i}||_ |jd g}|D]i}|jjt!|jd|jd d|jd |jdky#t"$r}t%j&d|d}~wwxYw)z Retrieve general information about the TV series from the streaming site. Raises: Exception: If there's an error fetching series information r/titles/-z html.parserdividappz data-pageversionpropstitleseasonsnumberSeason slug)rr$namer&zError collecting series info: N)rrgetrr r raise_for_statusrtextjsonloadsfindr title_inforaddr Exceptionloggingerror)rresponsesoup json_response title_data seasons_data season_dataes rcollect_info_titlezGetSerieInfo.collect_info_title&s_ $T\\:>>$((8TXTaTaSbbcdhdtdtcu?vwH  % % '! >D JJtyyu 'F'J'J;'WXM(3DL'**7B7;;GRHJ)DO&>>)R8L+  $$(("t,&??84";??8#<"=>$0 *   MM:1#> ?  sEE F&E??F number_seasonc < |jj|}|stjd|dy|jj }|j d|jdt|j|jd|jd|jd |}|jjd ijd ijd g}|D]v}|jjt!|jd |jd |jd|jd|jdxy#t"$r!}tjd|d|d}~wwxYw)a Retrieve episode information for a specific season. Args: number_season (int): Season number to fetch episodes for Raises: Exception: If there's an error fetching episode information r% not foundNtrue)z x-inertiazx-inertia-versionrrrz/season-r! loadedSeasonepisodesrr$r'duration)rvideo_idr$r'rAz%Error collecting episodes for season z: )rget_season_by_numberr1r2rcopyupdater rr(rr r r+r@r/rr0)rr;seasoncustom_headersr3r5epr9s rcollect_info_seasonz GetSerieInfo.collect_info_seasonJs ))>>}MF  jAB!\\..0N  ! !#%)\\# %^<@@DHH:XVZVcVcUddefjfvfvewwAN@OBPQH%MMO//<@@QSTXXYceghM$ ##Gvvd|VVD\66(+VVJ/ %   MMA-PRSTRUV W  s6E1D7E11 F:FFc|jjs|jt|jjS)zK Get the total number of seasons available for the series. )rr#r:len)rs rgetNumberSeasonzGetSerieInfo.getNumberSeasonus6##++  # # %4''//00r season_numberc|jj|}|stjd|dgS|jjs|j ||jjS)z9 Get all episodes for a specific season. r%r=)rrCr1r2r@rI)rrMrFs rgetEpisodeSeasonszGetSerieInfo.getEpisodeSeasons~sb%%::=I MMGM?*= >I''  $ $] 3'''r episode_indexc|j|}|r|dks|t|k\rtjd|d|y||S)zN Get information for a specific episode in a specific season. rzEpisode index z is out of range for season N)rOrKr1r2)rrMrPr@s r selectEpisodezGetSerieInfo.selectEpisodesS))-8=1, X0N MMN=/9UVcUde f &&r)NNNit)rN) __name__ __module__ __qualname__intstrrr:rIrLlistrOrrRrrr r sx+c+s+QT+ps+*"H'''V11 (s (t ( '3 's 'w 'rr ) r+r1bs4r$StreamingCommunity.utils.http_clientrr(StreamingCommunity.services._base.objectrrrr rZrrr^s* LSSF'F'r