|
|
import subprocess
|
|
|
|
|
|
from flask import Flask, render_template, request, send_file, session, redirect, url_for
|
|
|
from datetime import datetime
|
|
|
from common.logger import tool_logger, error_logger, system_logger
|
|
|
|
|
|
import os, uuid
|
|
|
|
|
|
# =========================
|
|
|
# 工具模块导入
|
|
|
# =========================
|
|
|
from tools.file_convert import file_convert_page, file_convert_download
|
|
|
from tools.temp_upload import (
|
|
|
temp_upload_page,
|
|
|
download_temp_file
|
|
|
)
|
|
|
|
|
|
from tools.excel_merge import (
|
|
|
read_file,
|
|
|
merge_df,
|
|
|
export_excel,
|
|
|
build_preview
|
|
|
)
|
|
|
|
|
|
from tools.smart_data_v2 import (
|
|
|
smart_data_page_v2,
|
|
|
smart_download_v2
|
|
|
)
|
|
|
|
|
|
from tools.weekly_permission import weekly_permission_page
|
|
|
|
|
|
# ⭐ 新增：邮件通知工具
|
|
|
from tools.mail_notify import mail_notify_page, mail_notify_progress
|
|
|
|
|
|
# ⭐ 新增：Base64 编解码工具
|
|
|
from tools.base64_codec import page as base64_page
|
|
|
|
|
|
# ⭐ 新增：JSON 格式化工具
|
|
|
from tools.json_tool import page as json_page
|
|
|
|
|
|
# ⭐ 新增：URL 编解码工具
|
|
|
from tools.url_codec import page as url_page
|
|
|
|
|
|
# ⭐ 新增：图片压缩工具
|
|
|
from tools.image_compress import page as image_compress_page
|
|
|
|
|
|
# ⭐ 新增：文本差异对比工具
|
|
|
from tools.text_diff import page as text_diff_page
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# Flask app
|
|
|
# =========================
|
|
|
app = Flask(__name__)
|
|
|
app.secret_key = "excel_merge_secret_key"
|
|
|
|
|
|
UPLOAD_FOLDER = "uploads"
|
|
|
OUTPUT_FOLDER = "output"
|
|
|
|
|
|
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
|
|
|
os.makedirs(OUTPUT_FOLDER, exist_ok=True)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# reset
|
|
|
# =========================
|
|
|
@app.route("/reset")
|
|
|
def reset():
|
|
|
session.clear()
|
|
|
return redirect(url_for("index"))
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# 首页
|
|
|
# =========================
|
|
|
@app.route("/")
|
|
|
def index():
|
|
|
return render_template("index.html")
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# Excel merge（原逻辑不变）
|
|
|
# =========================
|
|
|
def handle_excel_merge():
|
|
|
preview_data = None
|
|
|
download_file = None
|
|
|
message = None
|
|
|
|
|
|
if request.method == "POST":
|
|
|
try:
|
|
|
file1 = request.files.get("file1")
|
|
|
file2 = request.files.get("file2")
|
|
|
|
|
|
if file1 and file1.filename:
|
|
|
p1 = os.path.join(UPLOAD_FOLDER, f"{uuid.uuid4()}_{file1.filename}")
|
|
|
file1.save(p1)
|
|
|
session["file1_path"] = p1
|
|
|
session["file1_name"] = file1.filename
|
|
|
|
|
|
if file2 and file2.filename:
|
|
|
p2 = os.path.join(UPLOAD_FOLDER, f"{uuid.uuid4()}_{file2.filename}")
|
|
|
file2.save(p2)
|
|
|
session["file2_path"] = p2
|
|
|
session["file2_name"] = file2.filename
|
|
|
|
|
|
if not session.get("file1_path") or not session.get("file2_path"):
|
|
|
raise Exception("请上传两个文件")
|
|
|
|
|
|
key1 = request.form.get("key1", "").strip()
|
|
|
key2 = request.form.get("key2", "").strip()
|
|
|
join_type = request.form.get("join_type", "left")
|
|
|
|
|
|
session["key1"] = key1
|
|
|
session["key2"] = key2
|
|
|
session["join_type"] = join_type
|
|
|
|
|
|
df1 = read_file(session["file1_path"])
|
|
|
df2 = read_file(session["file2_path"])
|
|
|
|
|
|
merged_df = merge_df(df1, df2, key1, key2, join_type)
|
|
|
|
|
|
out_name = f"merged_{uuid.uuid4().hex}.xlsx"
|
|
|
out_path = os.path.join(OUTPUT_FOLDER, out_name)
|
|
|
|
|
|
export_excel(merged_df, out_path)
|
|
|
|
|
|
preview_data = build_preview(merged_df)
|
|
|
download_file = out_name
|
|
|
|
|
|
tool_logger.info(
|
|
|
f"[ExcelMerge] file1={session.get('file1_name')} "
|
|
|
f"file2={session.get('file2_name')} rows={len(merged_df)}"
|
|
|
)
|
|
|
|
|
|
message = (
|
|
|
f"整合成功，共 {len(merged_df)} 条记录 ｜ "
|
|
|
f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
|
|
|
)
|
|
|
|
|
|
except Exception as e:
|
|
|
error_logger.exception("[ExcelMergeError]")
|
|
|
message = f"错误：{e}"
|
|
|
|
|
|
return render_template(
|
|
|
"excel.html",
|
|
|
preview_data=preview_data,
|
|
|
download_file=download_file,
|
|
|
message=message,
|
|
|
file1_name=session.get("file1_name", ""),
|
|
|
file2_name=session.get("file2_name", ""),
|
|
|
key1=session.get("key1", ""),
|
|
|
key2=session.get("key2", ""),
|
|
|
join_type=session.get("join_type", "left"),
|
|
|
)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# download
|
|
|
# =========================
|
|
|
@app.route("/download/<filename>")
|
|
|
def download(filename):
|
|
|
return send_file(
|
|
|
os.path.join(OUTPUT_FOLDER, filename),
|
|
|
as_attachment=True
|
|
|
)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# TOOL HUB（统一入口）
|
|
|
# =========================
|
|
|
TOOL_ROUTES = {}
|
|
|
|
|
|
|
|
|
def tool_dispatch(name):
|
|
|
handler = TOOL_ROUTES.get(name)
|
|
|
if handler:
|
|
|
return handler()
|
|
|
|
|
|
return render_template(f"{name}.html")
|
|
|
|
|
|
|
|
|
@app.route("/tool/mail_notify/progress/<task_id>")
|
|
|
def mail_progress(task_id):
|
|
|
return mail_notify_progress(task_id)
|
|
|
|
|
|
|
|
|
@app.route("/tool/<name>", methods=["GET", "POST"])
|
|
|
def tool(name):
|
|
|
return tool_dispatch(name)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# 工具注册（核心扩展点）
|
|
|
# =========================
|
|
|
TOOL_ROUTES.update({
|
|
|
|
|
|
"excel": handle_excel_merge,
|
|
|
|
|
|
"smart_data_v2": smart_data_page_v2,
|
|
|
|
|
|
"file_convert": file_convert_page,
|
|
|
|
|
|
"weekly_permission": weekly_permission_page,
|
|
|
|
|
|
# ⭐ 新增邮件通知工具
|
|
|
"mail_notify": mail_notify_page,
|
|
|
|
|
|
# ⭐ 新增 Base64 编解码工具
|
|
|
"base64": base64_page,
|
|
|
|
|
|
# ⭐ 新增 JSON 格式化工具
|
|
|
"json": json_page,
|
|
|
|
|
|
# ⭐ 新增 URL 编解码工具
|
|
|
"url": url_page,
|
|
|
|
|
|
# ⭐ 新增图片压缩工具
|
|
|
"image_compress": image_compress_page,
|
|
|
|
|
|
# ⭐ 新增文本差异对比工具
|
|
|
"text_diff": text_diff_page,
|
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# temp upload
|
|
|
# =========================
|
|
|
@app.route("/tool/temp_upload", methods=["GET", "POST"])
|
|
|
def temp_upload():
|
|
|
return temp_upload_page()
|
|
|
|
|
|
|
|
|
@app.route("/temp/download/<filename>")
|
|
|
def temp_download(filename):
|
|
|
return download_temp_file(filename)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# smart data download
|
|
|
# =========================
|
|
|
@app.route("/tool/smart_data_v2/download/<filename>")
|
|
|
def smart_data_v2_download(filename):
|
|
|
return smart_download_v2(filename)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# file convert download
|
|
|
# =========================
|
|
|
@app.route("/download/file_convert/<filename>")
|
|
|
def file_convert_download_route(filename):
|
|
|
return file_convert_download(filename)
|
|
|
|
|
|
|
|
|
# =========================
|
|
|
# image compress download
|
|
|
# =========================
|
|
|
@app.route("/tool/image_compress/download/<filename>")
|
|
|
def image_compress_download(filename):
|
|
|
return send_file(os.path.join(OUTPUT_FOLDER, filename), as_attachment=True)
|
|
|
|
|
|
|
|
|
cleanup_process = None
|
|
|
def start_cleanup_task():
|
|
|
"""
|
|
|
启动定时清理任务
|
|
|
"""
|
|
|
global cleanup_process
|
|
|
cleanup_path = os.path.join(
|
|
|
os.path.dirname(__file__),
|
|
|
"cleanup_upload.py"
|
|
|
)
|
|
|
|
|
|
cleanup_process = subprocess.Popen(
|
|
|
["python3", cleanup_path]
|
|
|
)
|
|
|
|
|
|
system_logger.info(
|
|
|
f"[cleanup] started pid="
|
|
|
f"{cleanup_process.pid}"
|
|
|
)
|
|
|
print(
|
|
|
f"[cleanup] started pid="
|
|
|
f"{cleanup_process.pid}"
|
|
|
)
|
|
|
# =========================
|
|
|
# main
|
|
|
# =========================
|
|
|
if __name__ == "__main__":
|
|
|
|
|
|
# 防止 Flask debug 启动两次
|
|
|
if os.environ.get("WERKZEUG_RUN_MAIN") == "true":
|
|
|
start_cleanup_task()
|
|
|
|
|
|
app.run(host="0.0.0.0", port=8209, debug=True)
|
