import os
import sys
from flask import Flask, render_template, send_file
from app.DataBase import msg_db
from app.analysis import analysis
from app.person_pc import ContactPC, MePC
app = Flask(__name__)
wxid = ''
contact: ContactPC = None
@app.route("/")
def index():
# 渲染模板，并传递图表的 HTML 到模板中
return render_template("index.html")
@app.route('/home')
def home():
try:
first_message, first_time = msg_db.get_first_time_of_message(contact.wxid)
except TypeError:
return set_text('咱就是说，一次都没聊过就别分析了')
data = {
'sub_title': '二零二三年度报告',
'avatar_path': contact.avatar_path,
'nickname': contact.remark,
'first_time': first_time,
}
return render_template('home.html', **data)
@app.route('/wordcloud')
def one():
wxid = contact.wxid
# wxid = 'wxid_lltzaezg38so22'
world_cloud_data = analysis.wordcloud(wxid)
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)
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
<img src="https://res.wx.qq.com/t/wx_fed/we-emoji/res/v1.2.8/assets/newemoji/Yellowdog.png" id="旺柴" class="emoji_img">
</div>
</body>
</html>
''' % (text)
return html
@app.route('/test')
def test():
return set_text('以下内容仅对VIP开放')
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')
if __name__ == "__main__":
app.run(debug=True, host='0.0.0.0')