Xiaozhi MCP Aggregate

o LEh��@s�dd
lm
Z

ddlmZ
ddlmZ
ddlmZ
ddl m Z
ddlm Z
ddlmZ
ddlZddlZddlZdd lmZ
daej�d �
Zdd�Zd d�Zdefdd�ZdS)�)
� webdriver)
�By)
� WebDriverWait)
�expected_conditions)
�Service)
�Options)
�ActionChainsN)
�FastMCP�EDGE_DRIVER_PATHcCsTtd
ur(t
�}|�d�

|�ddg
�
|�d�

ttd�
}
tj|
|d�atd�

tS) u&获取或初始化Edge WebDriver实例Nz---disable-blink-features=AutomationControlled�excludeSwitcheszenable-automationz--start-maximized)
�executable_path)�service�optionsuWebDriver实例已初始化。) �driver_instance�EdgeOptions�add_argument�add_experimental_option�EdgeServicer r�Edge�print)Zedge_optionsr �r�E:\xiaozhi-mcp\tools\music.py� get_drivers







rc

Cs\z#t|d
��
t�tjdf�
�

|�tjd���
td�

t � d�

WdS


td�

YdS)u 关闭GD音乐台的公告弹窗�z.layui-layer-dialogz.layui-layer-close1u已关闭公告弹窗�
u未检测到公告弹窗N)r�until�EC�presence_of_element_locatedr�CSS_SELECTOR�find_element�clickr�time�sleep)
�driverrrr�close_popups

�



r$�mcpc
Cs4|��d
t
dtfdd��
}
|��dtfdd��
}dS)u7 注册音乐播放相关的MCP工具函数。 � song_name�returnc
 SsLt�}
z�d
|
j
v
r|
�d�

td�

t|
�

zt|
d��t�t j df�
�
��
td�

WntyE
}
z td|���

WYd}~nd}~w
wt|
d ��t�t j d f�
�
}|��
|�|�

td|���

t|
d��t�t j df�
�
}|��
t|
d ��t�t j d f�
�

td�

t|
d��t�t j d f�
�
}|
�d|�
|
�d|�
t�d�

t|
�
�|�
��
td�

t�d�

t|
d��t�t j df�
�
}|
�d|�
td�

t|
d ��dd��

td|�d��

dd|�d�d�WSt�
y%
}
z,tdt|�
���

t�d�
}d d!|�d"t|�
��d#|�d$�d%|�d&�gd'�WYd}~Sd}~w
w)(u� 在GD音乐台搜索并播放指定歌曲。 浏览器在播放后不会关闭。 参数: - song_name: 要搜索的歌曲名称。 返回: - 操作结果状态和信息。 zmusic.gdstudio.xyzzhttps://music.gdstudio.xyz/u正在打开GD音乐台...� z[data-action="search"]u已进入搜索界面uG未能点击搜索按钮，可能已在搜索界面或元素不可用: N�z search-wdu正在搜索歌曲: z.search-submitz.list-item[data-no]u搜索结果已加载z/arguments[0].scrollIntoView({block: 'center'});z*arguments[0].style.border='2px solid red';ru*已悬停鼠标，尝试显示播放按钮�z.list-item[data-no] .icon-playzarguments[0].click();u已点击播放按钮c

Ss2d
|�t
jd��d�
vp|�t
jd��d�
�d�
S)Nu暂停z .btn-play�title�classz icon-pause)rrr� get_attribute�endswith)
�
drrr�<lambda>xs
z>register_music_tools.<locals>.play_web_music.<locals>.<lambda>u歌曲 'u' 已开始播放。Tu' 已成功播放。��success�messageu播放失败: z %Y%m%d_%H%M%SFu播放歌曲 'u ' 失败: Zerror_z.pngZpage_source_z.html)r2r3Zdebug_files)r�current_url�getrr$rrr�element_to_be_clickablerrr � Exception�ID�clear� send_keys�visibility_of_element_locatedr�execute_scriptr!r"r�move_to_element�perform�str�strftime)r&r#�
eZsearch_inputZ submit_btnZfirst_song_item�play_button� timestamprrr�play_web_music-sn





�


�� 
�


� 
� 
�




�

�

<��z,register_music_tools.<locals>.play_web_musiccSs,trt�
�
d
atd�

ddd�Sddd�S)uv 关闭所有由该脚本打开的浏览器窗口。 返回: - 操作结果状态。 Nu浏览器已关闭。Tr1Fu*没有活动的浏览器实例可关闭。)r�quitrrrrr� close_browser�s 



z+register_music_tools.<locals>.close_browserN)�toolr?�dict)r%rDrFrrr�register_music_tools)s
W
rI)�seleniumr�selenium.webdriver.common.byr�selenium.webdriver.support.uirZselenium.webdriver.supportrrZselenium.webdriver.edge.servicerrZselenium.webdriver.edge.optionsrrZ'selenium.webdriver.common.action_chainsrr!�ctypes�os�mcp.server.fastmcpr r�environr5r rr$rIrrrr�<module>
s