music.cpython-310.pyc•4.82 kB
o
LEh� � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dl m
Z d dl m
Z d dlmZ d dlZd dlZd dlZd d lmZ daej�d
�Zd d � Zd
d� Zdefdd�Z dS )� )� webdriver)�By)�
WebDriverWait)�expected_conditions)�Service)�Options)� ActionChainsN)�FastMCP�EDGE_DRIVER_PATHc C sT t du r(t� } | �d� | �ddg� | �d� ttd�}tj|| d�a td� t S ) u& 获取或初始化Edge WebDriver实例Nz---disable-blink-features=AutomationControlled�excludeSwitcheszenable-automationz--start-maximized)�executable_path)�service�optionsu WebDriver实例已初始化。) �driver_instance� EdgeOptions� add_argument�add_experimental_option� EdgeServicer
r �Edge�print)Z edge_optionsr
� r � E:\xiaozhi-mcp\tools\music.py�
get_driver s
r c C s\ z#t | d��t�tjdf�� | �tjd��� td� t �
d� W dS td� Y dS )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)�driverr r r � close_popup s
�r$ �mcpc C s4 | � � dtdtfdd��}| � � dtfdd��}dS )u7
注册音乐播放相关的MCP工具函数。
� song_name�returnc
S sL t � }z�d|jvr|�d� td� t|� zt|d��t�t j
df��� � td� W n t yE } z
td|� �� W Y d}~nd}~ww t|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 �W S t �y% } z,td t|�� �� t�d �}d d!| � d"t|�� �d#|� d$�d%|� d&�gd'�W Y d}~S d}~ww )(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';r u* 已悬停鼠标，尝试显示播放按钮� z .list-item[data-no] .icon-playzarguments[0].click();u 已点击播放按钮c S s2 d| � tjd��d�v p| � tjd��d��d�S )Nu 暂停z .btn-play�title�classz
icon-pause)r r r �
get_attribute�endswith)�dr r r �<lambda>x s 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)r2 r3 Z debug_files)r � current_url�getr r$ r r r �element_to_be_clickabler r r � 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� timestampr r r �play_web_music- sn
� ��
�
�
�
�
�
�
<��z,register_music_tools.<locals>.play_web_musicc S s, t rt �� da td� ddd�S ddd�S )uv
关闭所有由该脚本打开的浏览器窗口。
返回:
- 操作结果状态。
Nu 浏览器已关闭。Tr1 Fu* 没有活动的浏览器实例可关闭。)r �quitr r r r r �
close_browser� s
z+register_music_tools.<locals>.close_browserN)�toolr? �dict)r% rD rF r r r �register_music_tools) s WrI ) �seleniumr � selenium.webdriver.common.byr � selenium.webdriver.support.uir Zselenium.webdriver.supportr r Z selenium.webdriver.edge.servicer r Z selenium.webdriver.edge.optionsr r Z'selenium.webdriver.common.action_chainsr r! �ctypes�os�mcp.server.fastmcpr r �environr5 r
r r$ rI r r r r �<module> s