a
    hw+                     @   sp   d dl Z d dlZd dlZddlmZ ddlmZ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 )
    N   )InfoExtractor   )ExtractorErrorfloat_or_noneint_or_nonetraverse_objtry_getunified_timestampurl_or_noneurljoinc                   @   s   e Zd ZdZdddddddd	d
dddddddddddddddg dd	dddddddd d!d"g d#d$d%d&dd'dd(id%d)dd*dd(id%d+dd,dd(igd-d.d/d0dd1d2d3d4d5g d6d7d8id9gZd:d; Zd<S )=SohuIEzJhttps?://(?P<mytv>my\.)?tv\.sohu\.com/.+?/(?(mytv)|n)(?P<id>\d+)\.shtml.*?z.This video is available only in Mainland Chinaz2http://tv.sohu.com/20130724/n382479172.shtml#superZ	382479172mp4u&   MV：Far East Movement《The Illest》)idexttitlezOn available in China)noteurl	info_dictskipz,http://tv.sohu.com/20150305/n409385080.shtmlZ	409385080u@   《2015湖南卫视羊年元宵晚会》唐嫣《花好月圆》zno longer available)r   r   r   z1http://my.tv.sohu.com/us/232799889/78693464.shtml786934645   【爱范品】第31期：MWC见不到的奇葩手机   爱范儿视频   T20150305uhttp://e3f49eaa46b57.cdn.sohucs.com//group1/M10/83/FA/MTAuMTAuODguODA=/6_14cbccdde5eg104SysCutcloud_78693464_7_0b.jpgu	   爱范儿u	   爱范品ZMWCu   手机	r   r   r   uploaderduration	timestampupload_date	thumbnailtagsr   r   Multipart videoz/http://my.tv.sohu.com/pl/8384802/78910339.shtml78910339=   【神探苍实战秘籍】第13期 战争之影 赫卡里姆
   小苍cany     @@$U20150313uhttp://e3f49eaa46b57.cdn.sohucs.com//group1/M11/89/57/MTAuMTAuODguODA=/6_14cea022a1dg102SysCutcloud_78910339_8_0b.jpgu   小苍MMu   英雄联盟u   实战秘籍r   r   r   r    r!   r"   r#   r$   r   Z78910339_part1i&  )r   r   r    r   Z78910339_part2i,  Z78910339_part3   )r   r   r   playlistz Video with title containing dashz1http://my.tv.sohu.com/us/249884221/78932792.shtmlZ78932792zyoutube-dl testing videoih  iLZUZ20150314zuhttp://e3f49eaa46b57.cdn.sohucs.com//group1/M02/8A/00/MTAuMTAuODguNzk=/6_14cee1be192g102SysCutcloud_78932792_7_7b.jpg)r   r   r   r    r!   r"   r#   r$   Zskip_downloadT)r   r   r   paramsc                    sp  dDfdd	} |}|d|dd u}|}tdd|}d|d	}|||}|d
 dkr|ddkrtj	 dddn
j	 d i }	dD ]@}
|d |
 d}|sqt|}||kr|n||||	|
< q|d d }g }t|D ] g }|	 D ]^\}
}|d }|d }t|d thfdd}|sdtd  |d }d}d }d}d|v r6d||  ddd}|d ur||d < d!|
 d" d  d#| }|dkr|d$| d%7 }d&| d'tj| |}|d( }|d)}|d7 }|d*krxtd+qx|||
