Files
nathannlu-ComfyUI-Cloud/python/upload/progress.py
T
2024-05-15 20:24:25 -07:00

23 lines
363 B
Python

import dataclasses
from typing import Dict
@dataclasses.dataclass
class FileProgress:
max: int
value: int
progress_dict: Dict[str, FileProgress] = {}
def progress_update(filename, value, max):
progress = progress_dict.get(filename, FileProgress(0, 0))
progress.value = value
progress.max = max
progress_dict[filename] = progress