mirror of
https://github.com/Bing-su/adetailer.git
synced 2026-01-26 19:29:54 +00:00
89 lines
2.0 KiB
Python
89 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from importlib.metadata import version # python >= 3.8
|
|
|
|
from packaging.version import parse
|
|
|
|
import_name = {"py-cpuinfo": "cpuinfo", "protobuf": "google.protobuf"}
|
|
|
|
|
|
def is_installed(
|
|
package: str, min_version: str | None = None, max_version: str | None = None
|
|
):
|
|
name = import_name.get(package, package)
|
|
try:
|
|
spec = importlib.util.find_spec(name)
|
|
except ModuleNotFoundError:
|
|
return False
|
|
|
|
if spec is None:
|
|
return False
|
|
|
|
if not min_version and not max_version:
|
|
return True
|
|
|
|
if not min_version:
|
|
min_version = "0.0.0"
|
|
if not max_version:
|
|
max_version = "99999999.99999999.99999999"
|
|
|
|
try:
|
|
pkg_version = version(package)
|
|
return parse(min_version) <= parse(pkg_version) <= parse(max_version)
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def run_pip(*args):
|
|
subprocess.run([sys.executable, "-m", "pip", "install", *args], check=False)
|
|
|
|
|
|
def run_uv_pip(*args):
|
|
subprocess.run([sys.executable, "-m", "uv", "pip", "install", *args], check=False)
|
|
|
|
|
|
def install():
|
|
deps = [
|
|
# requirements
|
|
("ultralytics", "8.2.0", None),
|
|
("mediapipe", "0.10.13", None),
|
|
("rich", "13.0.0", None),
|
|
# mediapipe
|
|
("protobuf", "4.25.3", "4.9999"),
|
|
]
|
|
|
|
if not is_installed("uv", "0.1.44", None):
|
|
run_pip("uv>=0.1.44")
|
|
|
|
os.environ["UV_PYTHON"] = sys.executable
|
|
|
|
pkgs = []
|
|
for pkg, low, high in deps:
|
|
if not is_installed(pkg, low, high):
|
|
if low and high:
|
|
cmd = f"{pkg}>={low},<={high}"
|
|
elif low:
|
|
cmd = f"{pkg}>={low}"
|
|
elif high:
|
|
cmd = f"{pkg}<={high}"
|
|
else:
|
|
cmd = pkg
|
|
pkgs.append(cmd)
|
|
|
|
run_uv_pip(*pkgs)
|
|
|
|
|
|
try:
|
|
import launch
|
|
|
|
skip_install = launch.args.skip_install
|
|
except Exception:
|
|
skip_install = False
|
|
|
|
if not skip_install:
|
|
install()
|