U
    g0                     @   s   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
 G dd deZG dd	 d	eZG d
d deZG dd deZG dd deZG dd deZG dd deZdS )    N   )InfoExtractor   )ExtractorErrorInAdvancePagedList
clean_htmlget_element_by_idremove_startc                
   @   sd   e 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dddddddgZdddZdS )
KuwoBaseIEaped   formatext
preferencemp3-320mp3Z320kmp3i@  P   )r   r   brabrr   zmp3-192Z192kmp3   F   zmp3-128Z128kmp3   <   Zwma   Zaac0   
   )r   r   r   r   Fc              
   C   s   g }| j D ]}|d |ddd| ddd}| jd|d	|d
 ||  d}|dkrj|sjtddd|dr
|||d
 |d
 |d |dd q
|S )Nr   r    ZMUSIC_Zconvert_urlurl)r   r   Zridtyperesponsez http://antiserver.kuwo.cn/anti.szDownload {} url infor   )notequeryheadersZIPDenyz#This song is blocked in this regionTexpected)zhttp://zhttps://r   r   )r   	format_idr   Zqualityr   )_FORMATSget_download_webpager   Zgeo_verification_headersr   
startswithappend)selfsong_idtolerate_ip_denyformatsZfile_formatr"   song_url r1   9/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/extractor/kuwo.py_get_formats   s4    

  

zKuwoBaseIE._get_formatsN)F)__name__
__module____qualname__r'   r3   r1   r1   r1   r2   r
      s   

	r
   c                	   @   sd   e Zd ZdZd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iddddgZdd ZdS ) KuwoIEFz	kuwo:songu   酷我音乐z.https?://(?:www\.)?kuwo\.cn/yinyue/(?P<id>\d+)z!http://www.kuwo.cn/yinyue/635632/Z635632r   u   爱我别走u	   张震岳Z20080122z$md5:ed13f58e3c3bf3f7fd9fbc4e5a7aa75c)idr   titlecreatorupload_datedescription6this song has been offline because of copyright issues)r   	info_dictskipz"http://www.kuwo.cn/yinyue/6446136/Z6446136r   u   心z$md5:5d0e947b242c35dc0eb1d2fce9fbf02cZIUZ20150518)r8   r   r9   r<   r:   r;   r   r   r   r>   paramsz3http://www.kuwo.cn/yinyue/3197154?catalog=yueku2016T)r   Zonly_matchingc                 C   s   |  |}| j||ddd\}}||jks2d|kr>tddd| d|d	}t| jd
|dddd}ttd|}|dkr~d }| |}| jd|ddd}	d }
|	d k	r| j	d|	 d|ddd}| jd|ddd}
|
r|

dd}
||||
||dS )NzDownload song detail infozUnable to get song detail infor!   errnoteuK   对不起，该歌曲由于版权问题已被下线，将返回网站首页r=   Tr$   z<p[^>]+id="lrcName">([^<]+)</p>z	song namez@<a[^>]+href="http://www\.kuwo\.cn/artist/content\?name=([^"]+)">singer nameF)fatalu   歌手Z
lrcContent   暂无z/<a[^>]+href="http://www\.kuwo\.cn/album/(\d+)/"zalbum idzhttp://www.kuwo.cn/album//zDownload album detail infozUnable to get album detail infou"   发行时间：(\d{4}-\d{2}-\d{2})zpublish time-r   )r8   r9   r:   r;   r<   r/   )	_match_idZ_download_webpage_handler   r   _html_search_regexr	   r   r   r3   r)   replace)r,   r   r-   webpageZurlh	song_namesinger_nameZlrc_contentr/   album_idZpublish_timeZalbum_info_pager1   r1   r2   _real_extractZ   sn    
  
    
  
   zKuwoIE._real_extractN)	r4   r5   r6   _WORKINGIE_NAMEIE_DESC
_VALID_URL_TESTSrP   r1   r1   r1   r2   r7   8   s>   	 r7   c                   @   s8   e Zd ZdZdZdZdZddddd	d
dZdd ZdS )KuwoAlbumIEFz
kuwo:albumu   酷我音乐 - 专辑z/https?://(?:www\.)?kuwo\.cn/album/(?P<id>\d+?)/z http://www.kuwo.cn/album/502294/Z502294u   Made Series 《M》z$md5:d463f0d8a0ff3c3ea3d6ed7452a9483fr8   r9   r<   r   )r   r>   Zplaylist_countc                    sn     |} j||ddd} d|d}tttd|| d} fdd	td
|D } ||||S )NzDownload album infozUnable to get album inforB   z1<div[^>]+class="comm"[^<]+<h1[^>]+title="([^"]+)"z
album nameintro	   简介：c                    s   g | ]}  |d qS Kuwo
url_result.0r0   r,   r1   r2   
<listcomp>   s    z-KuwoAlbumIE._real_extract.<locals>.<listcomp>zF<p[^>]+class="listen"><a[^>]+href="(http://www\.kuwo\.cn/yinyue/\d+/)")	rI   r)   rJ   r	   r   r   refindallplaylist_result)r,   r   rO   rL   Z
album_nameZalbum_introentriesr1   r`   r2   rP      s.    
   
zKuwoAlbumIE._real_extractN	r4   r5   r6   rQ   rR   rS   rT   _TESTrP   r1   r1   r1   r2   rV      s   
rV   c                   @   s4   e Zd ZdZdZdZdZdddidd	Zd
d ZdS )KuwoChartIEFz
kuwo:chartu   酷我音乐 - 排行榜z5https?://yinyue\.kuwo\.cn/billboard_(?P<id>[^.]+).htmu9   http://yinyue.kuwo.cn/billboard_香港中文龙虎榜.htmr8   u   香港中文龙虎榜   r   r>   playlist_mincountc                    sB     |} j||ddd} fddtd|D } ||S )NzDownload chart infozUnable to get chart inforB   c                    s   g | ]}  |d qS rZ   r\   r^   r`   r1   r2   ra      s    z-KuwoChartIE._real_extract.<locals>.<listcomp>z.<a[^>]+href="(http://www\.kuwo\.cn/yinyue/\d+))rI   r)   rb   rc   rd   )r,   r   Zchart_idrL   re   r1   r`   r2   rP      s    
  
 zKuwoChartIE._real_extractNrf   r1   r1   r1   r2   rh      s    rh   c                   @   sN   e Zd ZdZdZdZdZddddd	d
