211 lines
8.4 KiB
Python
211 lines
8.4 KiB
Python
# -*- coding: utf-8 -*-#
|
||
# -------------------------------------------------------------------------------
|
||
# Name: __init__.py
|
||
# Description:
|
||
# Author: xaoyaoo
|
||
# Date: 2023/12/14
|
||
# -------------------------------------------------------------------------------
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
import time
|
||
import uvicorn
|
||
import mimetypes
|
||
import logging
|
||
from logging.handlers import RotatingFileHandler
|
||
|
||
from uvicorn.config import LOGGING_CONFIG
|
||
from fastapi import FastAPI, Request, Path, Query
|
||
from fastapi.staticfiles import StaticFiles
|
||
from fastapi.exceptions import RequestValidationError
|
||
from starlette.middleware.cors import CORSMiddleware
|
||
from starlette.responses import RedirectResponse, FileResponse
|
||
|
||
from .utils import gc, is_port_in_use, server_loger
|
||
from .rjson import ReJson
|
||
from .remote_server import rs_api
|
||
from .local_server import ls_api
|
||
|
||
from pywxdump import __version__
|
||
|
||
|
||
def gen_fastapi_app(handler):
|
||
app = FastAPI(title="wxdump", description="微信工具", version=__version__,
|
||
terms_of_service="https://github.com/xaoyaoo/pywxdump",
|
||
contact={"name": "xaoyaoo", "url": "https://github.com/xaoyaoo/pywxdump"},
|
||
license_info={"name": "MIT License",
|
||
"url": "https://github.com/xaoyaoo/PyWxDump/blob/master/LICENSE"})
|
||
|
||
web_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "ui", "web") # web文件夹路径
|
||
|
||
# 跨域
|
||
origins = [
|
||
"http://localhost:5000",
|
||
"http://127.0.0.1:5000",
|
||
"http://localhost:8080", # 开发环境的客户端地址"
|
||
# "http://0.0.0.0:5000",
|
||
# "*"
|
||
]
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=origins, # 允许所有源
|
||
allow_credentials=True,
|
||
allow_methods=["*"], # 允许所有方法
|
||
allow_headers=["*"], # 允许所有头
|
||
)
|
||
|
||
@app.on_event("startup")
|
||
async def startup_event():
|
||
logger = logging.getLogger("uvicorn")
|
||
logger.addHandler(handler)
|
||
|
||
# 错误处理
|
||
@app.exception_handler(RequestValidationError)
|
||
async def request_validation_exception_handler(request: Request, exc: RequestValidationError):
|
||
# print(request.body)
|
||
return ReJson(1002, {"detail": exc.errors()})
|
||
|
||
# 首页
|
||
@app.get("/")
|
||
@app.get("/index.html")
|
||
async def index():
|
||
response = RedirectResponse(url="/s/index.html", status_code=307)
|
||
return response
|
||
|
||
# 路由挂载
|
||
app.include_router(rs_api, prefix='/api/rs', tags=['远程api'])
|
||
app.include_router(ls_api, prefix='/api/ls', tags=['本地api'])
|
||
|
||
# 根据文件类型，设置mime_type，返回文件
|
||
@app.get("/s/{filename:path}")
|
||
async def serve_file(filename: str):
|
||
# 构建完整的文件路径
|
||
file_path = os.path.join(web_path, filename)
|
||
file_path = os.path.abspath(file_path)
|
||
|
||
# 检查文件是否存在
|
||
if os.path.isfile(file_path):
|
||
# 获取文件 MIME 类型
|
||
mime_type, _ = mimetypes.guess_type(file_path)
|
||
# 如果 MIME 类型为空，则默认为 application/octet-stream
|
||
if mime_type is None:
|
||
mime_type = "application/octet-stream"
|
||
server_loger.warning(f"[+] 无法获取文件 MIME 类型，使用默认值：{mime_type}")
|
||
if file_path.endswith(".js"):
|
||
mime_type = "text/javascript"
|
||
server_loger.info(f"[+] 文件 {file_path} MIME 类型：{mime_type}")
|
||
# 返回文件
|
||
return FileResponse(file_path, media_type=mime_type)
|
||
|
||
# 如果文件不存在，返回 404
|
||
return {"detail": "Not Found"}, 404
|
||
|
||
# 静态文件挂载
|
||
# if os.path.exists(os.path.join(web_path, "index.html")):
|
||
# app.mount("/s", StaticFiles(directory=web_path), name="static")
|
||
|
||
return app
|
||
|
||
|
||
def start_server(port=5000, online=False, debug=False, isopenBrowser=True,
|
||
merge_path="", wx_path="", my_wxid="", ):
|
||
"""
|
||
启动flask
|
||
:param port: 端口号
|
||
:param online: 是否在线查看(局域网查看)
|
||
:param debug: 是否开启debug模式
|
||
:param isopenBrowser: 是否自动打开浏览器
|
||
:return:
|
||
"""
|
||
work_path = os.path.join(os.getcwd(), "wxdump_work") # 临时文件夹,用于存放图片等 # 全局变量
|
||
if not os.path.exists(work_path):
|
||
os.makedirs(work_path, exist_ok=True)
|
||
server_loger.info(f"[+] 创建临时文件夹：{work_path}")
|
||
print(f"[+] 创建临时文件夹：{work_path}")
|
||
|
||
# 日志处理，写入到文件
|
||
log_format = '[{levelname[0]}] {asctime} [{name}:{levelno}] {pathname}:{lineno} {message}'
|
||
log_datefmt = '%Y-%m-%d %H:%M:%S'
|
||
log_file_path = os.path.join(work_path, "wxdump.log")
|
||
file_handler = RotatingFileHandler(log_file_path, mode="a", maxBytes=10 * 1024 * 1024, backupCount=3)
|
||
formatter = logging.Formatter(fmt=log_format, datefmt=log_datefmt, style='{')
|
||
file_handler.setFormatter(formatter)
|
||
|
||
wx_core_logger = logging.getLogger("wx_core")
|
||
db_prepare = logging.getLogger("db_prepare")
|
||
|
||
# 这几个日志处理器为本项目的日志处理器
|
||
server_loger.addHandler(file_handler)
|
||
wx_core_logger.addHandler(file_handler)
|
||
db_prepare.addHandler(file_handler)
|
||
|
||
conf_file = os.path.join(work_path, "conf_auto.json") # 用于存放各种基础信息
|
||
auto_setting = "auto_setting"
|
||
env_file = os.path.join(work_path, ".env") # 用于存放环境变量
|
||
# set 环境变量
|
||
os.environ["PYWXDUMP_WORK_PATH"] = work_path
|
||
os.environ["PYWXDUMP_CONF_FILE"] = conf_file
|
||
os.environ["PYWXDUMP_AUTO_SETTING"] = auto_setting
|
||
|
||
with open(env_file, "w", encoding="utf-8") as f:
|
||
f.write(f"PYWXDUMP_WORK_PATH = '{work_path}'\n")
|
||
f.write(f"PYWXDUMP_CONF_FILE = '{conf_file}'\n")
|
||
f.write(f"PYWXDUMP_AUTO_SETTING = '{auto_setting}'\n")
|
||
|
||
if merge_path and os.path.exists(merge_path):
|
||
my_wxid = my_wxid if my_wxid else "wxid_dbshow"
|
||
gc.set_conf(my_wxid, "wxid", my_wxid) # 初始化wxid
|
||
gc.set_conf(my_wxid, "merge_path", merge_path) # 初始化merge_path
|
||
gc.set_conf(my_wxid, "wx_path", wx_path) # 初始化wx_path
|
||
db_config = {"key": my_wxid, "type": "sqlite", "path": merge_path}
|
||
gc.set_conf(my_wxid, "db_config", db_config) # 初始化db_config
|
||
gc.set_conf(auto_setting, "last", my_wxid) # 初始化last
|
||
|
||
# 检查端口是否被占用
|
||
if online:
|
||
host = '0.0.0.0'
|
||
else:
|
||
host = "127.0.0.1"
|
||
|
||
if is_port_in_use(host, port):
|
||
server_loger.error(f"Port {port} is already in use. Choose a different port.")
|
||
print(f"Port {port} is already in use. Choose a different port.")
|
||
input("Press Enter to exit...")
|
||
return # 退出程序
|
||
if isopenBrowser:
|
||
try:
|
||
# 自动打开浏览器
|
||
url = f"http://127.0.0.1:{port}/"
|
||
# 根据操作系统使用不同的命令打开默认浏览器
|
||
if sys.platform.startswith('darwin'): # macOS
|
||
subprocess.call(['open', url])
|
||
elif sys.platform.startswith('win'): # Windows
|
||
subprocess.call(['start', url], shell=True)
|
||
elif sys.platform.startswith('linux'): # Linux
|
||
subprocess.call(['xdg-open', url])
|
||
else:
|
||
server_loger.error(f"Unsupported platform, can't open browser automatically.", exc_info=True)
|
||
print("Unsupported platform, can't open browser automatically.")
|
||
except Exception as e:
|
||
server_loger.error(f"自动打开浏览器失败：{e}", exc_info=True)
|
||
|
||
time.sleep(1)
|
||
server_loger.info(f"启动flask服务，host:port：{host}:{port}")
|
||
print("[+] 请使用浏览器访问 http://127.0.0.1:5000/ 查看聊天记录")
|
||
global app
|
||
print("[+] 如需查看api文档，请访问 http://127.0.0.1:5000/docs ")
|
||
app = gen_fastapi_app(file_handler)
|
||
|
||
LOGGING_CONFIG["formatters"]["default"]["fmt"] = "[%(asctime)s] %(levelprefix)s %(message)s"
|
||
LOGGING_CONFIG["formatters"]["access"][
|
||
"fmt"] = '[%(asctime)s] %(levelprefix)s %(client_addr)s - "%(request_line)s" %(status_code)s'
|
||
config = uvicorn.Config(app=app, host=host, port=port, reload=debug, log_level="info", workers=1, env_file=env_file)
|
||
server = uvicorn.Server(config)
|
||
server.run()
|
||
# uvicorn.run(app=app, host=host, port=port, reload=debug, log_level="info", workers=1, env_file=env_file)
|
||
|
||
|
||
app = None
|
||
|
||
__all__ = ["start_server", "gen_fastapi_app"]
|