|
|
""" |
|
|
Temporary directory utilities for ComfyUI-SwissArmyKnife. |
|
|
Provides ComfyUI-aware temporary file and directory management that respects |
|
|
the --base-directory and --temp-directory flags. |
|
|
""" |
|
|
|
|
|
import os |
|
|
import tempfile |
|
|
from typing import Optional |
|
|
|
|
|
try: |
|
|
import folder_paths |
|
|
COMFYUI_AVAILABLE = True |
|
|
except ImportError: |
|
|
folder_paths = None |
|
|
COMFYUI_AVAILABLE = False |
|
|
|
|
|
|
|
|
def get_temp_directory() -> str: |
|
|
""" |
|
|
Get the base temporary directory for ComfyUI operations. |
|
|
|
|
|
This function respects ComfyUI's configuration: |
|
|
- Uses ComfyUI's user directory + '/temp' if available |
|
|
- Falls back to system temp directory if ComfyUI not available |
|
|
|
|
|
When using Docker with --base-directory, this ensures temp files |
|
|
are stored within the ComfyUI workspace rather than the container's /tmp. |
|
|
|
|
|
Returns: |
|
|
str: Path to the base temporary directory |
|
|
""" |
|
|
if COMFYUI_AVAILABLE and folder_paths is not None: |
|
|
try: |
|
|
|
|
|
user_dir = folder_paths.get_user_directory() |
|
|
temp_dir = os.path.join(user_dir, "temp") |
|
|
os.makedirs(temp_dir, exist_ok=True) |
|
|
return temp_dir |
|
|
except Exception as e: |
|
|
print(f"Warning: Could not get ComfyUI temp directory: {e}") |
|
|
|
|
|
|
|
|
return tempfile.gettempdir() |
|
|
|
|
|
|
|
|
def get_temp_subdirectory(subdir_name: str) -> str: |
|
|
""" |
|
|
Get or create a subdirectory within ComfyUI's temp directory. |
|
|
|
|
|
Args: |
|
|
subdir_name: Name of the subdirectory to create |
|
|
|
|
|
Returns: |
|
|
str: Path to the subdirectory |
|
|
|
|
|
Example: |
|
|
frames_dir = get_temp_subdirectory("comfyui_frames") |
|
|
""" |
|
|
base_temp = get_temp_directory() |
|
|
subdir = os.path.join(base_temp, subdir_name) |
|
|
os.makedirs(subdir, exist_ok=True) |
|
|
return subdir |
|
|
|
|
|
|
|
|
def get_temp_file_path(suffix: str = "", prefix: str = "tmp", subdir: Optional[str] = None) -> str: |
|
|
""" |
|
|
Generate a temporary file path within ComfyUI's temp directory. |
|
|
|
|
|
This is a replacement for tempfile.NamedTemporaryFile that respects |
|
|
ComfyUI's base directory configuration. |
|
|
|
|
|
Args: |
|
|
suffix: File extension (e.g., '.mp4', '.jpg') |
|
|
prefix: Filename prefix (default: 'tmp') |
|
|
subdir: Optional subdirectory within temp (e.g., 'video_processing') |
|
|
|
|
|
Returns: |
|
|
str: Full path to the temporary file (file is NOT created) |
|
|
|
|
|
Example: |
|
|
temp_video = get_temp_file_path(suffix='.mp4', subdir='videos') |
|
|
# Use the path to write your file |
|
|
with open(temp_video, 'wb') as f: |
|
|
f.write(video_data) |
|
|
""" |
|
|
if subdir: |
|
|
base_temp = get_temp_subdirectory(subdir) |
|
|
else: |
|
|
base_temp = get_temp_directory() |
|
|
|
|
|
|
|
|
import uuid |
|
|
unique_id = uuid.uuid4().hex[:8] |
|
|
filename = f"{prefix}_{unique_id}{suffix}" |
|
|
|
|
|
return os.path.join(base_temp, filename) |
|
|
|
