bilibili_search.cpython-310.pyc•5.7 kB
o
�3Eh0$ � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dl m
Z d dl Z d dl
Z
d dlZd dlmZ d dlmZ d Zd
ejd efd d
�Zd
ejd efdd�Zd
ejd efdd�Zd
ejd efdd�Zd ejfdd�Zd
ejded efdd�Z defdd�Z dS ) � )� webdriver)�By)�
WebDriverWait)�expected_conditions)�ServiceN)�Dict)�FastMCPzbilibili_cookies.json�driver�returnc
C s~ z$| � � }ttd��}t�||� W d � W dS 1 s w Y W dS ty> } ztd|� �� W Y d}~dS d}~ww )u 保存当前Cookies到文件�wNTu 保存Cookies失败: F)Z get_cookies�open� COOKIES_FILE�json�dump� Exception�print)r �cookies�f�e� r �'E:\xiaozhi-mcp\tools\bilibili_search.py� save_cookies s
����r c
C s� t j�t�sdS zXttd��
}t�|�}W d � n1 s w Y | �d� | �� |D ]/}d|�dd�vr<d|d< d |vrDd
|d < d |vrLd|d < d |v rXt |d �|d < | �
|� q.W d
S t yz } zt d|� �� W Y d}~dS d}~ww )u 从文件加载CookiesF�rN�https://www.bilibili.com� bilibili.com�domain� z
.bilibili.com�path�/�secureZexpiryTu 加载Cookies失败: )
�osr �existsr
r r �load�getZdelete_all_cookies�intZ
add_cookier r )r r r �cookier r r r � load_cookies s0 �
��r& c C s� t | �rA| �d� t�d� zt| d��t�tj df�� | �d� t�d� W dS t
j � t
�r8t
�t
� td� Y t| �S t| �S )u! 自动登录流程（优化版）r � � z5.header-avatar, .header-avatar-wrap, .nav-user-center� Tu1 Cookie 已失效或未登录，需要重新登录)r& r# �time�sleepr �until�EC�presence_of_element_locatedr � CSS_SELECTORr r r! r
�remover � manual_login)r r r r �
auto_login> s
�
r2 c
C st z t d� | �d� t| d��dd� � t�d� t| � W dS ty9 } zt d|� �� W Y d }~d
S d }~ww ) u! 手动登录流程（优化版）u, 请手动完成登录（30秒内完成）...z#https://passport.bilibili.com/login� c S s d| j vo d| j v S )N�loginr )Z current_url)�dr r r �<lambda>_ s z manual_login.<locals>.<lambda>r' Tu 登录失败或超时: NF)r r# r r, r* r+ r r )r r r r r r1 W s
�
��r1 c C sZ t �� } | �d� | �ddg� | �d� | �d� | �dd� t j| d�}|�d � |S )
u0 初始化浏览器驱动（防止自动关闭）z---disable-blink-features=AutomationControlledZexcludeSwitcheszenable-automationz--start-maximizedz
--log-level=3�detachT)�options�
)r Z EdgeOptions� add_argumentZadd_experimental_option�EdgeZimplicitly_wait)r8 r r r r � init_driverj s
r< �keywordc
C s� z�d|� �}| � |� t| d��t�tjdf�� t| d��t�tjdf��}| �d|� t �
d� | �d|� t| d ��t� d
�� | j �
| jd � t| d��t�tjd f�� td � z t| d��t�tjd
f��}|�d�dkr||�� td� W n Y zt| d��t�tjdf��}|�� td� W W dS td� Y W dS ty� } ztd|� �� W Y d}~dS d}~ww )u9 搜索并播放视频（只要打开视频就算成功）z(https://search.bilibili.com/all?keyword=� z.bili-video-card� z .bili-video-card:first-child az arguments[0].scrollIntoView();r) zarguments[0].click();r9 r' zbpx-player-containeru 视频页面已成功加载。z.bpx-player-ctrl-playz
data-state�pauseu 尝试点击播放按钮。z.bpx-player-ctrl-fullu 尝试全屏播放。u 全屏失败或已全屏。Tu 播放视频失败: NF)r# r r, r- r. r r/ Zelement_to_be_clickableZexecute_scriptr* r+ Znumber_of_windows_to_beZ switch_to�windowZwindow_handlesZ
CLASS_NAMEr �
get_attribute�clickr )r r= Z
search_urlZ first_videoZ play_buttonZfullscreen_btnr r r r �search_and_play_video~ sX
�
�
�
��
�
���rD �mcpc C s& | � � dtdtttf fdd��}dS )u" 注册B站视频播放功能到MCPr= r
c
S s� d}z*t � }t|�s|r|�� ddd�W S t|| �s'|r!|�� ddd�W S ddd�W S tyN } z|r:|�� ddt|�� �d�W Y d}~S d}~ww ) u�
在B站搜索并尝试全屏播放视频（成功后浏览器保持打开）
参数:
- keyword: 搜索关键词
返回:
- {"success": bool, "result": str}
NFu B站登录失败)�success�resultu 视频播放失败Tu- 视频已成功播放，浏览器保持打开u 发生错误: )r< r2 �quitrD r �str)r= r r r r r �play_bilibili_video� s$
��z3register_bilibili_tool.<locals>.play_bilibili_videoN)�toolrI r )rE rJ r r r �register_bilibili_tool� s rL ) Zseleniumr Z selenium.webdriver.common.byr Z selenium.webdriver.support.uir Zselenium.webdriver.supportr r- Z selenium.webdriver.edge.servicer Z EdgeServicer r r* �typingr �mcp.server.fastmcpr r
r; �boolr r& r2 r1 r< rI rD rL r r r r �<module> s$ "=