支持添加自定义停用词#196 #197 #68

app/config.py

@@ -1,4 +1,4 @@
-version = '0.2.9'
+version = '1.0.0'
 contact = '701805520'
 github = 'https://github.com/LC044/WeChatMsg'
 website = 'https://memotrace.lc044.love/'
@@ -9,7 +9,7 @@ description = [
     '3. 支持导出聊天记录<br>&nbsp;&nbsp;&nbsp;&nbsp;* csv<br>&nbsp;&nbsp;&nbsp;&nbsp;* html<br>&nbsp;&nbsp;&nbsp;&nbsp;* '
     'txt<br>',
     '4. 查找联系人<br>',
-    '5. 生成年度报告<br>',
+    '5. 生成年度报告——圣诞特别版<br>',
 ]
 about = f'''
     版本：{version}<br>

app/resources/stopwords.txt

@@ -13,27 +13,7 @@ wxid
 行
 拿
 你
-裂开
-苦涩
-叹气
-凋谢
-让我看看
-酷
-奋斗
-疑问
-擦汗
-抠鼻
-鄙视
-勾引
-奸笑
-嘿哈
-捂脸
-机智
-加油
-吃瓜
-尴尬
 乡村
-炸弹
 腹肌
 徐坤
 振兴
@@ -2557,5 +2537,4 @@ sup
 打
 🙄
 奥
-真
+真
-旺柴

app/ui/tool/setting/setting.py

@@ -0,0 +1,37 @@
+import json
+import os.path
+from PyQt5.QtCore import pyqtSignal
+from PyQt5.QtWidgets import QWidget, QMessageBox
+
+from .settingUi import Ui_Form
+
+
+class SettingControl(QWidget, Ui_Form):
+
+    def __init__(self, parent=None):
+        super(SettingControl, self).__init__(parent)
+        self.setupUi(self)
+        self.btn_addstopword.clicked.connect(self.add_stopwords)
+        self.read_data()
+
+    def read_data(self):
+        os.makedirs('./app/data', exist_ok=True)
+        stopwords = ['裂开','苦涩','叹气','凋谢','让我看看','酷','奋斗','疑问','擦汗','抠鼻','鄙视','勾引','奸笑','嘿哈','捂脸','机智','加油','吃瓜','尴尬','炸弹','旺柴']
+        if os.path.exists('./app/data/stopwords.txt'):
+            with open('./app/data/stopwords.txt', 'r', encoding='utf-8') as f:
+                stopwords = set(f.read().splitlines())
+                self.plainTextEdit.setPlainText(' '.join(stopwords))
+        else:
+            self.plainTextEdit.setPlainText(' '.join(stopwords))
+            stopwords = '\n'.join(stopwords)
+            with open('./app/data/stopwords.txt', 'w', encoding='utf-8') as f:
+                f.write(stopwords)
+
+
+
+    def add_stopwords(self):
+        text = self.plainTextEdit.toPlainText()
+        stopwords = '\n'.join(text.split())
+        with open('./app/data/stopwords.txt', 'w', encoding='utf-8') as f:
+            f.write(stopwords)
+        QMessageBox.about(self, "添加成功", "停用词添加成功")

app/ui/tool/tool_window.py

@@ -4,6 +4,7 @@ from PyQt5.QtWidgets import QWidget, QListWidgetItem, QLabel
 
 from app.ui.Icon import Icon
 from .pc_decrypt import DecryptControl
+from .setting.setting import SettingControl
 from .toolUI import Ui_Dialog
 
 # 美化样式表
@@ -60,13 +61,15 @@ class ToolWindow(QWidget, Ui_Dialog):
         self.listWidget.clear()
         self.listWidget.currentRowChanged.connect(self.setCurrentIndex)
         chat_item = QListWidgetItem(Icon.Decrypt_Icon, '解密', self.listWidget)
-        contact_item = QListWidgetItem(Icon.Contact_Icon, '别点', self.listWidget)
+        contact_item = QListWidgetItem(Icon.Contact_Icon, '设置', self.listWidget)
         myinfo_item = QListWidgetItem(Icon.Home_Icon, '别点', self.listWidget)
         tool_item = QListWidgetItem(Icon.Home_Icon, '别点', self.listWidget)
         decrypt_window = DecryptControl()
         decrypt_window.get_wxidSignal.connect(self.get_info_signal)
         decrypt_window.DecryptSignal.connect(self.decrypt_success_signal)
         self.stackedWidget.addWidget(decrypt_window)
+        setting_window = SettingControl()
+        self.stackedWidget.addWidget(setting_window)
         label = QLabel('都说了不让你点', self)
         label.setFont(QFont("微软雅黑", 50))
         label.setAlignment(Qt.AlignCenter)
@@ -74,7 +77,7 @@ class ToolWindow(QWidget, Ui_Dialog):
         # 这里加了一个margin边距(方便区分QStackedWidget和QLabel的颜色)
         # label.setStyleSheet('background: rgb(%d, %d, %d);margin: 50px;' % (
         #     randint(0, 255), randint(0, 255), randint(0, 255)))
-        self.stackedWidget.addWidget(label)
+
         self.stackedWidget.addWidget(label)
         self.stackedWidget.addWidget(label)
         self.listWidget.setCurrentRow(0)