office-pc-plug/office_helper.py

import functools
import json
import os.path
import sys

import threading

from core.api_app import start_flask
from core.monitor_file import start_watchdog
from config import base_path, dl_code
from tools.serve_client import ServerClient
from tools.logger_handle import logger

# 初始化 Flask 应用


# 初始化桌面通知工具


# 存储文档内容和推荐内容
current_text = ""
recommendation = "这是默认推荐内容"


def ApplicationInstance(func):
    @functools.wraps(func)
    def fun(*args, **kwargs):
        import socket
        try:
            global s
            s = socket.socket()
            host = socket.gethostname()
            s.bind((host, 60123))
        except:
            logger.error('不能重复打开已存在的程序')
            return None
        return func(*args, **kwargs)

    return fun


# 启动所有后台服务
@ApplicationInstance
def start_all_services(serve_client, work_path):
    # 启动 Flask 服务
    threading.Thread(target=start_flask, args=[serve_client, work_path], daemon=True).start()
    # 启动文件监控
    threading.Thread(target=start_watchdog, args=[serve_client, work_path], daemon=True).start()
    # 启动文档内容检查
    # threading.Thread(target=check_for_changes, daemon=True).start()
    logger.info('server running')
    while True:
        a = 0


if __name__ == "__main__":
    def set_working_directory():
        if getattr(sys, 'frozen', False):
            # 打包后的 exe
            os.chdir(os.path.dirname(sys.executable))
        else:
            # 普通 Python 脚本
            os.chdir(os.path.dirname(os.path.abspath(__file__)))


    with open(f'{dl_code}:\\ProgramData\\OfficeAssistant\\config.json', 'r', encoding='utf-8') as f:
        args = json.load(f)

    client = ServerClient(args['server'], args['username'], args['password'])
    if not os.path.exists(os.path.join(args['worker_path'], args['username'])):
        os.makedirs(os.path.join(args['worker_path'], args['username']))

    start_all_services(client, os.path.join(args['worker_path'], args['username']))