Upload dependency UI

This commit is contained in:
Nathan
2024-05-16 00:46:28 -07:00
parent 206b1dac4c
commit 4519fa7581
9 changed files with 208 additions and 122 deletions
+42
View File
@@ -0,0 +1,42 @@
import os
import re
import folder_paths
def update_requirements():
"""
Formats requirements.txt
"""
base = folder_paths.base_path
custom_nodes_dir = os.path.join(base, "custom_nodes")
# Loop over the files in the specified directory
for custom_node in os.listdir(custom_nodes_dir):
filepath = os.path.join(custom_nodes_dir, custom_node, "requirements.txt")
# Check if the file is a requirements.txt file
if os.path.exists(filepath):
# Read the content of the requirements.txt file
with open(filepath, 'r') as file:
lines = file.readlines()
# Update dependencies in the file content
updated_lines = []
for line in lines:
# Check if the line contains a git dependency without the specified format
match_git = re.match(r'^\s*git\+https://.*?/([^/]+)\.git', line)
match_git_plus = re.match(r'^\s*git\+https://.*?/([^/]+)$', line)
if match_git:
package_name = match_git.group(1)
updated_lines.append(f"{package_name} @ {line.strip()}\n")
elif match_git_plus:
package_name = match_git_plus.group(1).strip()
updated_lines.append(f"{package_name} @ {line.strip()}\n")
else:
updated_lines.append(line)
# Write the updated content back to the requirements.txt file
with open(filepath, 'w') as file:
file.writelines(updated_lines)
+5 -5
View File
@@ -4,11 +4,11 @@ from enum import Enum
from typing import Any, Dict
class TaskStatus(Enum):
STARTED = "Task started"
COMPLETED = "Task completed"
HASHING = "Task hashing"
UPLOADING = "Task uploading"
ERROR = "Task failed"
STARTED = "Started"
COMPLETED = "Completed"
HASHING = "Hashing"
UPLOADING = "Uploading"
ERROR = "Failed"
@dataclasses.dataclass
class Task: