Access package version

This commit is contained in:
huchenlei 2025-03-10 20:43:59 -04:00
parent 65ea778a5e
commit cfbe4b49ca

View File

@ -11,6 +11,7 @@ from dataclasses import dataclass
from functools import cached_property
from pathlib import Path
from typing import TypedDict, Optional
from importlib.metadata import version
import requests
from typing_extensions import NotRequired
@ -36,15 +37,14 @@ def check_frontend_version():
return tuple(map(int, version.split(".")))
try:
import comfyui_frontend_package
frontend_version = parse_version(comfyui_frontend_package.__version__)
frontend_version_str = version("comfyui-frontend-package")
frontend_version = parse_version(frontend_version_str)
with open(req_path, "r", encoding="utf-8") as f:
required_frontend = parse_version(f.readline().split("=")[-1])
if frontend_version < required_frontend:
logging.warning("________________________________________________________________________\nWARNING WARNING WARNING WARNING WARNING\n\nInstalled frontend version {} is lower than the recommended version {}.\n\n{}\n________________________________________________________________________".format('.'.join(map(str, frontend_version)), '.'.join(map(str, required_frontend)), frontend_install_warning_message()))
else:
logging.info("ComfyUI frontend version: {}".format(comfyui_frontend_package.__version__))
logging.info("ComfyUI frontend version: {}".format(frontend_version_str))
except Exception as e:
logging.error(f"Failed to check frontend version: {e}")