mirror of
https://github.com/facefusion/facefusion.git
synced 2026-06-23 20:49:59 +02:00
henryruhs
27 lines
710 B
Python
27 lines
710 B
Python
import signal
|
|
import sys
|
|
from time import sleep
|
|
|
|
from facefusion import process_manager, state_manager
|
|
from facefusion.temp_helper import clear_temp_directory
|
|
from facefusion.types import ErrorCode
|
|
|
|
|
|
def hard_exit(error_code : ErrorCode) -> None:
|
|
signal.signal(signal.SIGINT, signal.SIG_IGN)
|
|
sys.exit(error_code)
|
|
|
|
|
|
def conditional_exit(error_code : ErrorCode) -> None:
|
|
if state_manager.get_item('command') == 'headless-run':
|
|
hard_exit(error_code)
|
|
|
|
|
|
def graceful_exit(error_code : ErrorCode) -> None:
|
|
process_manager.stop()
|
|
while process_manager.is_processing():
|
|
sleep(0.5)
|
|
if state_manager.get_item('target_path'):
|
|
clear_temp_directory(state_manager.get_item('target_path'))
|
|
hard_exit(error_code)
|