U
    g|                     @   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_reszZ249240Zmp4u   Mintaapák - 213. epizódu
   Mintaapákit  z$md5:7350147e75485a59598e806c47967b07zre:^https?://.*\.jpg$Z20210825   )	idexttitleseriesdurationdescription	thumbnailrelease_dateepisode_numberZskip_downloadT)url	info_dictparamszhttps://tv2play.hu/taxi_2Z 585e58e2e090f34603804bb2c48e98d8Z199363zTaxi 2i  z$md5:47762155dc9a50241797ded101b1b08cZ20210118)r   r	   r
   r   r   r   r   r   )r   md5r   r   c                    s&    |} d| |}|d dkrXt|ddtd} fdd|D } j||d	S |d d
krlt|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/contentTypeZshowpage)Zpages.tabs.Z	ribbonIdsF)get_allZexpected_typec                    s&   g | ]} j d | t |dqS )zhttps://tv2play.hu/szalag/)Zievideo_id)
url_resultTV2HuSeriesIEZie_key).0Z	ribbon_idself :/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/extractor/tv2hu.py
<listcomp>3   s
    z)TV2HuIE._real_extract.<locals>.<listcomp>playlist_idvideor   ZplayerIdZ
seriesInfoz.https://tv2play.hu/api/streaming-url?playerId=r   )ZbitratesZhlsr
   ZseriesTitlelengthr   zhttps://tv2play.huZthumbnailUrlZ
uploadedAt. ZseasonNrZ	episodeNr)r   r
   r   r   r   r   r   Zseason_numberr   formats	subtitles)	_match_id_download_jsonr   listplaylist_resultr   strgetZ_proto_relative_urlZ#_extract_m3u8_formats_and_subtitlesreplace)r   r   r   	json_dataZ
ribbon_idsentriesZ	player_idZseries_jsonZvideo_json_urlZ
video_jsonZm3u8_urlr'   r(   r   r   r   _real_extract-   s:    


zTV2HuIE._real_extractN__name__
__module____qualname__ZIE_NAMEZ
_VALID_URLZ_TESTSr2   r   r   r   r   r      sD     "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   Z59)r   Z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=100000ZcardsZslugzhttps://tv2play.hu/r!   )r)   r*   r.   appendr   r   r,   )r   r   r"   r0   r1   Zcardr   r   r   r   r2   ]   s    

  zTV2HuSeriesIE._real_extractNr3   r   r   r   r   r   Q   s    r   N)commonr   utilsr   r   r   r   r   r   r   r   <module>   s   I