mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-31 04:14:38 +08:00
99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
"""
|
|
按解释器 ABI 选择主程序运行依赖 profile。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import tomllib
|
|
from collections.abc import Iterable
|
|
from pathlib import Path
|
|
|
|
from packaging.requirements import InvalidRequirement, Requirement
|
|
from packaging.utils import canonicalize_name
|
|
|
|
from app.foundation.environment import is_free_threaded_runtime
|
|
|
|
RUNTIME_STANDARD_GROUP = "runtime-standard"
|
|
RUNTIME_FREE_THREADED_GROUP = "runtime-free-threaded"
|
|
|
|
|
|
def runtime_dependency_group() -> str:
|
|
"""
|
|
返回当前解释器必须使用的互斥运行依赖组。
|
|
"""
|
|
if is_free_threaded_runtime():
|
|
return RUNTIME_FREE_THREADED_GROUP
|
|
return RUNTIME_STANDARD_GROUP
|
|
|
|
|
|
def runtime_sync_arguments() -> tuple[str, ...]:
|
|
"""
|
|
返回 uv sync 选择当前运行依赖组所需的稳定参数。
|
|
"""
|
|
return "--no-default-groups", "--group", runtime_dependency_group()
|
|
|
|
|
|
def iter_runtime_requirement_strings(project_file: Path) -> Iterable[str]:
|
|
"""
|
|
读取主项目依赖及当前运行 profile 的根依赖声明。
|
|
"""
|
|
with project_file.open("rb") as file:
|
|
document = tomllib.load(file)
|
|
|
|
project = document.get("project") or {}
|
|
for requirement in project.get("dependencies") or ():
|
|
if isinstance(requirement, str):
|
|
yield requirement
|
|
|
|
groups = document.get("dependency-groups") or {}
|
|
for requirement in groups.get(runtime_dependency_group()) or ():
|
|
if isinstance(requirement, str):
|
|
yield requirement
|
|
|
|
|
|
def iter_runtime_profile_requirement_strings(project_file: Path) -> Iterable[str]:
|
|
"""
|
|
读取当前解释器 profile 的根依赖声明。
|
|
"""
|
|
with project_file.open("rb") as file:
|
|
document = tomllib.load(file)
|
|
|
|
groups = document.get("dependency-groups") or {}
|
|
for requirement in groups.get(runtime_dependency_group()) or ():
|
|
if isinstance(requirement, str):
|
|
yield requirement
|
|
|
|
|
|
def runtime_excluded_dependency_pairs(project_file: Path) -> set[tuple[str, str]]:
|
|
"""
|
|
返回项目明确排除的“依赖包 -> 传递依赖”规范化名称对。
|
|
"""
|
|
with project_file.open("rb") as file:
|
|
document = tomllib.load(file)
|
|
|
|
exclusions = (
|
|
((document.get("tool") or {}).get("uv") or {}).get("exclude-dependencies")
|
|
or ()
|
|
)
|
|
pairs: set[tuple[str, str]] = set()
|
|
for exclusion in exclusions:
|
|
if not isinstance(exclusion, dict):
|
|
continue
|
|
package = exclusion.get("package") or {}
|
|
package_name = package.get("name") if isinstance(package, dict) else None
|
|
if not isinstance(package_name, str) or not package_name.strip():
|
|
continue
|
|
for dependency in exclusion.get("dependencies") or ():
|
|
if not isinstance(dependency, str):
|
|
continue
|
|
try:
|
|
dependency_name = Requirement(dependency).name
|
|
except InvalidRequirement:
|
|
continue
|
|
pairs.add((canonicalize_name(package_name), canonicalize_name(dependency_name)))
|
|
return pairs
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(runtime_dependency_group())
|