code/default/lib/noarch/env_info.py - master - githubmirror/XX-Net

code/default/lib/noarch/env_info.py
207 строк6 KB
micheal
5.16.4 improve performance
25 авг 2025, 02:10
25 авг 2025, 02:10758928d
import os
import platform
import sys
from pathlib import Path
import json

import utils

current_path = os.path.dirname(os.path.abspath(__file__))
default_path = os.path.abspath(os.path.join(current_path, os.path.pardir, os.path.pardir))
data_path = os.path.abspath(os.path.join(default_path, os.path.pardir, os.path.pardir, 'data'))


def win32_version():
    import ctypes
    class OSVERSIONINFOEXW(ctypes.Structure):
        _fields_ = [('dwOSVersionInfoSize', ctypes.c_ulong),
                    ('dwMajorVersion', ctypes.c_ulong),
                    ('dwMinorVersion', ctypes.c_ulong),
                    ('dwBuildNumber', ctypes.c_ulong),
                    ('dwPlatformId', ctypes.c_ulong),
                    ('szCSDVersion', ctypes.c_wchar * 128),
                    ('wServicePackMajor', ctypes.c_ushort),
                    ('wServicePackMinor', ctypes.c_ushort),
                    ('wSuiteMask', ctypes.c_ushort),
                    ('wProductType', ctypes.c_byte),
                    ('wReserved', ctypes.c_byte)]

    """
    Get's the OS major and minor versions.  Returns a tuple of
    (OS_MAJOR, OS_MINOR).
    """
    os_version = OSVERSIONINFOEXW()
    os_version.dwOSVersionInfoSize = ctypes.sizeof(os_version)
    retcode = ctypes.windll.Ntdll.RtlGetVersion(ctypes.byref(os_version))
    if retcode != 0:
        raise Exception("Failed to get OS version")

    return os_version.dwMajorVersion


def win32_version_string():
    import ctypes
    class OSVERSIONINFOEXW(ctypes.Structure):
        _fields_ = [('dwOSVersionInfoSize', ctypes.c_ulong),
                    ('dwMajorVersion', ctypes.c_ulong),
                    ('dwMinorVersion', ctypes.c_ulong),
                    ('dwBuildNumber', ctypes.c_ulong),
                    ('dwPlatformId', ctypes.c_ulong),
                    ('szCSDVersion', ctypes.c_wchar * 128),
                    ('wServicePackMajor', ctypes.c_ushort),
                    ('wServicePackMinor', ctypes.c_ushort),
                    ('wSuiteMask', ctypes.c_ushort),
                    ('wProductType', ctypes.c_byte),
                    ('wReserved', ctypes.c_byte)]

    """
    Get's the OS major and minor versions.  Returns a tuple of
    (OS_MAJOR, OS_MINOR).
    """
    os_version = OSVERSIONINFOEXW()
    os_version.dwOSVersionInfoSize = ctypes.sizeof(os_version)
    retcode = ctypes.windll.Ntdll.RtlGetVersion(ctypes.byref(os_version))
    if retcode != 0:
        raise Exception("Failed to get OS version")

    version_string = "Version:%d-%d; Build:%d; Platform:%d; CSD:%s; ServicePack:%d-%d; Suite:%d; ProductType:%d" % (
        os_version.dwMajorVersion, os_version.dwMinorVersion,
        os_version.dwBuildNumber,
        os_version.dwPlatformId,
        os_version.szCSDVersion,
        os_version.wServicePackMajor, os_version.wServicePackMinor,
        os_version.wSuiteMask,
        os_version.wReserved
    )

    return version_string


def linux_distribution():
    try:
        with open("/etc/os-release", "br") as fd:
            kvs = {}
            for line in fd.readlines():
                kv = line.split(b"=")
                if kv[0] == b"NAME":
                    v = kv[1].replace(b"\"", b"")
                    kvs[kv[0]] = v
        if b"PRETTY_NAME" in kvs:
            return kvs[b"PRETTY_NAME"]
        elif b"NAME" in kvs:
            return kvs[b"NAME"]
        else:
            return None
    except Exception as e:
        return None


def os_detail():
    if sys.platform == "win32":
        return win32_version_string()
    elif sys.platform.startswith("linux"):
        distribution = linux_distribution()
        if distribution is None:
            return "plat:%s release:%s ver:%s" % (platform.platform(), platform.release(), platform.version())
        else:
            return utils.to_str(distribution)
    elif sys.platform == "darwin":
        release, versioninfo, machine = platform.mac_ver()
        return "Release:%s; Version:%s Machine:%s" % (release, versioninfo, machine)
    else:
        return "None"


def get_system_date_path():
    home = Path.home()
    if sys.platform == "win32":
        return os.environ.get("APPDATA")
    elif sys.platform == "darwin":
        return os.path.join(home, "Library", "Application Support")
    else:
        return os.path.join(home, ".local", "share")

def get_user_data_path():
    """
    获取用户数据目录路径

    Returns:
        Path: 用户数据目录的 Path 对象

    Raises:
        OSError: 当无法确定用户目录时
    """
    try:
        home = Path.home()
    except RuntimeError:
        raise OSError("无法确定用户主目录")

    if sys.platform == "win32":
        # Windows: %APPDATA%
        appdata = os.environ.get("APPDATA")
        if appdata and Path(appdata).exists():
            return Path(appdata)
        # Fallback
        fallback = home / "AppData" / "Roaming"
        fallback.mkdir(parents=True, exist_ok=True)
        return fallback

    elif sys.platform == "darwin":
        # macOS: ~/Library/Application Support
        app_support = home / "Library" / "Application Support"
        app_support.mkdir(parents=True, exist_ok=True)
        return app_support

    else:
        # Linux/Unix: 遵循 XDG Base Directory Specification
        xdg_data_home = os.environ.get("XDG_DATA_HOME")
        if xdg_data_home:
            data_dir = Path(xdg_data_home)
        else:
            data_dir = home / ".local" / "share"

        data_dir.mkdir(parents=True, exist_ok=True)
        return data_dir


def is_in_system_application_path(p):
    if sys.platform == "win32":
        if "Program Files" in p:
            return True
    elif sys.platform == "darwin":
        if "Contents" in p:
            return True
    else:
        return False


def get_app_name():
    app_info_file = os.path.join(default_path, os.path.pardir, "app_info.json")
    try:
        with open(app_info_file, "r") as fd:
            dat = json.load(fd)
        return dat["app_name"]
    except Exception as e:
        print("get app name fail:%r", e)
    return "Dashboard"


def use_default_data_path():
    if os.path.isdir(data_path):
        return True

    try:
        os.mkdir(data_path)
        return True
    except Exception as e:
        return False


app_name = get_app_name()

# check and update data path
if not os.path.isdir(data_path) and \
        (is_in_system_application_path(data_path) or not use_default_data_path() or app_name not in ["XX-Net"]):
    data_path = os.path.join(get_user_data_path(), app_name)
    if not os.path.isdir(data_path):
        os.mkdir(data_path)