dddddddgZdZdd Z	dS )KuwoSingerIEFzkuwo:singeru   酷我音乐 - 歌手z2https?://(?:www\.)?kuwo\.cn/mingxing/(?P<id>[^/]+)z'http://www.kuwo.cn/mingxing/bruno+mars/z
bruno+marsu   Bruno Mars)r8   r9   iI  rj   z)http://www.kuwo.cn/mingxing/Ali/music.htmZAli_   zRegularly stalls travis build)r   r>   rk   r?      c                    sx    jddd}d|d}d|d td|d	} fd
d}t||j}||S )NzDownload singer infozUnable to get singer inforB   z<h1>([^<]+)</h1>rD   zdata-artistid="(\d+)"z	artist idzdata-page="(\d+)"z
page countc              	      sN   j dd| d  d| d   | jdd}fddtd	|D S )
Nz+http://www.kuwo.cn/artist/contentMusicsAjaxzDownload song list page #r   zUnable to get song list page #)ZartistIdZpnZrn)r!   rC   r"   c                    s"   g | ]}  tj|d qS rZ   )r]   urllibparseurljoinr^   )r,   r   r1   r2   ra      s   zAKuwoSingerIE._real_extract.<locals>.page_func.<locals>.<listcomp>z0<div[^>]+class="name"><a[^>]+href="(/yinyue/\d+))r)   	PAGE_SIZErb   rc   )Zpage_numrL   Z	artist_idr,   Z	singer_idr   r1   r2   	page_func   s     z-KuwoSingerIE._real_extract.<locals>.page_func)rI   r)   rJ   intr   rr   rd   )r,   r   rL   rN   Z
page_countrt   re   r1   rs   r2   rP      s2    
        zKuwoSingerIE._real_extractN)
r4   r5   r6   rQ   rR   rS   rT   rU   rr   rP   r1   r1   r1   r2   rl      s(   rl   c                   @   s8   e Zd ZdZdZdZdZddddd	d
dZdd ZdS )KuwoCategoryIEFzkuwo:categoryu   酷我音乐 - 分类z3https?://yinyue\.kuwo\.cn/yy/cinfo_(?P<id>\d+?).htmz(http://yinyue.kuwo.cn/yy/cinfo_86375.htmZ86375u   八十年代精选u*   这些都是属于八十年代的回忆！rW      rj   c                    s     |} j||ddd} d|d}ttd| | d}|dkrPd }  d	|d
|} fdd|d D } ||||S )NzDownload category infozUnable to get category inforB   z&<h1[^>]+title="([^<>]+?)">[^<>]+?</h1>zcategory namerX   rY   rF   zvar\s+jsonm\s*=\s*([^;]+);zcategory songsc                    s"   g | ]}  d |d dqS )zhttp://www.kuwo.cn/yinyue/{}/Zmusicridr[   )r]   r   )r_   songr`   r1   r2   ra   "  s   z0KuwoCategoryIE._real_extract.<locals>.<listcomp>Z	musiclist)rI   r)   rJ   r	   r   stripZ_parse_jsonrd   )r,   r   Zcategory_idrL   Zcategory_nameZcategory_descZjsonmre   r1   r`   r2   rP     s:    
      
zKuwoCategoryIE._real_extractNrf   r1   r1   r1   r2   rv     s   
rv   c                   @   s\   e Zd ZdZdZdZdZddddd	d
ddidZej	ddddddddfZ	dd Z
dS )KuwoMvIEFzkuwo:mvu   酷我音乐 - MVz,https?://(?:www\.)?kuwo\.cn/mv/(?P<id>\d+?)/zhttp://www.kuwo.cn/mv/6480076/Z6480076Zmp4z
My HouseMVZ2PM)r8   r   r9   r:   r   mvr@   Zmkv   r      c           	      C   s   |  |}| j||d| d| d}td|}|rN|d}|d}ntd| j|dd	}| jd
| |d| dd}||dd ||||dS )NzDownload mv detail info: zUnable to get mv detail info: rB   zH<h1[^>]+title="(?P<song>[^"]+)">[^<]+<span[^>]+title="(?P<singer>[^"]+)"rx   Zsingerz#Unable to find song or singer namesT)r.   z)http://www.kuwo.cn/yy/st/mvurl?rid=MUSIC_z	Download z MV URL)r!   r{   )r   r&   )r8   r9   r:   r/   )rI   r)   rb   searchgroupr   r3   r+   )	r,   r   r-   rL   ZmobjrM   rN   r/   Zmv_urlr1   r1   r2   rP   B  s<    
  
 
zKuwoMvIE._real_extractN)r4   r5   r6   rQ   rR   rS   rT   rg   r
   r'   rP   r1   r1   r1   r2   rz   )  s&   
 

rz   )rb   urllib.parsero   commonr   utilsr   r   r   r   r	   r
   r7   rV   rh   rl   rv   rz   r1   r1   r1   r2   <module>   s   	*P%:(