office-pc-plug/core/api_app.py

import logging
import os
import shutil
import socket

import pythoncom
import pywintypes
import win32com.client
from flask import Flask, jsonify, request, current_app
from flask_cors import CORS
from win10toast import ToastNotifier

from core.wps_handle import is_file_open_in_wps, bring_wps_window_to_front
from config import file_type_map
from tools.file_manager import get_file_md5
from tools.logger_handle import logger
from tools.oss_client import oss_handle

# ✅ 修复主机名非 UTF-8 解码错误
socket.getfqdn = lambda name=None: 'localhost'
toaster = ToastNotifier()

app = Flask(__name__)


class InterceptHandler(logging.Handler):
    def emit(self, record):
        # 将标准日志记录转为 loguru
        try:
            level = logger.level(record.levelname).name
        except ValueError:
            level = record.levelno
        logger.opt(depth=6, exception=record.exc_info).log(level, record.getMessage())


logging.basicConfig(handlers=[InterceptHandler()], level=logging.INFO)
for name in ("flask", "werkzeug", "watchdog"):
    logging.getLogger(name).handlers = [InterceptHandler()]
    logging.getLogger(name).propagate = False
CORS(app)


@logger.catch()
def open_file_by_wps(file_path):
    ext = file_path.split(".")[-1]
    shutil.rmtree(win32com.client.gencache.GetGeneratePath())
    if ext not in file_type_map:
        return jsonify({'code': 3001, 'msg': '不支持该类型的文件'})

    if not os.path.exists(file_path):
        return jsonify({'code': 3006, 'msg': '文件不存在'})

    pythoncom.CoInitialize()
    try:
        if ext in ['csv', 'xlsx', 'xls']:
            # wps = win32com.client.gencache.EnsureDispatch(file_type_map['xlsx'][0])
            wps = win32com.client.Dispatch(file_type_map[ext][0])

            wps.Visible = True
            getattr(wps, file_type_map[ext][1]).Open(os.path.abspath(file_path))
        else:
            wps = win32com.client.Dispatch(file_type_map[ext][0])
            wps.Visible = True
            getattr(wps, file_type_map[ext][1]).open(os.path.abspath(file_path))
    except pywintypes.com_error as e:
        logger.exception(e)

    bring_wps_window_to_front(file_path)
    pythoncom.CoUninitialize()

    return jsonify({'code': 1000, 'msg': '操作完成'})


@logger.catch()
@app.route('/download_and_open_file', methods=['POST'])
def download_and_open_with_wps():
    '''
    1000: 操作完成
    2001: 检测到文档正在被编辑，无法下载
    2002: 检测到远程文件已经更新本地文件也发生改动，终止操作
    3001: 不支持该类型的文件
    3002: 下载文件失败
    :return:
    '''
    file_id = request.get_json()['file_id']
    # 判断是否为exe模式执行

    file_info = app.config['serve_client'].get_file_info(file_id)
    if not file_info:
        return jsonify({'code': 3006, 'msg': '文件不存在'})

    local_file = os.path.join(current_app.config['work_path'], file_info['filePath'].replace('/', '_'))

    # 判断本地是否存在已下载的同名文件
    if not os.path.exists(local_file) or not os.path.exists(local_file + '.metadata.json'):
        app.config['serve_client'].download_file(file_info,
                                                 current_app.config['work_path'])

        return open_file_by_wps(local_file)

    # 判断本地文件是否被打开
    if is_file_open_in_wps(local_file):
        bring_wps_window_to_front(local_file)
        return jsonify({'code': 1000, 'msg': '检测到文件已被打开'})

    # 判断本地文件和线上文件是否一致
    local_file_metadata = oss_handle.load_metadata(local_file)
    if local_file_metadata['update_time'] == file_info['updateTime']:
        return open_file_by_wps(local_file)

    if local_file_metadata['md5'] == app.config['serve_client'].get_file_md5(local_file):
        app.config['serve_client'].download_file(file_info, current_app.config['work_path'])

        return open_file_by_wps(local_file)

    return jsonify({'code': 2002, 'msg': '检测到远程文件已经更新本地文件也发生改动，终止操作'})


@logger.catch()
@app.route('/upload_local_file', methods=['POST'])
def upload_local_file():
    file_name = request.get_json()['file_name']
    if is_file_open_in_wps(os.path.join(current_app.config['work_path'], file_name)):
        return jsonify({'code': 3004, 'msg': '操作失败，文件已被打开无法操作'})
    metadata = app.config['serve_client'].load_metadata(os.path.join(current_app.config['work_path'], file_name))
    code = app.config['serve_client'].upload_file(os.path.join(current_app.config['work_path'], file_name))
    if not code:
        return jsonify({'code': 3003, 'msg': '操作失败，上传文件失败'})

    try:
        os.remove(os.path.join(current_app.config['work_path'], file_name))
        os.remove(os.path.join(current_app.config['work_path'], file_name + '.metadata'))

        file_info = app.config['serve_client'].get_file_info(metadata['file_id'])
        if not file_info:
            return jsonify({'code': 3006, 'msg': '文件不存在'})

        app.config['serve_client'].download_file(file_info, current_app.config['work_path'])
    except Exception as e:
        logger.exception(e)
        return jsonify({'code': 1000, 'msg': '文件上传成功，本地文件清除失败'})

    return jsonify({'code': 1000, 'msg': '操作成功'})


