U
    g&                     @   sT   d dl mZ ddlmZmZmZmZmZmZm	Z	 G dd deZ
G dd deZdS )	   )InfoExtractor   )ExtractorErrorfloat_or_noneint_or_nonejoin_nonemptyparse_iso8601	qualitiestry_getc                   @   sR   e Zd ZdZdZdgZdddddd	Zd
dddddZdd Zdd Z	dd Z
dS )SRGSSRIEa  (?x)
                    (?:
                        https?://tp\.srgssr\.ch/p(?:/[^/]+)+\?urn=urn|
                        srgssr
                    ):
                    (?P<bu>
                        srf|rts|rsi|rtr|swi
                    ):(?:[^:]+:)?
                    (?P<type>
                        video|audio
                    ):
                    (?P<id>
                        [0-9a-f\-]{36}|\d+
                    )
                    FZCHz_To protect children under the age of 12, this video is only available between 8 p.m. and 6 a.m.z`To protect children under the age of 18, this video is only available between 11 p.m. and 5 a.m.z?For legal reasons, this video is only available in Switzerland.z2The video cannot be transmitted for legal reasons.z8This video is not yet available. Please try again later.)ZAGERATING12ZAGERATING18GEOBLOCKZLEGALZ	STARTDATEdefritZrmen)ZsrfZrtsZrsiZrtrZswic                 C   sL   | j d|d| dddpi }t|dd }|rH|d|kr>dnd	| 7 }|S )
Nz%http://tp.srgssr.ch/akahd/token?acl=*zDownloading z tokenF)fatalc                 S   s   | d d S )NtokenZ
authparams )xr   r   ;/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/extractor/srgssr.py<lambda>4       z-SRGSSRIE._get_tokenized_src.<locals>.<lambda>?&)_download_jsonr
   )selfurlZvideo_id	format_idr   Zauth_paramsr   r   r   _get_tokenized_src0   s     
 zSRGSSRIE._get_tokenized_srcc           	         s   |dkrddini }| j d| d| d  d |dd	 }zt fd
d|D }W n tk
rr   tdY nX |d}|r|| jkr| j| }|dkr| j|| jd t| j d| dd|S )NvideoZonlyChaptersTz*https://il.srgssr.ch/integrationlayer/2.0/z/mediaComposition//z.json)queryZchapterListc                 3   s    | ]}| d  kr|V  qdS )idN)get).0r   media_idr   r   	<genexpr>?   s     z+SRGSSRIE._get_media_data.<locals>.<genexpr>zNo media information foundZblockReasonr   )msgZ	countriesz said: )expected)	r   nextStopIterationr   r#   _ERRORSZraise_geo_restricted_GEO_COUNTRIESZIE_NAME)	r   bu
media_typer&   r!   Zfull_media_data
media_dataZblock_reasonmessager   r%   r   _get_media_data9   s6     

  zSRGSSRIE._get_media_datac              	   C   s(  |  | \}}}| |||}|d }g }i }tddg}	|dpJg D ]}
|
d}|s`qL|
d}|
d}t||
d|}|d	kr|
d
dkr| |||}| ||\}}|| | 	||}n:|dkr.| j
||dd|dd\}}|| | 	||}qL|dkrL||||	|d qLt|ddkrdD ]D}|d| d}|shqH|d }|d| ||	|d qH|dkr|dpg D ]F}|d}|sq|dp| j| }||g d|i q|||dt|d|dt|d d!||d"S )#NtitleZSDZHDZresourceListr   protocolqualityencoding)ZHDSHLS	tokenTypeZAKAMAIr7   mp4Zm3u8_nativeF)Zm3u8_idr   )HTTPHTTPS)r   r   r5   position    )SHZpodcastZdUrlDzPODCAST-r   ZsubtitleListlocaledescriptiondateZimageUrldurationi  )r"   r3   rB   	timestamp	thumbnailrD   	subtitlesformats)_match_valid_urlgroupsr2   r	   r#   r   r   Z%_extract_akamai_formats_and_subtitlesextendZ_merge_subtitlesZ#_extract_m3u8_formats_and_subtitlesappendr   _DEFAULT_LANGUAGE_CODES
setdefaultr   r   )r   r   r.   r/   r&   r0   r3   rH   rG   qsourceZ
format_urlr4   r5   r   ZfmtssubsZ	m3u8_fmtsZ	m3u8_subspZpodcast_urlsubZsub_urllangr   r   r   _real_extractO   s    



   

    


	

 
zSRGSSRIE._real_extractN)__name__
__module____qualname__
_VALID_URLZ_GEO_BYPASSr-   r,   rM   r   r2   rU   r   r   r   r   r      s$   		r   c                   @   s   e Zd ZdZdZddddddd	d
dddg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#d$d%dd&d'd(d)d*d+d,d-	ddidd.dd/d0dd/d1dd/d2dd/gZd3d4 Zd5S )6SRGSSRPlayIEz:srf.ch, rts.ch, rsi.ch, rtr.ch and swissinfo.ch play sitesa  (?x)
                    https?://
                        (?:(?:www|play)\.)?
                        (?P<bu>srf|rts|rsi|rtr|swissinfo)\.ch/play/(?:tv|radio)/
                        (?:
                            [^/]+/(?P<type>video|audio)/[^?]+|
                            popup(?P<type_2>video|audio)player
                        )
                        \?.*?\b(?:id=|urn=urn:[^:]+:video:)(?P<id>[0-9a-f\-]{36}|\d+)
                    zrhttp://www.srf.ch/play/tv/10vor10/video/snowden-beantragt-asyl-in-russland?id=28e1a57d-5b76-4399-8ab3-9097f071e6c5Z 6db2226ba97f62ad42ce09783680046cz$28e1a57d-5b76-4399-8ab3-9097f071e6c5r9   Z20130701z"Snowden beantragt Asyl in RusslandiwQgrht\@zre:^https?://.*1383719781\.png$)r"   extupload_dater3   rE   rD   rF   zUnable to download f4m manifest)r   md5	info_dictZexpected_warningszhttp://www.rtr.ch/play/radio/actualitad/audio/saira-tujetsch-tuttina-cuntinuar-cun-sedrun-muster-turissem?id=63cb0778-27f8-49af-9284-8c7a8c6d15fcz$63cb0778-27f8-49af-9284-8c7a8c6d15fcZmp3Z20151013u?   Saira: Tujetsch - tuttina cuntinuar cun Sedrun Mustér Turissemi(VgVu@)r"   r[   r\   r3   rE   rD   Zskip_downloadT)r   r^   paramsz5http://www.rts.ch/play/tv/-/video/le-19h30?id=6348260Z 67a2a9ae4e8e62a68d0e9820cc9782dfZ6348260gףp=
