utils_env.py

# -*- coding: utf-8 -*-

import os
import platform

# 缓存全局的环境
ENV = ''


def get_env_str() -> str:
    """
    尝试获取当前系统的环境，返回字符。

    :return: Windows / Linux / Mac / github / v2p / ql / 空
    """
    global ENV
    if ENV:
        return ENV

    v2p_file = '/usr/local/app/script/Lists/task.list'
    ql_file = '/ql/config/env.sh'

    print('尝试检查运行环境...')
    if os.getenv('GITHUB_ACTIONS'):
        print('成功，当前环境为: github action 面板。')
        env = 'github'
    elif os.path.exists(v2p_file):
        print('成功，当前环境为: elecV2P 面板。')
        env = 'v2p'
    elif os.path.exists(ql_file):
        print('成功，当前环境为: 青龙面板。')
        env = 'ql'

    # 面板判断优先于系统判断
    elif (e := platform.system()) == "Windows" or "Linux" or "Mac":
        print(f'成功，当前环境为 {e}。')
        env = e
    else:
        print('失败！请检查环境。')
        env = ''

    ENV = env
    print('环境检查结束。\n')
    return env


def get_env_int() -> int:
    """
    尝试获取当前系统的环境，返回数字。

    :return: 空: -1 / Windows: 0 / Linux: 1 / Mac: 2 / github: 3 / v2p: 4 / ql: 5
    """
    env = get_env_str()
    if env == 'Windows':
        return 0
    if env == 'Linux':
        return 1
    if env == 'Mac':
        return 2
    if env == 'github':
        return 3
    if env == 'v2p':
        return 4
    if env == 'ql':
        return 5

    return -1


def get_file_path(file_name: str) -> str:
    """
    根据文件名返回对应环境中位置。

    :param file_name: 文件名，不含任何 "/" (即目录)
    :return: 如果有面板，返回面板默认配置文件夹，否则返回当前目录下文件。 <br/> 如果路径下文件不存在，返回空串。
    """
    env_i = get_env_int()
    print(f'配置文件 ({file_name}) 检查开始...')
    paths = [
        file_name,
        file_name,
        file_name,
        file_name,
        f'/usr/local/app/script/Lists/{file_name}',
        f'/ql/config/{file_name}',
    ]

    if env_i < 0:
        print('无法判断环境，选择当前目录为配置文件夹目录。')
        env_i = 0
    if not os.path.exists(paths[env_i]):
        print(f'未找到配置文件（不一定是错误），路径为: {paths[env_i]}。')
        print('配置文件检查结束。\n')
        return ''
    print(f'在 {paths[env_i]} 发现配置文件。')
    print('配置文件检查结束。\n')
    return paths[env_i]