@logger.catch()
@app.route('/update_local_file', methods=['POST'])
def update_local_file():
    file_name = request.get_json()['file_name']
    if is_file_open_in_wps(os.path.join(current_app.config['work_path'], file_name)):
        return jsonify({'code': 3004, 'msg': '操作失败，文件已被打开无法操作'})

    try:
        metadata = oss_handle.load_metadata(os.path.join(current_app.config['work_path'], file_name))

        os.remove(os.path.join(current_app.config['work_path'], file_name))
        os.remove(os.path.join(current_app.config['work_path'], file_name + '.metadata'))
    except Exception as e:
        logger.exception(e)
        return jsonify({'code': 3005, 'msg': '操作失败，无权限操作本地文件'})

    try:
        file_info = app.config['serve_client'].get_file_info(metadata['file_id'])
        if not app.config['serve_client'].download_file(file_info, current_app.config['work_path']):
            return jsonify({'code': 3005, 'msg': '文件下载失败'})
    except Exception as e:
        logger.exception(e)
        return jsonify({'code': 3005, 'msg': '文件下载失败'})

    return jsonify({'code': 1000, 'msg': '操作成功'})


@logger.catch()
@app.route('/file_state_list', methods=['GET'])
def file_state_list():
    '''
    state : 0 正常， 1: 云端有更新 2: 本地有变动且无冲突可以上传 3: 本地文件和云端文件有冲突
    :return:
    '''
    file_info_list = []
    for file_name in os.listdir(current_app.config['work_path']):
        suffer = file_name.split('.')[-1]
        if suffer not in file_type_map:
            continue

        if not os.path.exists(os.path.join(current_app.config['work_path'], file_name + '.metadata')):
            continue

        metadata = oss_handle.load_metadata(os.path.join(current_app.config['work_path'], file_name))
        file_info = {
            'file_name': file_name,
            'show_name': metadata['file_name'],
            'cloud_update_time': metadata['update_time'],
            'source_md5': metadata['md5'],
            'state': 0
        }
        cloud_file_info = current_app.config['serve_client'].get_file_info(metadata['file_id'])
        if cloud_file_info == 2:
            file_info['state'] = 4
            file_info_list.append(file_info)
            continue

        if not cloud_file_info:
            file_info_list.append(file_info)
            continue

        if cloud_file_info['updateTime'] != metadata['update_time']:
            file_info['state'] = 1

        try:
            file_md5 = get_file_md5(os.path.join(current_app.config['work_path'], file_name))
            if file_md5 != metadata['md5']:
                if file_info['state'] == 1:
                    file_info['state'] = 3
                else:
                    file_info['state'] = 2
        except Exception as e:
            logger.exception(e)

        file_info_list.append(file_info)

    return jsonify({'code': 1000, 'msg': '操作成功', 'data': file_info_list})


@logger.catch()
@app.route('/remove_file', methods=['POST'])
def remove_file():
    file_name = request.get_json()['file_name']
    file_path = os.path.join(current_app.config['work_path'], file_name)
    if not os.path.exists(file_path):
        return jsonify({'code': '3006', 'msg': '文件不存在'})

    try:
        os.remove(file_path)
        os.remove(f'{file_path}.metadata')
    except Exception as e:
        logger.exception(e)
        return jsonify({'code': 3005, 'msg': '操作失败，无权限操作本地文件'})

    return jsonify({'code': 1000, 'msg': '操作成功'})


@logger.catch()
@app.route('/open_file', methods=['POST'])
def open_file():
    file_name = request.get_json()['file_name']
    file_path = os.path.join(current_app.config['work_path'], file_name)

    if not os.path.exists(file_path):
        return jsonify({'code': '3006', 'msg': '文件不存在'})

    if is_file_open_in_wps(file_path):
        bring_wps_window_to_front(file_path)
        return jsonify({'code': 1000, 'msg': '检测到文件已被打开'})

    return open_file_by_wps(file_path)


def start_flask(serve_client, work_path):
    app.config['serve_client'] = serve_client
    app.config['work_path'] = work_path
    # app.logger.handlers.clear()
    # app.logger.addHandler(InterceptHandler())

    app.run(host='0.0.0.0', port=5855)