@zLe 19h30Z20141201i|Tzre:^https?://.*\.image)r"   Z
display_idr[   rD   r3   r\   rE   rF   )r   r]   r^   r_   z_http://play.swissinfo.ch/play/tv/business/video/why-people-were-against-tax-reforms?id=42960270Z42960270z#Why people were against tax reformsz$md5:7ac442c558e9630e947427469c4b824dg     W@Z20170215iwXz8re:https?://www\.swissinfo\.ch/srgscalableimage/42961964zcount:9)	r"   r[   r3   rB   rD   r\   rE   rF   rG   zShttps://www.srf.ch/play/tv/popupvideoplayer?id=c4dba0ca-e75b-43b2-a34f-f708a4932e01)r   Zonly_matchingzhttps://www.srf.ch/play/tv/10vor10/video/snowden-beantragt-asyl-in-russland?urn=urn:srf:video:28e1a57d-5b76-4399-8ab3-9097f071e6c5zIhttps://www.rts.ch/play/tv/19h30/video/le-19h30?urn=urn:rts:video:6348260zVhttps://www.srf.ch/play/radio/popupaudioplayer?id=50b20dc8-f05b-4972-bf03-e438ff2833ebc                 C   sX   |  |}|d}|dp&|d}|d}| d|d d  d| d| dS )	Nr.   typeZtype_2r"   zsrgssr:   :ZSRGSSR)rI   groupZ
url_result)r   r   Zmobjr.   r/   r&   r   r   r   rU      s
    


zSRGSSRPlayIE._real_extractN)rV   rW   rX   ZIE_DESCrY   Z_TESTSrU   r   r   r   r   rZ      s   	
   KrZ   N)commonr   utilsr   r   r   r   r   r	   r
   r   rZ   r   r   r   r   <module>   s   $ 