o
    i|                     @   s@   d dl mZ ddlmZmZ G dd deZG dd deZdS )	   )InfoExtractor   )UnsupportedErrortraverse_objc                   @   sd   e Zd ZdZdZddddddd	d
ddd	ddidddddddddd
ddddidgZdd ZdS )TV2HuIEz
tv2play.huz9https?://(?:www\.)?tv2play\.hu/(?!szalag/)(?P<id>[^#&?]+)z6https://tv2play.hu/mintaapak/mintaapak_213_epizod_resz249240mp4u   Mintaapák - 213. epizódu
   Mintaapákit  z$md5:7350147e75485a59598e806c47967b07zre:^https?://.*\.jpg$20210825   )	idexttitleseriesdurationdescription	thumbnailrelease_dateepisode_numberskip_downloadT)url	info_dictparamszhttps://tv2play.hu/taxi_2 585e58e2e090f34603804bb2c48e98d8199363zTaxi 2i  z$md5:47762155dc9a50241797ded101b1b08c20210118)r   r   r   r   r   r   r   r   )r   md5r   r   c                    s&    |} d| |}|d dkr,t|ddtd} fdd|D } j||d	S |d d
kr6t|t|d }|d}|di } d| |d } ||}	 t|	d}
 	|
|\}}||d |d|d|dd|d |d
dd|d|d||dS )Nzhttps://tv2play.hu/api/search/contentTypeshowpage)pages.tabs.	ribbonIdsF)get_allexpected_typec                    s&   g | ]} j d | t |dqS )zhttps://tv2play.hu/szalag/)ievideo_id)
url_resultTV2HuSeriesIEie_key).0	ribbon_idself J/home/ubuntu/.local/lib/python3.10/site-packages/yt_dlp/extractor/tv2hu.py
<listcomp>3   s
    
z)TV2HuIE._real_extract.<locals>.<listcomp>playlist_idvideor   playerId
seriesInfoz.https://tv2play.hu/api/streaming-url?playerId=r   )bitrateshlsr   seriesTitlelengthr   zhttps://tv2play.huthumbnailUrl
uploadedAt. seasonNr	episodeNr)r   r   r   r   r   r   r   season_numberr   formats	subtitles)	_match_id_download_jsonr   listplaylist_resultr   strget_proto_relative_url#_extract_m3u8_formats_and_subtitlesreplace)r+   r   r$   	json_data
ribbon_idsentries	player_idseries_jsonvideo_json_url
video_jsonm3u8_urlr?   r@   r,   r*   r-   _real_extract-   s:   


zTV2HuIE._real_extractN__name__
__module____qualname__IE_NAME
_VALID_URL_TESTSrR   r,   r,   r,   r-   r      sB    "r   c                   @   s.   e Zd ZdZdZddddidgZdd	 Zd
S )r&   ztv2playseries.huz5https?://(?:www\.)?tv2play\.hu/szalag/(?P<id>[^#&?]+)z1https://tv2play.hu/szalag/59?rendezes=nepszerusegi  r   59)r   playlist_mincountr   c                 C   sj   |  |}| d| d|}g }|dg D ]}|d}|r-|| d| t| q| j||dS )Nzhttps://tv2play.hu/api/ribbons/z/0?size=100000cardsslugzhttps://tv2play.hu/r/   )rA   rB   rF   appendr%   r   rD   )r+   r   r0   rJ   rL   cardr$   r,   r,   r-   rR   ]   s   

zTV2HuSeriesIE._real_extractNrS   r,   r,   r,   r-   r&   Q   s    r&   N)commonr   utilsr   r   r   r&   r,   r,   r,   r-   <module>   s    I