system.cpython-310.pyc•6.6 kB
o
&EhB! � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
e �d�ZejdkrLejjdd� ejjdd� dd � Zd
e
fd d �ZdS )
� N)�Path)�FastMCPZ system_tools�win32�utf-8��encodingc C s t �� d S )u! 获取当前用户的桌面路径ZDesktop)r �home� r r � E:\xiaozhi-mcp\tools\system.py�get_desktop_path s r �mcpc C s� | � � dtfdd��}| � � ddtdtdtfdd��}| � � d tdtfd
d ��}| � � d tdtfd d
��}| � � dtfdd��}| � � dtfdd��}| � � dtdtfdd��}d S )N�returnc
S s� zUt jdd�} t �� }t �� }|jd }|jd }|j}t �d�}|jd }|jd }|j} t �� }
d| |d�t |d�t |d�|d�t |d�t |d�| d�d |
id
�d �W S t
yx } zt � d t
| �� �� d
t
| �d�W Y d} ~ S d} ~ ww )u�
获取服务器状态监控信息。
返回:
- 包含CPU、内存、磁盘等使用情况的字典
� )�intervali @�/T)�
usage_percentZ
core_count� )Ztotal_gbZused_gbr � boot_time)Zcpu�memory�disk�system)�success�resultu 获取服务器状态失败: F�r �errorN)�psutil� cpu_percent� cpu_countZvirtual_memory�total�used�percent�
disk_usager �round� Exception�loggerr �str) r r r Z memory_totalZ memory_usedZmemory_percentr Z
disk_totalZ disk_usedZ disk_percentr �er r r
�get_server_status sF
���������z0register_system_tools.<locals>.get_server_status� �text�filenamec
S s2 zu|r?t � }|| }t|ddd��
}|�| � W d � n1 s!w Y t�dt|�g� t�d|� �� dt|�d�W S tj ddd d
d d ��}|�| � |j
}W d � n1 s\w Y t�d|g� t�d
|� �� d|d�W S t y� } zt� dt|�� �� d
t|�d�W Y d}~S d}~ww )aM
Open Notepad with text and optionally save to a specific file.
Args:
text: The text to display in Notepad
filename: If provided, save to this file on desktop (e.g., "poem.txt")
Returns:
dict: {"success": bool, "filepath": str (if saved), "error": str (if failed)}
�wr r N� notepad.exezOpened Notepad with file: T)r �filepathz.txtFZnotepad_)�moder �suffix�delete�prefixz$Opened Notepad with temporary file: z"Failed to open Notepad with text: r )
r �open�write�
subprocess�Popenr% r$ �info�tempfile�NamedTemporaryFile�namer# r )r) r* Zdesktopr- �fZ temp_pathr& r r r
�open_notepad_with_textV s:
��
� ��z5register_system_tools.<locals>.open_notepad_with_text� program_namec
S s� ddddddddd d
� }| � � }||vr d d | � �d
�S z"|dkr+tj|d dd� nt�|| � t�d| � �� ddiW S tyd } zt�d| � dt|�� �� d t|�d
�W Y d}~S d}~ww )uw
Open common Windows programs. Supported programs:
- 'notepad' (记事本)
- 'explorer' (文件资源管理器)
- 'cmd' (命令提示符)
- 'calculator' (计算器)
- 'control' (控制面板)
- 'taskmgr' (任务管理器)
- 'mspaint' (画图)
- 'wordpad' (写字板)
- 'wechat' (微信)
r, z explorer.exezcmd.exezcalc.exez control.exez taskmgr.exe� mspaint.exe� write.exez"D:\\WeChat\\WeChat.exe") �notepadZexplorer�cmd�
calculatorZcontrolZtaskmgr�mspaint�wordpad�wechatF�Unsupported program: r rD T)�shellzSuccessfully opened r zFailed to open �: N)�lowerr4 r5 r$ r6 r# r r% )r< Zprograms�program_name_lowerr& r r r
� open_program| s0 �
��z+register_system_tools.<locals>.open_programc
S s� dddddd�}| � � }||vrdd| � �d �S z%|| }t�d
g�D ]
}|jd
|kr0|�� q#t�d | � �� d d
iW S tyc } zt�d| � dt|�� �� dt|�d �W Y d}~S d}~ww )z�
Close a running program by its process name.
Supported programs: 'calculator', 'notepad', 'mspaint', etc.
zCalculator.exer, r= r> z
WeChat.exe)rA r? rB rC rD FrE r r9 zSuccessfully closed r TzFailed to close rG N) rH r Z process_iterr6 �killr$ r# r r% )r< Z
process_namesrI Z process_name�procr& r r r
�
close_program� s, ��
��z,register_system_tools.<locals>.close_programc
S sh zt �dd� t�d� ddiW S ty3 } zt�dt| �� �� dt| �d�W Y d } ~ S d } ~ ww )
z<Minimize all windows to show the desktop (Win+D equivalent).�win�dzShowing desktopr TzFailed to show desktop: Fr N)� pyautoguiZhotkeyr$ r6 r# r r% �r& r r r
� show_desktop� s
��z+register_system_tools.<locals>.show_desktopc
S sf zt �d� t�d� ddiW S ty2 } zt�dt| �� �� dt| �d�W Y d} ~ S d} ~ ww ) z$Lock the Windows screen immediately.z'rundll32.exe user32.dll,LockWorkStationzScreen locked successfullyr TzFailed to lock screen: Fr N��osr r$ r6 r# r r% rQ r r r
� lock_screen� s
��z*register_system_tools.<locals>.lock_screen�commandc
S sx zt �d| � �� t�d| � �� ddiW S ty; } zt�d| � dt|�� �� dt|�d�W Y d }~S d }~ww )
z�
Run a Windows command or open a specific file/path.
Examples:
- 'C:\Program Files\MyApp\app.exe'
- 'shell:startup' (opens the startup folder)
- 'ms-settings:' (opens Windows settings)
zstart zExecuted command: r TzFailed to execute command rG Fr NrS )rV r& r r r
� run_command� s
��z*register_system_tools.<locals>.run_command)r( N)�tool�dictr% )r r' r; rJ rM rR rU rW r r r
�register_system_tools s 7%)
rZ )�logging�requestsr rT �sysr4 Z
webbrowserr7 �timerP �pathlibr �mcp.server.fastmcpr � getLoggerr$ �platform�stderr� reconfigure�stdoutr rZ r r r r
�<module> s$