Merge branch 'LC044:master' into master

app/DataBase/__init__.py

@@ -10,8 +10,6 @@
 from .hard_link import HardLink
 from .micro_msg import MicroMsg
 from .media_msg import MediaMsg
-# from . import data
-# from . import output
 from .misc import Misc
 from .msg import Msg
 from .msg import MsgType
@@ -31,4 +29,12 @@ def close_db():
     media_msg_db.close()
 
 
+def init_db():
+    misc_db.init_database()
+    msg_db.init_database()
+    micro_msg_db.init_database()
+    hard_link_db.init_database()
+    media_msg_db.init_database()
+
+
 __all__ = ['output', 'misc_db', 'micro_msg_db', 'msg_db', 'hard_link_db', 'MsgType', "media_msg_db"]

app/DataBase/msg.py

@@ -206,8 +206,8 @@ class Msg:
                     ('', '', ['', ''], ''),
                     ('', '', '', '')
                 ))
-        print(keyword,res)
         return res
+
     def get_contact(self, contacts):
         if not self.open_flag:
             return None
@@ -227,6 +227,7 @@ class Msg:
                 contacts[i].append(0)
         contacts.sort(key=lambda cur_contact: cur_contact[-1], reverse=True)
         return contacts
+
     def get_messages_by_days(self, username_, is_Annual_report_=False, year_='2023'):
         if is_Annual_report_:
             sql = '''

app/DataBase/output_pc.py

@@ -274,7 +274,6 @@ class ChildThread(QThread):
                 f'''{str_time} {name}\n[语音]\n\n'''
             )
 
-
     def emoji(self, doc, message):
         origin_docx_path = f"{os.path.abspath('.')}/data/聊天记录/{self.contact.remark}"
         str_content = message[7]
@@ -348,7 +347,6 @@ class ChildThread(QThread):
                     f'''{str_time} {name}\n{content.get('title')}\n引用:未知\n\n'''
                 )
 
-
     def system_msg(self, doc, message):
         str_content = message[7]
         is_send = message[4]
@@ -501,6 +499,7 @@ class ChildThread(QThread):
                 elif type_ == 49 and sub_type == 57:
                     self.refermsg(f, message)
         self.okSignal.emit(1)
+
     def run(self):
         if self.output_type == Output.DOCX:
             return

app/analysis/analysis.py

@@ -3,6 +3,7 @@ from collections import Counter
 from PyQt5.QtCore import QFile, QTextStream, QIODevice
 
 import sys
+
 sys.path.append('.')
 
 from app.DataBase import msg_db, MsgType
@@ -60,15 +61,7 @@ def wordcloud(wxid, is_Annual_report=False, year='2023', who='1'):
     keyword, max_num = text_data[0]
     w = (
         WordCloud(init_opts=opts.InitOpts(width=f"{wordcloud_width}px", height=f"{wordcloud_height}px"))
-        .add(series_name="聊天文字", data_pair=text_data, word_size_range=[20, 100])
-        .set_global_opts(
-            title_opts=opts.TitleOpts(
-                title=f"词云图", subtitle=f"总计{total_msg_len}字",
-                title_textstyle_opts=opts.TextStyleOpts(font_size=23)
-            ),
-            tooltip_opts=opts.TooltipOpts(is_show=True),
-            legend_opts=opts.LegendOpts(is_show=False)
-        )
+        .add(series_name="聊天文字", data_pair=text_data, word_size_range=[5, 40])
     )
     # return w.render_embed()
     return {

app/config.py

@@ -1,9 +1,21 @@
-version = '0.2.8'
+version = '0.2.9'
 contact = '701805520'
+github = 'https://github.com/LC044/WeChatMsg'
+website = 'http://memotrace.lc044.love/'
+copyright = '©2023-2023 ShuaikangZhou'
 description = [
     '1. 支持获取个人信息<br>',
     '2. 支持显示聊天界面<br>',
-    '3. 支持导出聊天记录<br>&nbsp;&nbsp;&nbsp;&nbsp;* csv<br>&nbsp;&nbsp;&nbsp;&nbsp;* html<br>&nbsp;&nbsp;&nbsp;&nbsp;* txt<br>',
+    '3. 支持导出聊天记录<br>&nbsp;&nbsp;&nbsp;&nbsp;* csv<br>&nbsp;&nbsp;&nbsp;&nbsp;* html<br>&nbsp;&nbsp;&nbsp;&nbsp;* '
+    'txt<br>',
     '4. 查找联系人<br>',
     '5. 生成年度报告<br>',
 ]
+about = f'''
+    版本：{version}<br>
+    QQ交流群:{contact},加群要求请阅读文档<br>
+    地址：<a href='{github}'>{github}</a><br>
+    官网：<a href='{website}'>{website}</a><br>
+    新特性:<br>{''.join(['' + i for i in description])}<br>
+    Copyright {copyright}
+'''

app/person.py

@@ -53,6 +53,7 @@ class ContactPC:
         self.smallHeadImgBLOG = b''
         self.avatar = QPixmap()
         self.avatar_path = Icon.Default_avatar_path
+        self.is_chatroom = self.wxid.__contains__('@chatroom')
 
     def set_avatar(self, img_bytes):
         if not img_bytes:

app/ui/mainview.py

@@ -225,10 +225,7 @@ class MainWinController(QMainWindow, mainwindow.Ui_MainWindow):
         """
         关于
         """
-        QMessageBox.about(self, "关于",
-                          f'''版本：{config.version}<br>QQ交流群:{config.contact},加群要求请阅读文档<br>地址：<a href='https://github.com/LC044/WeChatMsg'>https://github.com/LC044/WeChatMsg</a><br>新特性:<br>{''.join(['' + i for i in config.description])}
-                            '''
-                          )
+        QMessageBox.about(self, "关于",config.about)
 
     def decrypt_success(self):
         QMessageBox.about(self, "解密成功", "请重新启动")