FilesExpand file tree

master

/

install_util.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (43 loc) · 1.67 KB

master

/

install_util.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (43 loc) · 1.67 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

from pathlib import Path

import sys

import logging

import re

# The path to the requirements.txt file

requirements_path = Path(__file__).parents[1] / "requirements.txt"

def get_missing_requirements_message():

"""The warning message to display when a package is missing."""

extra = ""

if sys.flags.no_user_site:

extra = "-s "

return f"""

Please install the updated requirements.txt file by running:

{sys.executable} {extra}-m pip install -r {requirements_path}

If you are on the portable package you can run: update\\update_comfyui.bat to solve this problem.

""".strip()

def is_valid_version(version: str) -> bool:

"""Validate if a string is a valid semantic version (X.Y.Z format)."""

pattern = r"^(\d+)\.(\d+)\.(\d+)$"

return bool(re.match(pattern, version))

PACKAGE_VERSIONS = {}

def get_required_packages_versions():

if len(PACKAGE_VERSIONS) > 0:

return PACKAGE_VERSIONS.copy()

out = PACKAGE_VERSIONS

try:

with open(requirements_path, "r", encoding="utf-8") as f:

for line in f:

line = line.strip().replace(">=", "==")

s = line.split("==")

if len(s) == 2:

version_str = s[-1]

if not is_valid_version(version_str):

logging.debug(f"Invalid version format for {s[0]} in requirements.txt: {version_str}")

continue

out[s[0]] = version_str

return out.copy()

except FileNotFoundError:

logging.error("requirements.txt not found.")

return None

except Exception as e:

logging.error(f"Error reading requirements.txt: {e}")

return None