office-pc-plug/office_helper.py

import functools
import json
import os.path
import sys
import threading
import time

from api.ws_app import start_ws_server
from tools.check import enable_wps_login_force

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

from api.api_app import start_flask
from modify_folder.monitor_file import start_watchdog
from tools.logger_handle import logger
from tools.serve_client import server_client
from config import *


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, storage_path):
    enable_wps_login_force()
    # 启动 Flask 服务
    threading.Thread(target=start_flask, args=[serve_client, storage_path], daemon=True).start()
    # 启动文件监控
    # threading.Thread(target=start_watchdog, args=[serve_client, storage_path], daemon=True).start()

    # threading.Thread(target=start_ws_server, daemon=True).start()
    logger.info('server running')

    while True:
        time.sleep(10)


if __name__ == "__main__":
    if not (args.get('username') and args.get('password')):
        logger.error('The config is missing critical information')
        sys.exit(0)
    storage_path = os.path.join(args['work_path'], args['username'])
    if not os.path.exists(storage_path):
        os.makedirs(storage_path)
    logger.info(f'VERSION: {VERSION}')

    start_all_services(server_client, storage_path)