nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-04-05 03:38:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(string): 优化 版本比较 方法

Browse Source
This commit is contained in:
Aqr-K
2025-01-05 16:22:28 +08:00
parent 3078c076dc
commit 9fab50edb0
3 changed files with 119 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/core/plugin.py
View File

@@ -682,7 +682,7 @@ class PluginManager(metaclass=Singleton):
# 相同 ID 的插件保留版本号最大的版本
max_versions = {}
for p in all_plugins:
if p.id not in max_versions or StringUtils.compare_version(p.plugin_version, max_versions[p.id]) > 0:
if p.id not in max_versions or StringUtils.compare_version(p.plugin_version, max_versions[p.id], ">"):
max_versions[p.id] = p.plugin_version
result = [p for p in all_plugins if p.plugin_version == max_versions[p.id]]
logger.info(f"共获取到 {len(result)} 个线上插件")
@@ -823,7 +823,7 @@ class PluginManager(metaclass=Singleton):
plugin.has_update = False
if plugin_static:
installed_version = getattr(plugin_static, "plugin_version")
if StringUtils.compare_version(installed_version, plugin_info.get("version")) < 0:
if StringUtils.compare_version(installed_version, plugin_info.get("version"), "<"):
# 需要更新
plugin.has_update = True
# 运行状态

2
app/helper/resource.py
View File

@@ -70,7 +70,7 @@ class ResourceHelper(metaclass=Singleton):
local_version = self.siteshelper.indexer_version
else:
continue
if StringUtils.compare_version(version, local_version) > 0:
if StringUtils.compare_version(version, local_version, ">"):
logger.info(f"{rname} 资源包有更新，最新版本：v{version}")
else:
continue

135
app/utils/string.py
View File

@@ -740,27 +740,124 @@ class StringUtils:
return ''.join(common_prefix)
@staticmethod
def compare_version(v1: str, v2: str) -> int:
def compare_version(v1: str, v2: str, compare_type, verbose: bool = False) \
-> Tuple[Optional[bool], str | Exception] | Optional[bool]:
"""
比较两个版本号的大小，v1 > v2时返回1，v1 < v2时返回-1，v1 = v2时返回0
比较两个版本号的大小
:param v1: 比对的来源版本号
:param v2: 比对的目标版本号
:param verbose: 是否输出比对结果的时候输出详细消息，默认 False 不输出
:param compare_type: 识别模式。支持直接使用符号进行比对
'ge' or '>=' ：来源 >= 目标
'le' or '<=' ：来源 <= 目标
'eq' or '==' ：来源 == 目标
'gt' or '>' ：来源 > 目标
'lt' or '<' ：来源 < 目标
:return
"""
if not v1 or not v2:
return 0
v1 = v1.replace('v', '')
v2 = v2.replace('v', '')
v1 = [int(x) for x in v1.split('.')]
v2 = [int(x) for x in v2.split('.')]
for i in range(min(len(v1), len(v2))):
if v1[i] > v2[i]:
return 1
elif v1[i] < v2[i]:
return -1
if len(v1) > len(v2):
return 1
elif len(v1) < len(v2):
return -1
else:
return 0
def preprocess_version(version: str) -> list:
"""
预处理版本号，去除首尾空字符串与换行符，去除开头大小写v，并拆分版本号
"""
return re.split(r'[.-]', version.strip().lstrip('vV'))
def conversion_version(version_list) -> list:
"""
英文字符转换为数字
stable = -1，rc = -2，beta = -3，alpha = -4
其余不符合的，都为-5
"""
# 内置版本号转换字典
version_map = {"stable": -1, "rc": -2, "beta": -3, "alpha": -4}
# 不符合的版本号
other = -5
result = []
for item in version_list:
if item.isdigit():
result.append(int(item))
else:
value = version_map.get(item, other)
result.append(value)
return result
try:
if not v1 or not v2:
raise ValueError("要比较的版本号不全")
if not compare_type:
raise ValueError("缺少比对模式，无法比对")
if compare_type not in {"ge", "gt", "le", "lt", "eq", "==", ">=", ">", "<=", "<"}:
raise ValueError(f"设置的版本比对模式 {compare_type} 不是有效的模式！")
# 拆分获取版本号各个分段值做成列表
input_ver_list = conversion_version(preprocess_version(version=v1))
targe_ver_list = conversion_version(preprocess_version(version=v2))
# 补全版本号位置，保持长度一致
max_length = max(len(input_ver_list), len(targe_ver_list))
input_ver_list += [0] * (max_length - len(input_ver_list))
targe_ver_list += [0] * (max_length - len(targe_ver_list))
ver_comparison, ver_comparison_err = None, None
for i, t in zip(input_ver_list, targe_ver_list):
# 来源==目标
if compare_type in {"eq", "=="}:
if i != t:
ver_comparison, ver_comparison_err = None, "不等于"
break
else:
ver_comparison, ver_comparison_err = "等于", None
# 来源>=目标
elif compare_type in {"ge", ">="}:
if i > t:
ver_comparison, ver_comparison_err = "大于", None
break
elif i < t:
ver_comparison, ver_comparison_err = None, "小于"
break
else:
ver_comparison, ver_comparison_err = "等于", None
# 来源>目标
elif compare_type in {"gt", ">"}:
if i > t:
ver_comparison, ver_comparison_err = "大于", None
break
elif i < t:
ver_comparison, ver_comparison_err = None, "小于"
break
else:
ver_comparison, ver_comparison_err = None, "等于"
# 来源<=目标
elif compare_type in {"le", "<="}:
if i > t:
ver_comparison, ver_comparison_err = None, "大于"
break
elif i < t:
ver_comparison, ver_comparison_err = "小于", None
break
else:
ver_comparison, ver_comparison_err = "等于", None
# 来源<目标
elif compare_type in {"lt", "<"}:
if i > t:
ver_comparison, ver_comparison_err = None, "大于"
break
elif i < t:
ver_comparison, ver_comparison_err = "小于", None
break
else:
ver_comparison, ver_comparison_err = None, "等于"
msg = f"版本号 {v1} {ver_comparison if ver_comparison else ver_comparison_err} 目标版本号 {v2} ！"
return (True if ver_comparison else False, msg) if verbose else True if ver_comparison else False
except Exception as e:
return (None, e) if verbose else None
@staticmethod
def diff_time_str(time_str: str):