新增手动获取基址

app/analysis/analysis.py

@@ -1,77 +1,9 @@
 import os
 from collections import Counter
-
-from PyQt5.QtCore import QFile, QTextStream, QIODevice
-
 import sys
-
-sys.path.append('.')
-
 from app.DataBase import msg_db, MsgType
 from pyecharts import options as opts
 from pyecharts.charts import WordCloud, Calendar, Bar, Line
-from app.resources import resource_rc
-
-var = resource_rc.qt_resource_name
-charts_width = 800
-charts_height = 450
-wordcloud_width = 780
-wordcloud_height = 720
-
-
-def wordcloud(wxid, is_Annual_report=False, year='2023', who='1'):
-    import jieba
-    txt_messages = msg_db.get_messages_by_type(wxid, MsgType.TEXT, year)
-    if not txt_messages:
-        return {
-            'chart_data': None,
-            'keyword': "没有聊天你想分析啥",
-            'max_num': "0",
-            'dialogs': []
-        }
-    # text = ''.join(map(lambda x: x[7], txt_messages))
-    text = ''.join(map(lambda x: x[7] if x[4] == int(who) else '', txt_messages))  # 1“我”说的话，0“Ta”说的话
-
-    total_msg_len = len(text)
-    # 使用jieba进行分词，并加入停用词
-    words = jieba.cut(text)
-    # 统计词频
-    word_count = Counter(words)
-    # 过滤停用词
-    stopwords_file = './app000/data/stopwords.txt'
-    try:
-        with open(stopwords_file, "r", encoding="utf-8") as stopword_file:
-            stopwords = set(stopword_file.read().splitlines())
-    except:
-        file = QFile(':/data/stopwords.txt')
-        if file.open(QIODevice.ReadOnly | QIODevice.Text):
-            stream = QTextStream(file)
-            stream.setCodec('utf-8')
-            content = stream.readAll()
-            file.close()
-            stopwords = set(content.splitlines())
-    filtered_word_count = {word: count for word, count in word_count.items() if len(word) > 1 and word not in stopwords}
-
-    # 转换为词云数据格式
-    data = [(word, count) for word, count in filtered_word_count.items()]
-    # text_data = data
-    data.sort(key=lambda x: x[1], reverse=True)
-
-    text_data = data[:100] if len(data) > 100 else data
-    # 创建词云图
-    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=[5, 40])
-    )
-    # return w.render_embed()
-    return {
-        'chart_data': w.dump_options_with_quotes(),
-        'keyword': keyword,
-        'max_num': str(max_num),
-        'dialogs': msg_db.get_messages_by_keyword(wxid, keyword, num=5, max_len=12)
-    }
-
 
 def wordcloud_(wxid, time_range=None):
     import jieba
@@ -115,7 +47,7 @@ def wordcloud_(wxid, time_range=None):
     # 创建词云图
     keyword, max_num = text_data[0]
     w = (
-        WordCloud(init_opts=opts.InitOpts(width=f"{wordcloud_width}px", height=f"{wordcloud_height}px"))
+        WordCloud(init_opts=opts.InitOpts())
         .add(series_name="聊天文字", data_pair=text_data, word_size_range=[5, 100])
     )
     # return w.render_embed()
@@ -267,7 +199,7 @@ def hour_count(wxid, is_Annual_report=False, year='2023'):
     y_data = list(map(lambda x: x[1], msg_data))
     x_axis = list(map(lambda x: x[0], msg_data))
     h = (
-        Line(init_opts=opts.InitOpts(width=f"{charts_width}px", height=f"{charts_height}px"))
+        Line(init_opts=opts.InitOpts())
         .add_xaxis(xaxis_data=x_axis)
         .add_yaxis(
             series_name="聊天频率",