facefusion/facefusion/temp_helper.py

import os

from facefusion import state_manager
from facefusion.filesystem import create_directory, get_file_extension, get_file_name, move_file, remove_directory, resolve_file_pattern
from facefusion.types import FrameSet


def get_temp_file_path(file_path : str) -> str:
	temp_directory_path = get_temp_directory_path(file_path)
	temp_file_extension = get_file_extension(file_path)
	return os.path.join(temp_directory_path, 'temp' + temp_file_extension)


def move_temp_file(file_path : str, move_path : str) -> bool:
	temp_file_path = get_temp_file_path(file_path)
	return move_file(temp_file_path, move_path)


def resolve_temp_frame_set(target_path : str) -> FrameSet:
	temp_frame_pattern = get_temp_frame_pattern(target_path, '*')
	temp_frame_set = {}

	for temp_frame_path in resolve_file_pattern(temp_frame_pattern):
		frame_number = int(get_file_name(temp_frame_path))
		temp_frame_set[frame_number] = temp_frame_path

	return temp_frame_set


def get_temp_frame_pattern(target_path : str, temp_frame_prefix : str) -> str:
	temp_directory_path = get_temp_directory_path(target_path)
	return os.path.join(temp_directory_path, temp_frame_prefix + '.' + state_manager.get_item('temp_frame_format'))


def get_temp_directory_path(file_path : str) -> str:
	temp_file_name = get_file_name(file_path)
	return os.path.join(state_manager.get_item('temp_path'), 'facefusion', temp_file_name)


def create_temp_directory(file_path : str) -> bool:
	temp_directory_path = get_temp_directory_path(file_path)
	return create_directory(temp_directory_path)


def clear_temp_directory(file_path : str) -> bool:
	temp_directory_path = get_temp_directory_path(file_path)
	return remove_directory(temp_directory_path)