tt| fd,d-t|d.t|d/t|d0d1 q | d2 d  ||d d3   |d4 qt|dkr|d }|d< nd5||t|dd6thfd7}|rtjd8|d9dd:}nt|d;thf}d<|r0|d= nd it|dd>thfd?d@thfddAthfddBtjhfdC|S )ENFc                    s.   |r
d}nd} j ||  d|    dS )Nz+http://my.tv.sohu.com/play/videonew.do?vid=z-http://hot.vrs.sohu.com/vrs_flash.action?vid=zDownloading JSON data for )headers)Z_download_jsonZgeo_verification_headers)vid_idmytvZbase_data_url)selfvideo_id J/mnt/pikpak/tmp/myenv/lib/python3.9/site-packages/yt_dlp/extractor/sohu.py_fetch_datak   s    z)SohuIE._real_extract.<locals>._fetch_datar   r5   u.   ( - 高清正版在线观看)? - 搜狐视频$ zvar vid ?= ?["\'](\d+)["\']z
video pathZplayr   status   z, said: There's something wrong in the video.T)expectedz= said: The video is only licensed to users in Mainland China.)ZnorhighsuperZoriZh2644kZh2654kdataZVidZtotalBlocksallot)ZclipsURLZ
mp4PlayUrl)get_allzUnable to extract url for clip suznewflv.sohu.ccgslb.netr   	   Zh5n)ZprotfilenewprodrbZidczDownloading z video URL part z of z	 (retry #)http://z/?r   nid   zFailed to get video URLc                    s   | d   S )NZ
clipsBytesr8   )x)ir8   r9   <lambda>       z&SohuIE._real_extract.<locals>.<lambda>widthheightfps)r   	format_idfilesizerR   rS   rT   Z_partZclipsDuration)r   r   r    formatsZmulti_videoZtotalDuration)_typeentriesr   r   r    z.publishTime:\s*["\'](\d+-\d+-\d+ \d+:\d+)["\']zpublish time)fatalZtv_application_timer!   ip  ZsubNameZwm_dataZwm_usernameZcoverImgtag)	alt_titler   r#   r$   )F)Z_match_valid_urlgroupZ_download_webpageresubZ_og_search_titleZ_html_search_regexgetr   ZIE_NAMEZraise_geo_restrictedstrrangeitemsr   r   Z_parse_jsonurllibparse	urlencodeappendr   r	   lenr   r
   Z_search_regexsplit)r6   r   r:   Zmobjr5   Zwebpager   ZvidZvid_dataZformats_jsonrU   r4   
part_countr1   rW   Zformat_datarB   rA   Zclip_urlrD   Z	video_urlZcdn_idretriesr2   Zdownload_noteZ	part_infoinfoZpublish_timer8   )rO   r6   r7   r9   _real_extracti   s    















zSohuIE._real_extractN__name__
__module____qualname__Z
_VALID_URLZ_TESTSrm   r8   r8   r8   r9   r      s   
	$Sr   c                   @   s|   e Zd ZdZdddddddd	d
dddddddddddddg dd	dddddddd d!d"g d#d$d%d&gZd'd( Zd)S )*SohuVIEz8https?://tv\.sohu\.com/v/(?P<id>[\w=-]+)\.html(?:$|[#?])r&   z?https://tv.sohu.com/v/MjAyMzA2MTQvbjYwMTMxNTE5Mi5zaHRtbA==.htmlZ	601315192u   《淬火丹心》第1集u   “点天灯”发生事故gMb@idZ20230614zMhttp://photocdn.tv.sohu.com/img/20230614/vrsa_hor_1686738763256_454010551.jpg)r   r   r\   r    r!   r"   r#   rE   zOnly available in China)r   r   r   playlist_mincountr   z?https://tv.sohu.com/v/dXMvMjMyNzk5ODg5Lzc4NjkzNDY0LnNodG1s.htmlr   r   r   r   r   r   r   r   r   r   r%   zFhttps://tv.sohu.com/v/dXMvMjQyNTYyMTYzLzc4OTEwMzM5LnNodG1s.html?src=plr'   r(   r)   r*   r+   r,   r-   r.   r/      )r   r   r   rs   c                 C   sF   |  |}t| }td|r(dnd}| td| d|tS )Nz\d+/n\d+\.shtmltvzmy.tvrK   z
.sohu.com/)	Z	_match_idbase64urlsafe_b64decodedecoder^   matchZ
url_resultr   r   )r6   r   Z
encoded_idpathZ	subdomainr8   r8   r9   rm     s    
zSohuVIE._real_extractNrn   r8   r8   r8   r9   rr      sR   	
+rr   )rv   r^   urllib.parserd   commonr   utilsr   r   r   r   r	   r
   r   r   r   rr   r8   r8   r8   r9   <module>   s   ( ^