office-pc-plug/tools/check.py

import json
import os
import ipaddress
from urllib.parse import urlparse

import pythoncom
import win32com

from config import base_path, file_type_map
from tools.oss_client import MinioClient
from tools.logger_handle import logger

REPLY = {'html': None, 'cookies': None, 'token': None}


def extract_domain_or_ip(url):
    """提取网页中的域名或 IP+端口"""
    try:
        parsed_url = urlparse(url)
        host = parsed_url.hostname  # 提取主机部分（域名或 IP）
        port = parsed_url.port  # 提取端口

        result = host
        # 如果是 IP 地址，则保留端口
        if is_ip_address(host):
            result = f"u{host}:{port}" if port else 'u' + host

        return result.replace('.', '_').replace(':', '_')  # 仅返回域名
    except Exception as e:
        print(f"解析错误: {e}")
        return None


def is_ip_address(host):
    """判断是否是 IP 地址"""
    try:
        ipaddress.ip_address(host)
        return True
    except ValueError:
        return False


def detect_oss_file_changes(oss_client, file_info):
    cloud_file_md5 = oss_client.get_cloud_file_md5(file_info['bucket_name'], file_info['file_path'])
    with open(os.path.join(base_path, file_info['file_name'] + '.metadata'), 'r', encoding='utf-8') as f:
        metadata = json.load(f)

    return cloud_file_md5 == metadata['md5']


def detect_local_file_changes(file_path):
    with open(os.path.join(base_path, file_path + '.metadata'), 'r', encoding='utf-8') as f:
        metadata = json.load(f)

    current_file_md5 = MinioClient.get_file_md5(file_path)
    return metadata['md5'] == current_file_md5


def is_file_open_in_wps(file_path: str) -> bool:
    abs_path = os.path.abspath(file_path).lower()
    suffix = file_path.split('.')[-1]
    try:
        pythoncom.CoInitialize()
        app = win32com.client.GetActiveObject(file_type_map[suffix][0])

        for wb in getattr(app, file_type_map[suffix][1]):
            if wb.FullName.lower() == abs_path:
                return True
    except Exception:
        return False
    finally:
        pythoncom.CoUninitialize()
    return False


def enable_wps_login_force():
    import winreg

    key_path = r"Software\kingsoft\Office\6.0\plugins\officespace\flogin"
    try:
        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, key_path, 0, winreg.KEY_SET_VALUE)
        winreg.SetValueEx(key, "enableForceLoginForHasInstallDevice", 0, winreg.REG_SZ, "false")
        winreg.CloseKey(key)
        logger.info("WPS注册表:enableForceLoginForHasInstallDevice=false 修改成功")
    except Exception as e:
        logger.info(f"WPS注册表修改失败: {e}")