note.cpython-310.pyc•22 kB
o
]Eh�g � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z d dl m
Z
mZ d dlZe�d�Zej�d�ZdadaG d d
� d
�Ze� Zd d � Zd
d� ZdS )� N)� webdriver)�Service)�Options)�WebDriverException�SessionNotCreatedExceptionZsticky_notes_manager�EDGE_DRIVER_PATHFc @ s� e Zd Zd+dd�Zdd� Zdd� Zd d
� Zd d � Zd
efdd�Z d,dededede
fdd�Z d-de deded ede
f
d d �Z
de de
fd d �Zd-d!ededede
fd"d#�Zde
fd$d%�Zd&ede
fd'd(�Zd&edefd)d*�ZdS ).�StickyNoteManager�sticky_notes.json�sticky_notes.htmlc C s$ || _ || _g | _d| _| �� d S )N� )� data_file�html_output_file�notes�next_id� _load_notes)�selfr r
� r � E:\xiaozhi-mcp\tools\note.py�__init__ s
zStickyNoteManager.__init__c
C sX t j�| j�r�zat| jddd��>}t�|�}|�dg �| _| jD ]}d|vr*d|d< d|vr2d|d< q | jrDt d d
� | jD ��d | _
nd | _
W d � n1 sQw Y t � d
| j� dt
| j�� d�� W d S tjy~ t �d� g | _d | _
Y d S ty� } zt �d|� �� g | _d | _
W Y d }~d S d }~ww t � d� g | _d | _
d S )u 从文件中加载便签数据�r�utf-8��encodingr �
importance� 普通�category� 未分类c s s � | ]}|d V qdS ��idNr ��.0�noter r r � <genexpr>) s � z0StickyNoteManager._load_notes.<locals>.<genexpr>r Nu 已从 'u ' 加载 �
条便签u6 便签数据文件损坏或为空，将创建新文件u 加载便签数据失败: u0 便签数据文件不存在，将创建新文件)�os�path�existsr �open�json�load�getr �maxr �logger�info�len�JSONDecodeError�warning� Exception�error)r �f�datar! �er r r r s<
� � �&
��
z StickyNoteManager._load_notesc
C s� z9d| j i}t| jddd��}tj||ddd� W d � n1 s"w Y t�d t| j �� d
| j� d �� W dS tyT } zt� d |� �� W Y d}~dS d}~ww )
u 将便签数据保存到文件r �wr r � F)�indent� ensure_asciiNu
已保存 u 条便签到 '�'u 保存便签数据失败: )
r r' r r( �dumpr, r- r. �IOErrorr2 )r r4 r3 r5 r r r � _save_notes: s
�& ��z StickyNoteManager._save_notesc C s t j �� �d�S )u 生成当前时间戳z%Y-%m-%d %H:%M:%S)�datetime�now�strftime)r r r r �_generate_timestampD s z%StickyNoteManager._generate_timestampc
C s, t durzt j t W S ty da Y nw t du r�z*t� }|�d� |�ddg� |�d� ttd�}tj ||d�a da
t � d � t W S t
y_ } zt �d
|� �� W Y d}~t S d}~w tyy } zt �d |� �� W Y d}~t S d}~w ty� } zt �d |� �� W Y d}~t S d}~ww t S )
u:
获取或初始化 Edge WebDriver 实例
Nz---disable-blink-features=AutomationControlled�excludeSwitcheszenable-automationz--start-maximized)�executable_path)�service�optionsTu Edge浏览器已启动ud Edge浏览器启动失败：版本不匹配。请确保Edge浏览器和驱动版本一致。错误: u Edge浏览器启动失败：u* Edge浏览器启动失败：未知错误: )�_edge_driver_instance� current_urlr � EdgeOptions� add_argument�add_experimental_option� EdgeServicer r �Edge�_browser_initializedr, r- r r2 r1 )r � edge_optionsrD r5 r r r �_get_edge_driverH s@ �
������z"StickyNoteManager._get_edge_driver� file_pathc
C s� | � � }|s ddd�S z4dtj�|�� �}|jr|jnd}||kr+|�� t�d� n
|�|� t�d� t �
d� d d
d �W S t yc } zt� d |� �� d
a
dd |� �d�W Y d
}~S d
}~ww )u1
刷新浏览器中的HTML页面
Fu5 无法启动Edge浏览器，请检查Edge驱动配置��successr2 zfile://� u 便签页面已刷新u 便签页面已打开g �?Tu 便签显示已更新�rR �messageu 浏览器操作失败：N)rO r$ r% �abspathrG �refreshr, r- r* �time�sleepr r2 rF )r rP �driverZ
target_urlrG r5 r r r �_refresh_browser_pagek s&
��z'StickyNoteManager._refresh_browser_pager r �contentr r �returnc C s� |sddd�S |dvr
d}| j || �� ||d�}| j�|� | j d7 _ | �� | j| jd�}|d s6|S d
d |d � d
�|d�S )u!
添加新便签
Fu 便签内容不能为空rQ )r � 重要� 紧急r )r r\ � timestampr r r ��notes_to_displayrR Tu 已成功添加便签(ID: r �))rR rU r! )r rA r �appendr= �generate_html_report)r r\ r r r! � html_resultr r r �add_note� s(
� �zStickyNoteManager.add_noteN�note_id� new_content�new_importance� new_categoryc s� d}| j D ]*}|d � kr/d}|dur ||d< | �� |d< |dur%||d< |dur-||d< nq|s;dd � � d
�d �S | �� | j| j d �}|d
sL|S dd� � d�t� fdd�| j D �d�d�S )u$
修改现有便签
Fr TNr\ r` r r � 未找到ID为 �
的便签rQ ra rR u 已成功修改便签(ID: rc c 3 s � | ] }|d � kr|V qdS r r r �rh r r r" � s � z0StickyNoteManager.modify_note.<locals>.<genexpr>)rR rU Z updated_note)r rA r= re �next)r rh ri rj rk Z
note_foundr! rf r rn r � modify_note� s.
�
�z StickyNoteManager.modify_notec s| t | j�}� fdd�| jD �| _t | j�|kr dd� � d�d�S | �� | j| jd�}|d s1|S d d
� � d �t | j�d �S )
u
删除便签
c s g | ]
}|d � kr|�qS )r r r rn r r �
<listcomp>� s z1StickyNoteManager.delete_note.<locals>.<listcomp>Frl rm rQ ra rR Tu 已成功删除便签(ID: rc )rR rU Zremaining_notes)r. r r= re )r rh Z
initial_countrf r rn r � delete_note� s
�z StickyNoteManager.delete_note�keywordc sx | j }�r�fdd�|D �}�r�fdd�|D �}� r$� fdd�|D �}| j|d�}|d s0|S ddt|�� d �|d
�S ) u
搜索便签
c s$ g | ]}� � � |d � � v r|�qS )r\ )�lowerr )rs r r rq � s
�z2StickyNoteManager.search_notes.<locals>.<listcomp>c �( g | ]}|� d d��� � �� kr|�qS )r rS �r* rt r )r r r rq � �
�c ru )r rS rv r )r r r rq � rw ra rR Tu 找到 u 条匹配的便签)rR rU �matches)r re r. )r rs r r Zmatching_notesrf r )r r rs r � search_notes� s(
�
�
� �z StickyNoteManager.search_notesc C s6 | j | jd�}|d s
|S ddt| j�� d�| jd�S )u$
列出所有便签
ra rR Tu 共 r# )rR rU r )re r r. )r rf r r r �list_all_notes s �z StickyNoteManager.list_all_notesrb c
C s� z&t | jddd��}|�| �|�� W d � n1 s w Y ddd�W S ty@ } zdd |� �d
�W Y d}~S d}~ww ) u"
生成HTML报告
r6 r r NTu HTML报告已生成rT Fu 生成HTML文件失败: rQ )r' r
�write�_generate_html_contentr< )r rb r3 r5 r r r re s � ��z&StickyNoteManager.generate_html_reportc C sP dt t| j�� d t t|�� d t|�t| j�kr#dt|�� d�nd d }|s/|d7 }ns|D ]p}|d �d
d ��d d
��dd��dd��dd�}|�dd�}|�dd�}d}|�� dkrbd}n|�� dkrjd }d }|�� d krud }n|�� d kr~d!}n|�� d"kr�d#}|d$|d% � d&|� d'|� d(|d) � d*|� d+|� d,�
7 }q1|d-7 }|S ).u"
生成HTML内容
u� <!DOCTYPE html>
<html lang="zh-CN">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>我的便签</title>
<link href="https://fonts.googleapis.com/css2?family=Noto+Sans+SC:wght@300;400;500;700&display=swap" rel="stylesheet">
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0-beta3/css/all.min.css">
<style>
:root {
--primary-color: #4361ee;
--secondary-color: #3f37c9;
--success-color: #4cc9f0;
--danger-color: #f72585;
--warning-color: #f8961e;
--info-color: #4895ef;
--light-color: #f8f9fa;
--dark-color: #212529;
--work-color: #4361ee;
--life-color: #4cc9f0;
--study-color: #7209b7;
--other-color: #6c757d;
}
body {
font-family: 'Noto Sans SC', sans-serif;
margin: 0;
padding: 20px;
background-color: #f5f7fa;
color: #333;
line-height: 1.6;
}
.container {
max-width: 1200px;
margin: 0 auto;
padding: 20px;
}
header {
text-align: center;
margin-bottom: 30px;
padding-bottom: 20px;
border-bottom: 1px solid #e0e0e0;
}
h1 {
color: var(--primary-color);
font-weight: 700;
margin-bottom: 10px;
font-size: 2.5rem;
}
.subtitle {
color: #6c757d;
font-weight: 300;
font-size: 1.1rem;
}
.stats {
display: flex;
justify-content: center;
gap: 20px;
margin-bottom: 20px;
flex-wrap: wrap;
}
.stat-card {
background: white;
border-radius: 8px;
padding: 15px 20px;
box-shadow: 0 2px 4px rgba(0,0,0,0.05);
min-width: 120px;
text-align: center;
}
.stat-value {
font-size: 1.5rem;
font-weight: 700;
color: var(--primary-color);
margin-bottom: 5px;
}
.stat-label {
font-size: 0.85rem;
color: #6c757d;
}
.note-grid {
display: grid;
grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
gap: 25px;
margin-top: 20px;
}
.note-card {
background: white;
border-radius: 12px;
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.05);
transition: all 0.3s ease;
overflow: hidden;
display: flex;
flex-direction: column;
height: 100%;
}
.note-card:hover {
transform: translateY(-5px);
box-shadow: 0 10px 20px rgba(0, 0, 0, 0.1);
}
.note-header {
padding: 15px 20px;
background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
display: flex;
justify-content: space-between;
align-items: center;
}
.note-id {
font-size: 0.85rem;
color: #6c757d;
font-weight: 500;
}
.note-content {
padding: 20px;
flex-grow: 1;
font-size: 1rem;
color: #495057;
border-bottom: 1px solid #f1f1f1;
}
.note-footer {
padding: 15px 20px;
background-color: #f8f9fa;
}
.note-timestamp {
font-size: 0.8rem;
color: #6c757d;
margin-bottom: 10px;
display: flex;
align-items: center;
}
.note-timestamp i {
margin-right: 5px;
font-size: 0.9rem;
}
.note-tags {
display: flex;
flex-wrap: wrap;
gap: 8px;
margin-top: 10px;
}
.tag {
font-size: 0.75rem;
padding: 4px 10px;
border-radius: 50px;
font-weight: 500;
display: inline-flex;
align-items: center;
}
.importance-stars {
display: flex;
gap: 3px;
}
.star {
color: #ffc107;
font-size: 0.9rem;
}
.star.empty {
color: #e0e0e0;
}
/* 重要性星级样式 */
.importance-1 .star:nth-child(1) { color: #ffc107; }
.importance-1 .star:nth-child(n+2) { color: #e0e0e0; }
.importance-2 .star:nth-child(-n+2) { color: #ffc107; }
.importance-2 .star:nth-child(n+3) { color: #e0e0e0; }
.importance-3 .star { color: #ffc107; }
/* 类别标签颜色 */
.category-work {
background-color: rgba(67, 97, 238, 0.1);
color: var(--work-color);
border: 1px solid rgba(67, 97, 238, 0.2);
}
.category-life {
background-color: rgba(76, 201, 240, 0.1);
color: var(--life-color);
border: 1px solid rgba(76, 201, 240, 0.2);
}
.category-study {
background-color: rgba(114, 9, 183, 0.1);
color: var(--study-color);
border: 1px solid rgba(114, 9, 183, 0.2);
}
.category-other {
background-color: rgba(108, 117, 125, 0.1);
color: var(--other-color);
border: 1px solid rgba(108, 117, 125, 0.2);
}
.no-notes {
text-align: center;
padding: 50px 20px;
grid-column: 1 / -1;
}
.no-notes i {
font-size: 3rem;
color: #adb5bd;
margin-bottom: 15px;
}
.no-notes h3 {
color: #6c757d;
font-weight: 400;
margin-bottom: 10px;
}
.no-notes p {
color: #adb5bd;
font-size: 0.9rem;
}
.search-info {
text-align: center;
margin-bottom: 20px;
color: #6c757d;
font-size: 0.9rem;
}
</style>
</head>
<body>
<div class="container">
<header>
<h1>我的便签</h1>
<p class="subtitle">记录生活中的每一个重要时刻</p>
</header>
<div class="stats">
<div class="stat-card">
<div class="stat-value">u� </div>
<div class="stat-label">总便签数</div>
</div>
<div class="stat-card">
<div class="stat-value">uu </div>
<div class="stat-label">当前显示</div>
</div>
</div>
u& <div class="search-info">正在显示 u 条便签</div>rS z*
<div class="note-grid">
u�
<div class="no-notes">
<i class="far fa-sticky-note"></i>
<h3>没有找到便签</h3>
<p>尝试修改搜索条件或添加新便签</p>
</div>
r\ �&z&�<z<�>z>�"z"r: z'r r r r r r^ � r_ � zcategory-otheru 工作z
category-worku 生活z
category-lifeu 学习zcategory-studyz�
<div class="note-card">
<div class="note-header">
<span class="note-id">#r zH</span>
<div class="importance-stars importance-aP ">
<i class="fas fa-star star"></i>
<i class="fas fa-star star"></i>
<i class="fas fa-star star"></i>
</div>
</div>
<div class="note-content">
z�
</div>
<div class="note-footer">
<div class="note-timestamp">
<i class="far fa-clock"></i>
r` z�
</div>
<div class="note-tags">
<span class="tag z>">
<i class="fas fa-tag"></i> z�
</span>
</div>
</div>
</div>
z+
</div>
</div>
</body>
</html>
)�strr. r �replacer* rt ) r rb �htmlr! Zescaped_contentr r Zimportance_levelZcategory_classr r r r| s� � �� � ��
� �� � �� $
� ��
� ��
0 �� ��� � z(StickyNoteManager._generate_html_content)r r
�r r �NNN)�__name__�
__module__� __qualname__r r r= rA rO r� r[ �dictrg �intrp rr ry rz �listre r| r r r r r s,
# ! ���
�" # r c C s� | � � ddtdtdtdtfdd��}| � � dd
td td td
tdtf
dd��}| � � d
tdtfdd��}| � � ddtdtdtdtfdd��}| � � dtfdd��}| � � dtfdd��}d S )u"
注册便签管理工具
r r r\ r r r] c S �> t �| ||�}|d r t �t j�}|d s |�dd�|d< |S )u�
添加新便签
:param content: 便签内容(必填)
:param importance: 重要性(普通/重要/紧急)
:param category: 分类(工作/生活/学习/其他)
:return: 操作结果和便签信息
rR r2 � 浏览器刷新失败�browser_message)�_sticky_note_managerrg r[ r
r* )r\ r r �result�browser_resultr r r �add_sticky_note{ s z4register_sticky_notes_tools.<locals>.add_sticky_noteNrh ri rj rk c S s@ t �| |||�}|d r t �t j�}|d s |�dd�|d< |S )u#
修改现有便签
:param note_id: 要修改的便签ID(必填)
:param new_content: 新内容(可选)
:param new_importance: 新重要性(可选)
:param new_category: 新分类(可选)
:return: 操作结果和更新后的便签信息
rR r2 r� r� )r� rp r[ r
r* )rh ri rj rk r� r� r r r �modify_sticky_note� s z7register_sticky_notes_tools.<locals>.modify_sticky_notec S s: t �| �}|d rt �t j�}|d s|�dd�|d< |S )u{
删除便签
:param note_id: 要删除的便签ID
:return: 操作结果和剩余便签数
rR r2 r� r� )r� rr r[ r
r* )rh r� r� r r r �delete_sticky_note� s
z7register_sticky_notes_tools.<locals>.delete_sticky_noters c S r� )u�
搜索便签
:param keyword: 关键词搜索(可选)
:param importance: 按重要性筛选(可选)
:param category: 按分类筛选(可选)
:return: 操作结果和匹配的便签列表
rR r2 r� r� )r� ry r[ r
r* )rs r r r� r� r r r �search_sticky_notes� s
z8register_sticky_notes_tools.<locals>.search_sticky_notesc S s8 t �� } | d rt �t j�}|d s|�dd�| d< | S )uW
列出所有便签
:return: 操作结果和所有便签列表
rR r2 r� r� )r� rz r[ r
r* )r� r� r r r �list_all_sticky_notes� s z:register_sticky_notes_tools.<locals>.list_all_sticky_notesc S s& t jt jd�} | d s
| S t �t j�S )uF
刷新便签HTML页面
:return: 操作结果
ra rR )r� re r r[ r
)rf r r r �show_sticky_notes_html� s � z;register_sticky_notes_tools.<locals>.show_sticky_notes_htmlr� r� )�toolr� r� r� )�mcpr� r� r� r� r� r� r r r �register_sticky_notes_toolsw s4 ��� � �� �
r� c
C sp t r6z0z t �� t�d� W n ty( } zt�d| � �� W Y d} ~ n
d} ~ ww W da dS W da dS da w dS )u#
关闭Edge浏览器驱动
u Edge浏览器已关闭u 关闭Edge浏览器失败: N)rF �quitr, r- r r2 )r5 r r r �close_edge_driver� s �����r� )r> r( r$ �logging�seleniumr Z selenium.webdriver.edge.servicer rK Z selenium.webdriver.edge.optionsr rH �selenium.common.exceptionsr r rX � getLoggerr, �environr* r rF rM r r� r� r� r r r r �<module> s*
f c