696 lines
22 KiB
Python
696 lines
22 KiB
Python
# -*- coding: utf-8 -*-#
|
|
# -------------------------------------------------------------------------------
|
|
# Name: chat_api.py
|
|
# Description:
|
|
# Author: xaoyaoo
|
|
# Date: 2024/01/02
|
|
# -------------------------------------------------------------------------------
|
|
import os
|
|
import time
|
|
import shutil
|
|
from collections import Counter
|
|
from urllib.parse import quote
|
|
from typing import List, Optional
|
|
|
|
from pydantic import BaseModel
|
|
from fastapi import APIRouter,Response
|
|
from starlette.responses import StreamingResponse, FileResponse
|
|
|
|
import pywxdump
|
|
from pywxdump import decrypt_merge,get_core_db
|
|
from pywxdump.db import DBHandler, download_file, dat2img
|
|
from pywxdump.db.export import export_csv, export_json, export_html
|
|
|
|
from .rjson import ReJson, RqJson
|
|
from .utils import error9999, gc, asyncError9999
|
|
|
|
rs_api = APIRouter()
|
|
|
|
|
|
# 是否初始化
|
|
@rs_api.api_route('/is_init', methods=["GET", 'POST'])
|
|
@error9999
|
|
def is_init():
|
|
"""
|
|
是否初始化
|
|
:return:
|
|
"""
|
|
|
|
local_wxids = gc.get_local_wxids()
|
|
if len(local_wxids) > 1:
|
|
return ReJson(0, True)
|
|
return ReJson(0, False)
|
|
|
|
|
|
# start 以下为聊天联系人相关api *******************************************************************************************
|
|
|
|
@rs_api.api_route('/mywxid', methods=["GET", 'POST'])
|
|
@error9999
|
|
def mywxid():
|
|
"""
|
|
获取我的微信id
|
|
:return:
|
|
"""
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
return ReJson(0, {"my_wxid": my_wxid})
|
|
|
|
|
|
@rs_api.api_route('/user_session_list', methods=["GET", 'POST'])
|
|
@error9999
|
|
def user_session_list():
|
|
"""
|
|
获取联系人列表
|
|
:return:
|
|
"""
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
ret = db.get_session_list()
|
|
return ReJson(0, list(ret.values()))
|
|
|
|
|
|
@rs_api.api_route('/user_labels_dict', methods=["GET", 'POST'])
|
|
@error9999
|
|
def user_labels_dict():
|
|
"""
|
|
获取标签字典
|
|
:return:
|
|
"""
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
user_labels_dict = db.get_labels()
|
|
return ReJson(0, user_labels_dict)
|
|
|
|
|
|
@rs_api.post('/user_list')
|
|
@error9999
|
|
def user_list(word: str = "", wxids: List[str] = None, labels: List[str] = None):
|
|
"""
|
|
获取联系人列表，可用于搜索
|
|
:return:
|
|
"""
|
|
if isinstance(wxids, str) and wxids == '' or wxids is None: wxids = []
|
|
if isinstance(labels, str) and labels == '' or labels is None: labels = []
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
users = db.get_user(word, wxids, labels)
|
|
return ReJson(0, users)
|
|
|
|
|
|
# end 以上为聊天联系人相关api *********************************************************************************************
|
|
|
|
# start 以下为聊天记录相关api *********************************************************************************************
|
|
|
|
class MsgCountRequest(BaseModel):
|
|
wxids: Optional[List[str]]
|
|
|
|
|
|
@rs_api.post('/msg_count')
|
|
@error9999
|
|
def msg_count(request: MsgCountRequest):
|
|
"""
|
|
获取联系人的聊天记录数量
|
|
:return:
|
|
"""
|
|
wxids = request.wxids
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_db_config()
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
count = db.get_msgs_count(wxids)
|
|
return ReJson(0, count)
|
|
|
|
|
|
class MsgListRequest(BaseModel):
|
|
wxid: str
|
|
start: int
|
|
limit: int
|
|
|
|
|
|
@rs_api.api_route('/msg_list', methods=["GET", 'POST'])
|
|
@error9999
|
|
def get_msgs(request: MsgListRequest):
|
|
"""
|
|
获取联系人的聊天记录
|
|
:return:
|
|
"""
|
|
wxid = request.wxid
|
|
start = request.start
|
|
limit = request.limit
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
msgs, users = db.get_msgs(wxid=wxid, start_index=start, page_size=limit)
|
|
return ReJson(0, {"msg_list": msgs, "user_list": users})
|
|
|
|
|
|
@rs_api.get('/imgsrc')
|
|
@asyncError9999
|
|
async def get_imgsrc(src: str):
|
|
"""
|
|
获取图片,
|
|
1. 从网络获取图片，主要功能只是下载图片，缓存到本地
|
|
2. 读取本地图片
|
|
:return:
|
|
"""
|
|
imgsrc = src
|
|
if not imgsrc:
|
|
return ReJson(1002)
|
|
if imgsrc.startswith("FileStorage"): # 如果是本地图片文件则调用get_img
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
wx_path = gc.get_conf(my_wxid, "wx_path")
|
|
|
|
img_path = imgsrc.replace("\\\\", "\\")
|
|
|
|
img_tmp_path = os.path.join(gc.work_path, my_wxid, "img")
|
|
original_img_path = os.path.join(wx_path, img_path)
|
|
if os.path.exists(original_img_path):
|
|
rc, fomt, md5, out_bytes = dat2img(original_img_path)
|
|
if not rc:
|
|
return ReJson(1001, body=original_img_path)
|
|
imgsavepath = os.path.join(str(img_tmp_path), img_path + "_" + "".join([md5, fomt]))
|
|
if os.path.exists(imgsavepath):
|
|
return FileResponse(imgsavepath)
|
|
if not os.path.exists(os.path.dirname(imgsavepath)):
|
|
os.makedirs(os.path.dirname(imgsavepath))
|
|
with open(imgsavepath, "wb") as f:
|
|
f.write(out_bytes)
|
|
return Response(content=out_bytes, media_type="image/jpeg")
|
|
else:
|
|
return ReJson(1001, body=f"{original_img_path} not exists")
|
|
elif imgsrc.startswith("http://") or imgsrc.startswith("https://"):
|
|
# 将?后面的参数连接到imgsrc
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
|
|
img_tmp_path = os.path.join(gc.work_path, my_wxid, "imgsrc")
|
|
if not os.path.exists(img_tmp_path):
|
|
os.makedirs(img_tmp_path)
|
|
file_name = imgsrc.replace("http://", "").replace("https://", "").replace("/", "_").replace("?", "_")
|
|
file_name = file_name + ".jpg"
|
|
# 如果文件名过长，则将文件明分为目录和文件名
|
|
if len(file_name) > 255:
|
|
file_name = file_name[:255] + "/" + file_name[255:]
|
|
|
|
img_path_all = os.path.join(str(img_tmp_path), file_name)
|
|
if os.path.exists(img_path_all):
|
|
return FileResponse(img_path_all)
|
|
else:
|
|
# proxies = {
|
|
# "http": "http://127.0.0.1:10809",
|
|
# "https": "http://127.0.0.1:10809",
|
|
# }
|
|
proxies = None
|
|
download_file(imgsrc, img_path_all, proxies=proxies)
|
|
if os.path.exists(img_path_all):
|
|
return FileResponse(img_path_all)
|
|
else:
|
|
return ReJson(4004, body=imgsrc)
|
|
else:
|
|
return ReJson(1002, body=imgsrc)
|
|
|
|
|
|
@rs_api.api_route('/video', methods=["GET", 'POST'])
|
|
def get_video(src: str):
|
|
"""
|
|
获取视频
|
|
:return:
|
|
"""
|
|
videoPath = src
|
|
if not videoPath:
|
|
return ReJson(1002)
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
wx_path = gc.get_conf(my_wxid, "wx_path")
|
|
|
|
videoPath = videoPath.replace("\\\\", "\\")
|
|
|
|
video_tmp_path = os.path.join(gc.work_path, my_wxid, "video")
|
|
original_img_path = os.path.join(wx_path, videoPath)
|
|
if not os.path.exists(original_img_path):
|
|
return ReJson(5002)
|
|
# 复制文件到临时文件夹
|
|
assert isinstance(video_tmp_path, str)
|
|
video_save_path = os.path.join(video_tmp_path, videoPath)
|
|
if not os.path.exists(os.path.dirname(video_save_path)):
|
|
os.makedirs(os.path.dirname(video_save_path))
|
|
if os.path.exists(video_save_path):
|
|
return FileResponse(path=video_save_path)
|
|
shutil.copy(original_img_path, video_save_path)
|
|
return FileResponse(path=video_save_path)
|
|
|
|
|
|
@rs_api.api_route('/audio', methods=["GET", 'POST'])
|
|
def get_audio(src: str):
|
|
"""
|
|
获取语音
|
|
:return:
|
|
"""
|
|
savePath = src.replace("audio\\", "")
|
|
if not savePath:
|
|
return ReJson(1002)
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
|
|
savePath = os.path.join(gc.work_path, my_wxid, "audio", savePath) # 这个是从url中获取的
|
|
if os.path.exists(savePath):
|
|
assert isinstance(savePath, str)
|
|
return FileResponse(path=savePath, media_type='audio/mpeg')
|
|
|
|
MsgSvrID = savePath.split("_")[-1].replace(".wav", "")
|
|
if not savePath:
|
|
return ReJson(1002)
|
|
|
|
# 判断savePath路径的文件夹是否存在
|
|
if not os.path.exists(os.path.dirname(savePath)):
|
|
os.makedirs(os.path.dirname(savePath))
|
|
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
wave_data = db.get_audio(MsgSvrID, is_play=False, is_wave=True, save_path=savePath, rate=24000)
|
|
if not wave_data:
|
|
return ReJson(1001, body="wave_data is required")
|
|
|
|
if os.path.exists(savePath):
|
|
assert isinstance(savePath, str)
|
|
return FileResponse(path=savePath, media_type='audio/mpeg')
|
|
else:
|
|
return ReJson(4004, body=savePath)
|
|
|
|
|
|
class FileInfoRequest(BaseModel):
|
|
file_path: str
|
|
|
|
|
|
@rs_api.api_route('/file_info', methods=["GET", 'POST'])
|
|
def get_file_info(request: FileInfoRequest):
|
|
file_path = request.file_path
|
|
if not file_path:
|
|
return ReJson(1002)
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
wx_path = gc.get_conf(my_wxid, "wx_path")
|
|
|
|
all_file_path = os.path.join(wx_path, file_path)
|
|
if not os.path.exists(all_file_path):
|
|
return ReJson(5002)
|
|
file_name = os.path.basename(all_file_path)
|
|
file_size = os.path.getsize(all_file_path)
|
|
return ReJson(0, {"file_name": file_name, "file_size": str(file_size)})
|
|
|
|
|
|
@rs_api.get('/file')
|
|
def get_file(src: str):
|
|
"""
|
|
获取文件
|
|
:return:
|
|
"""
|
|
file_path = src
|
|
if not file_path:
|
|
return ReJson(1002)
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
wx_path = gc.get_conf(my_wxid, "wx_path")
|
|
|
|
all_file_path = os.path.join(wx_path, file_path)
|
|
if not os.path.exists(all_file_path):
|
|
return ReJson(5002)
|
|
|
|
def file_iterator(file_path, chunk_size=8192):
|
|
with open(file_path, "rb") as f:
|
|
while True:
|
|
chunk = f.read(chunk_size)
|
|
if not chunk:
|
|
break
|
|
yield chunk
|
|
|
|
headers = {
|
|
"Content-Disposition": f'attachment; filename*=UTF-8\'\'{quote(os.path.basename(all_file_path))}',
|
|
}
|
|
return StreamingResponse(file_iterator(all_file_path), media_type="application/octet-stream", headers=headers)
|
|
|
|
|
|
# end 以上为聊天记录相关api *********************************************************************************************
|
|
|
|
# start 导出聊天记录 *****************************************************************************************************
|
|
class ExportEndbRequest(BaseModel):
|
|
wx_path: str = ""
|
|
outpath: str = ""
|
|
key: str = ""
|
|
|
|
|
|
@rs_api.api_route('/export_endb', methods=["GET", 'POST'])
|
|
def get_export_endb(request: ExportEndbRequest):
|
|
"""
|
|
导出加密数据库
|
|
:return:
|
|
"""
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
|
|
wx_path = request.wx_path
|
|
if not wx_path:
|
|
wx_path = gc.get_conf(my_wxid, "wx_path")
|
|
if not os.path.exists(wx_path if wx_path else ""):
|
|
return ReJson(1002, body=f"wx_path is required: {wx_path}")
|
|
|
|
# 分割wx_path的文件名和父目录
|
|
code, wxdbpaths = get_core_db(wx_path)
|
|
if not code:
|
|
return ReJson(2001, body=wxdbpaths)
|
|
|
|
outpath = os.path.join(gc.work_path, "export", my_wxid, "endb")
|
|
if not os.path.exists(outpath):
|
|
os.makedirs(outpath)
|
|
|
|
for wxdb in wxdbpaths:
|
|
# 复制wxdb->outpath, os.path.basename(wxdb)
|
|
assert isinstance(outpath, str) # 为了解决pycharm的警告, 无实际意义
|
|
wxdb_path = wxdb.get("db_path")
|
|
shutil.copy(wxdb_path, os.path.join(outpath, os.path.basename(wxdb_path)))
|
|
return ReJson(0, body=outpath)
|
|
|
|
|
|
class ExportDedbRequest(BaseModel):
|
|
wx_path: str = ""
|
|
outpath: str = ""
|
|
key: str = ""
|
|
|
|
|
|
@rs_api.api_route('/export_dedb', methods=["GET", "POST"])
|
|
def get_export_dedb(request: ExportDedbRequest):
|
|
"""
|
|
导出解密数据库
|
|
:return:
|
|
"""
|
|
key = request.key
|
|
wx_path = request.wx_path
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
|
|
if not key:
|
|
key = gc.get_conf(my_wxid, "key")
|
|
if not wx_path:
|
|
wx_path = gc.get_conf(my_wxid, "wx_path")
|
|
|
|
if not key:
|
|
return ReJson(1002, body=f"key is required: {key}")
|
|
if not wx_path:
|
|
return ReJson(1002, body=f"wx_path is required: {wx_path}")
|
|
if not os.path.exists(wx_path):
|
|
return ReJson(1001, body=f"wx_path not exists: {wx_path}")
|
|
|
|
outpath = os.path.join(gc.work_path, "export", my_wxid, "dedb")
|
|
if not os.path.exists(outpath):
|
|
os.makedirs(outpath)
|
|
assert isinstance(outpath, str)
|
|
code, merge_save_path = decrypt_merge(wx_path=wx_path, key=key, outpath=outpath)
|
|
time.sleep(1)
|
|
if code:
|
|
return ReJson(0, body=merge_save_path)
|
|
else:
|
|
return ReJson(2001, body=merge_save_path)
|
|
|
|
|
|
class ExportCsvRequest(BaseModel):
|
|
wxid: str
|
|
|
|
|
|
@rs_api.api_route('/export_csv', methods=["GET", 'POST'])
|
|
def get_export_csv(request: ExportCsvRequest):
|
|
"""
|
|
导出csv
|
|
:return:
|
|
"""
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
|
|
wxid = request.wxid
|
|
# st_ed_time = request.json.get("datetime", [0, 0])
|
|
if not wxid:
|
|
return ReJson(1002, body=f"username is required: {wxid}")
|
|
# if not isinstance(st_ed_time, list) or len(st_ed_time) != 2:
|
|
# return ReJson(1002, body=f"datetime is required: {st_ed_time}")
|
|
# start, end = st_ed_time
|
|
# if not isinstance(start, int) or not isinstance(end, int) or start >= end:
|
|
# return ReJson(1002, body=f"datetime is required: {st_ed_time}")
|
|
|
|
outpath = os.path.join(gc.work_path, "export", my_wxid, "csv", wxid)
|
|
if not os.path.exists(outpath):
|
|
os.makedirs(outpath)
|
|
|
|
code, ret = export_csv(wxid, outpath, db_config, my_wxid=my_wxid)
|
|
if code:
|
|
return ReJson(0, ret)
|
|
else:
|
|
return ReJson(2001, body=ret)
|
|
|
|
|
|
class ExportJsonRequest(BaseModel):
|
|
wxid: str
|
|
|
|
|
|
@rs_api.api_route('/export_json', methods=["GET", 'POST'])
|
|
def get_export_json(request: ExportJsonRequest):
|
|
"""
|
|
导出json
|
|
:return:
|
|
"""
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
|
|
wxid = request.wxid
|
|
if not wxid:
|
|
return ReJson(1002, body=f"username is required: {wxid}")
|
|
|
|
outpath = os.path.join(gc.work_path, "export", my_wxid, "json", wxid)
|
|
if not os.path.exists(outpath):
|
|
os.makedirs(outpath)
|
|
|
|
code, ret = export_json(wxid, outpath, db_config, my_wxid=my_wxid)
|
|
if code:
|
|
return ReJson(0, ret)
|
|
else:
|
|
return ReJson(2001, body=ret)
|
|
|
|
|
|
class ExportHtmlRequest(BaseModel):
|
|
wxid: str
|
|
|
|
|
|
@rs_api.api_route('/export_html', methods=["GET", 'POST'])
|
|
def get_export_html(request: ExportHtmlRequest):
|
|
"""
|
|
导出json
|
|
:return:
|
|
"""
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
|
|
wxid = request.wxid
|
|
if not wxid:
|
|
return ReJson(1002, body=f"username is required: {wxid}")
|
|
|
|
html_outpath = os.path.join(gc.work_path, "export", my_wxid, "html")
|
|
if not os.path.exists(html_outpath):
|
|
os.makedirs(html_outpath)
|
|
assert isinstance(html_outpath, str)
|
|
outpath = os.path.join(html_outpath, wxid)
|
|
if os.path.exists(outpath):
|
|
shutil.rmtree(outpath, ignore_errors=True)
|
|
# 复制pywxdump/ui/web/*到outpath
|
|
web_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "ui", "web")
|
|
shutil.copytree(web_path, outpath)
|
|
|
|
code, ret = export_html(wxid, outpath, db_config, my_wxid=my_wxid)
|
|
|
|
if code:
|
|
return ReJson(0, ret)
|
|
else:
|
|
return ReJson(2001, body=ret)
|
|
|
|
|
|
# end 导出聊天记录 *******************************************************************************************************
|
|
|
|
# start 聊天记录分析api **************************************************************************************************
|
|
class DateCountRequest(BaseModel):
|
|
wxid: str = ""
|
|
start_time: int = 0
|
|
end_time: int = 0
|
|
time_format: str = "%Y-%m-%d"
|
|
|
|
|
|
@rs_api.api_route('/date_count', methods=["GET", 'POST'])
|
|
def get_date_count(request: DateCountRequest):
|
|
"""
|
|
获取日期统计
|
|
:return:
|
|
"""
|
|
wxid = request.wxid
|
|
start_time = request.start_time
|
|
end_time = request.end_time
|
|
time_format = request.time_format
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
date_count = db.get_date_count(wxid=wxid, start_time=start_time, end_time=end_time, time_format=time_format)
|
|
return ReJson(0, date_count)
|
|
|
|
|
|
class TopTalkerCountRequest(BaseModel):
|
|
top: int = 10
|
|
start_time: int = 0
|
|
end_time: int = 0
|
|
|
|
|
|
@rs_api.api_route('/top_talker_count', methods=["GET", 'POST'])
|
|
def get_top_talker_count(request: TopTalkerCountRequest):
|
|
"""
|
|
获取最多聊天的人
|
|
:return:
|
|
"""
|
|
top = request.top
|
|
start_time = request.start_time
|
|
end_time = request.end_time
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
date_count = DBHandler(db_config, my_wxid=my_wxid).get_top_talker_count(top=top, start_time=start_time,
|
|
end_time=end_time)
|
|
return ReJson(0, date_count)
|
|
|
|
|
|
class WordCloudRequest(BaseModel):
|
|
target: str = "signature"
|
|
|
|
|
|
@rs_api.api_route('/wordcloud', methods=["GET", 'POST'])
|
|
@error9999
|
|
def get_wordcloud(request: WordCloudRequest):
|
|
try:
|
|
import jieba
|
|
except ImportError:
|
|
return ReJson(9999, body="jieba is required")
|
|
|
|
target = request.target
|
|
if not target:
|
|
return ReJson(1002, body="target is required")
|
|
|
|
my_wxid = gc.get_conf(gc.at, "last")
|
|
if not my_wxid: return ReJson(1001, body="my_wxid is required")
|
|
db_config = gc.get_conf(my_wxid, "db_config")
|
|
db = DBHandler(db_config, my_wxid=my_wxid)
|
|
|
|
if target == "signature":
|
|
users = db.get_user()
|
|
signature_list = []
|
|
for wxid, user in users.items():
|
|
ExtraBuf = user.get("ExtraBuf", {})
|
|
signature = ExtraBuf.get("个性签名", "") if ExtraBuf else ""
|
|
if signature:
|
|
signature_list.append(signature)
|
|
signature_str = " ".join(signature_list)
|
|
words = jieba.lcut(signature_str)
|
|
words = [word for word in words if len(word) > 1]
|
|
count_dict = dict(Counter(words))
|
|
return ReJson(0, count_dict)
|
|
elif target == "nickname":
|
|
users = db.get_user()
|
|
nickname_list = []
|
|
for wxid, user in users.items():
|
|
nickname = user.get("nickname", "")
|
|
if nickname:
|
|
nickname_list.append(nickname)
|
|
nickname_str = " ".join(nickname_list)
|
|
words = jieba.lcut(nickname_str)
|
|
words = [word for word in words if len(word) > 1]
|
|
count_dict = dict(Counter(words))
|
|
return ReJson(0, count_dict)
|
|
|
|
return ReJson(1002, body="target is required")
|
|
|
|
|
|
# end 聊天记录分析api ****************************************************************************************************
|
|
|
|
# 关于、帮助、设置 *******************************************************************************************************
|
|
@rs_api.api_route('/check_update', methods=["GET", 'POST'])
|
|
@error9999
|
|
def check_update():
|
|
"""
|
|
检查更新
|
|
:return:
|
|
"""
|
|
url = "https://api.github.com/repos/xaoyaoo/PyWxDump/tags"
|
|
try:
|
|
import requests
|
|
res = requests.get(url)
|
|
if res.status_code == 200:
|
|
data = res.json()
|
|
NEW_VERSION = data[0].get("name")
|
|
if NEW_VERSION[1:] != pywxdump.__version__:
|
|
msg = "有新版本"
|
|
else:
|
|
msg = "已经是最新版本"
|
|
return ReJson(0, body={"msg": msg, "latest_version": NEW_VERSION,
|
|
"latest_url": "https://github.com/xaoyaoo/PyWxDump/releases/tag/" + NEW_VERSION})
|
|
else:
|
|
return ReJson(2001, body="status_code is not 200")
|
|
except Exception as e:
|
|
return ReJson(9999, msg=str(e))
|
|
|
|
|
|
@rs_api.api_route('/version', methods=["GET", "POST"])
|
|
@error9999
|
|
def version():
|
|
"""
|
|
版本
|
|
:return:
|
|
"""
|
|
return ReJson(0, pywxdump.__version__)
|
|
|
|
|
|
@rs_api.api_route('/get_readme', methods=["GET", 'POST'])
|
|
@error9999
|
|
def get_readme():
|
|
"""
|
|
版本
|
|
:return:
|
|
"""
|
|
url = "https://raw.githubusercontent.com/xaoyaoo/PyWxDump/master/doc/README_CN.md"
|
|
import requests
|
|
res = requests.get(url)
|
|
if res.status_code == 200:
|
|
data = res.text
|
|
data = data.replace("# <center>PyWxDump</center>", "")
|
|
return ReJson(0, body=data)
|
|
else:
|
|
return ReJson(2001, body="status_code is not 200")
|
|
|
|
|
|
# END 关于、帮助、设置 ***************************************************************************************************
|