105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
|
# =======================================
|
|||
|
|
# =============== 通用工具 ===============
|
|||
|
|
# =======================================
|
|||
|
|
|
|||
|
|
import re
|
|||
|
|
import os
|
|||
|
|
from PySide2.QtGui import QClipboard
|
|||
|
|
from PySide2.QtCore import QFileInfo
|
|||
|
|
from PySide2.QtQml import QJSValue
|
|||
|
|
from urllib.parse import unquote # 路径解码
|
|||
|
|
|
|||
|
|
from umi_log import logger
|
|||
|
|
|
|||
|
|
Clipboard = QClipboard() # 剪贴板
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 传入文件名，检测是否含非法字符。没问题返回True
|
|||
|
|
def allowedFileName(fn):
|
|||
|
|
pattern = r'[\\/:*?"<>|]'
|
|||
|
|
if re.search(pattern, fn):
|
|||
|
|
return False # 转布尔值
|
|||
|
|
else:
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 复制文本到剪贴板
|
|||
|
|
def copyText(text):
|
|||
|
|
Clipboard.setText(text)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# QUrl列表 转 String列表
|
|||
|
|
def QUrl2String(urls):
|
|||
|
|
resList = []
|
|||
|
|
for url in urls:
|
|||
|
|
if url.isLocalFile():
|
|||
|
|
u = unquote(url.toLocalFile()) # 解码路径
|
|||
|
|
if QFileInfo(u).exists(): # 检查路径是否真的存在
|
|||
|
|
resList.append(u)
|
|||
|
|
return resList
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 初始化配置项字典数值，等同于 Configs.qml 的 function initConfigDict
|
|||
|
|
# 主要是为了补充type和default
|
|||
|
|
def initConfigDict(dic):
|
|||
|
|
toDict = {}
|
|||
|
|
|
|||
|
|
def handleConfigItem(config, key): # 处理一个配置项
|
|||
|
|
# 类型：指定type
|
|||
|
|
if not config["type"] == "":
|
|||
|
|
if config["type"] == "file": # 文件选择
|
|||
|
|
config["default"] = "" if not config["default"] is None else None
|
|||
|
|
elif config["type"] == "var" and config["default"] is None: # 任意类型
|
|||
|
|
config["default"] = ""
|
|||
|
|
# 类型：省略type
|
|||
|
|
else:
|
|||
|
|
if isinstance(config["default"], bool): # 布尔
|
|||
|
|
config["type"] = "boolean"
|
|||
|
|
elif "optionsList" in config: # 枚举
|
|||
|
|
config["type"] = "enum"
|
|||
|
|
if len(config["optionsList"]) == 0:
|
|||
|
|
logger.error(f"处理配置项异常：{key}枚举列表为空。")
|
|||
|
|
return
|
|||
|
|
if config["default"] is None:
|
|||
|
|
config["default"] = config["optionsList"][0][0]
|
|||
|
|
elif isinstance(config["default"], str): # 文本
|
|||
|
|
config["type"] = "text"
|
|||
|
|
elif isinstance(config["default"], (int, float)): # 数字
|
|||
|
|
config["type"] = "number"
|
|||
|
|
elif "btnsList" in config: # 按钮组
|
|||
|
|
config["type"] = "buttons"
|
|||
|
|
return
|
|||
|
|
else:
|
|||
|
|
logger.error(f"未知类型的配置项：{key}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
def handleConfigGroup(group, prefix=""): # 处理一个配置组
|
|||
|
|
for key in group:
|
|||
|
|
config = group[key]
|
|||
|
|
if not isinstance(config, dict):
|
|||
|
|
continue
|
|||
|
|
# 补充空白参数
|
|||
|
|
if "type" not in config: # 类型
|
|||
|
|
config["type"] = ""
|
|||
|
|
if "default" not in config: # 默认值
|
|||
|
|
config["default"] = None
|
|||
|
|
if "advanced" not in config: # 是否为高级选项
|
|||
|
|
config["advanced"] = False
|
|||
|
|
# 记录完整key
|
|||
|
|
fullKey = prefix + key
|
|||
|
|
if config["type"] == "group": # 若是配置项组，递归遍历
|
|||
|
|
handleConfigGroup(config, fullKey + ".") # 前缀加深一层
|
|||
|
|
else: # 若是配置项
|
|||
|
|
toDict[fullKey] = config
|
|||
|
|
handleConfigItem(config, fullKey)
|
|||
|
|
|
|||
|
|
handleConfigGroup(dic)
|
|||
|
|
return toDict
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 整理 argd 参数字典，将 float 恢复 int 类型，如 12.0 → 12
|
|||
|
|
def argdIntConvert(argd):
|
|||
|
|
for k, v in argd.items():
|
|||
|
|
if isinstance(v, float) and v.is_integer():
|
|||
|
|
argd[k] = int(v)
|