2023-12-03 21:25:50 +08:00
|
|
|
import json
|
2023-12-05 00:13:20 +08:00
|
|
|
import os
|
|
|
|
|
import sys
|
2023-12-03 21:25:50 +08:00
|
|
|
|
2023-12-05 00:13:20 +08:00
|
|
|
from flask import Flask, render_template, send_file
|
2023-11-28 21:32:24 +08:00
|
|
|
from pyecharts import options as opts
|
|
|
|
|
from pyecharts.charts import Bar
|
|
|
|
|
from pyecharts.globals import ThemeType
|
|
|
|
|
|
2023-12-03 21:25:50 +08:00
|
|
|
from app.DataBase import msg_db
|
|
|
|
|
from app.analysis import analysis
|
2023-12-05 00:13:20 +08:00
|
|
|
from app.person_pc import ContactPC, MePC
|
2023-12-03 21:25:50 +08:00
|
|
|
|
2023-11-28 21:32:24 +08:00
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
2023-12-05 00:13:20 +08:00
|
|
|
wxid = ''
|
|
|
|
|
contact: ContactPC = None
|
|
|
|
|
|
2023-11-28 21:32:24 +08:00
|
|
|
|
|
|
|
|
@app.route("/")
|
|
|
|
|
def index():
|
|
|
|
|
# 创建一个简单的柱状图
|
|
|
|
|
bar = (
|
|
|
|
|
Bar(init_opts=opts.InitOpts(theme=ThemeType.LIGHT))
|
|
|
|
|
.add_xaxis(["A", "B", "C", "D", "E"])
|
|
|
|
|
.add_yaxis("Series", [5, 20, 36, 10, 75])
|
|
|
|
|
.set_global_opts(title_opts=opts.TitleOpts(title="Flask and Pyecharts Interaction"))
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# 将图表转换成 HTML
|
|
|
|
|
chart_html = bar.render_embed()
|
|
|
|
|
|
|
|
|
|
# 渲染模板，并传递图表的 HTML 到模板中
|
|
|
|
|
return render_template("index.html", chart_html=chart_html)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@app.route("/index")
|
|
|
|
|
def index0():
|
2023-12-03 21:25:50 +08:00
|
|
|
return render_template("index1.html")
|
2023-11-28 21:32:24 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@app.route('/home')
|
|
|
|
|
def home():
|
2023-12-05 20:39:21 +08:00
|
|
|
try:
|
|
|
|
|
data = {
|
|
|
|
|
'sub_title': '二零二三年度报告',
|
|
|
|
|
'avatar_path': contact.avatar_path,
|
|
|
|
|
'nickname': contact.remark,
|
|
|
|
|
'first_time': msg_db.get_first_time_of_message(contact.wxid)[1],
|
|
|
|
|
}
|
|
|
|
|
return render_template('home.html', **data)
|
|
|
|
|
except IndexError:
|
|
|
|
|
return set_text('咱就是说，一次都没聊过就别分析了')
|
|
|
|
|
|
2023-11-28 21:32:24 +08:00
|
|
|
|
|
|
|
|
|
2023-12-05 00:13:20 +08:00
|
|
|
@app.route('/wordcloud')
|
2023-11-28 21:32:24 +08:00
|
|
|
def one():
|
2023-12-05 00:13:20 +08:00
|
|
|
wxid = contact.wxid
|
2023-12-03 22:19:25 +08:00
|
|
|
# wxid = 'wxid_lltzaezg38so22'
|
2023-12-03 21:25:50 +08:00
|
|
|
world_cloud_data = analysis.wordcloud(wxid)
|
2023-12-05 00:13:20 +08:00
|
|
|
|
|
|
|
|
with open('wordcloud.html', 'w', encoding='utf-8') as f:
|
|
|
|
|
f.write(render_template('wordcloud.html', **world_cloud_data))
|
|
|
|
|
return render_template('wordcloud.html', **world_cloud_data)
|
2023-12-03 21:25:50 +08:00
|
|
|
|
|
|
|
|
|
2023-12-05 20:39:21 +08:00
|
|
|
def set_text(text):
|
|
|
|
|
html = '''
|
|
|
|
|
<!DOCTYPE html>
|
|
|
|
|
<html lang="en">
|
|
|
|
|
<head>
|
|
|
|
|
<meta charset="UTF-8">
|
|
|
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
|
|
|
<title>Centered Text</title>
|
|
|
|
|
<style>
|
|
|
|
|
body {
|
|
|
|
|
display: flex;
|
|
|
|
|
align-items: center;
|
|
|
|
|
justify-content: center;
|
|
|
|
|
height: 100vh;
|
|
|
|
|
margin: 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
.centered-text {
|
|
|
|
|
font-size: 2em;
|
|
|
|
|
text-align: center;
|
|
|
|
|
}
|
|
|
|
|
</style>
|
|
|
|
|
</head>
|
|
|
|
|
<body>
|
|
|
|
|
<div class="centered-text">
|
|
|
|
|
<!-- 这里是要显示的四个大字 -->
|
|
|
|
|
%s
|
|
|
|
|
</div>
|
|
|
|
|
</body>
|
|
|
|
|
</html>
|
|
|
|
|
''' % (text)
|
|
|
|
|
return html
|
|
|
|
|
|
|
|
|
|
|
2023-12-03 21:25:50 +08:00
|
|
|
@app.route('/test')
|
|
|
|
|
def test():
|
2023-12-05 20:39:21 +08:00
|
|
|
return set_text('以下内容仅对VIP开放🐶')
|
2023-11-28 21:32:24 +08:00
|
|
|
|
|
|
|
|
|
2023-12-05 00:13:20 +08:00
|
|
|
def run(port=21314):
|
|
|
|
|
app.run(debug=True, host='0.0.0.0', port=port, use_reloader=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def resource_path(relative_path):
|
|
|
|
|
""" Get absolute path to resource, works for dev and for PyInstaller """
|
|
|
|
|
base_path = getattr(sys, '_MEIPASS', os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
|
return os.path.join(base_path, relative_path)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@app.route('/data/avatar/<filename>')
|
|
|
|
|
def get_image(filename):
|
|
|
|
|
try:
|
|
|
|
|
# 返回动态生成的图片
|
|
|
|
|
return send_file(os.path.join("../../data/avatar/", filename), mimetype='image/png')
|
|
|
|
|
except:
|
|
|
|
|
return send_file(os.path.join(f"{os.getcwd()}/data/avatar/", filename), mimetype='image/png')
|
|
|
|
|
|
|
|
|
|
|
2023-11-28 21:32:24 +08:00
|
|
|
if __name__ == "__main__":
|
|
|
|
|
app.run(debug=True, host='0.0.0.0')
|
