mirror of
https://github.com/facefusion/facefusion.git
synced 2026-06-02 19:01:35 +02:00
Henry Ruhs
35 lines
795 B
Python
35 lines
795 B
Python
import os
|
|
import signal
|
|
import sys
|
|
from time import sleep
|
|
from types import FrameType
|
|
|
|
from facefusion import process_manager, state_manager
|
|
from facefusion.temp_helper import clear_temp_directory
|
|
from facefusion.types import ErrorCode
|
|
|
|
|
|
def fatal_exit(error_code : ErrorCode) -> None:
|
|
os._exit(error_code)
|
|
|
|
|
|
def hard_exit(error_code : ErrorCode) -> None:
|
|
sys.exit(error_code)
|
|
|
|
|
|
def signal_exit(signum : int, frame : FrameType) -> None:
|
|
graceful_exit(0)
|
|
|
|
|
|
def graceful_exit(error_code : ErrorCode) -> None:
|
|
signal.signal(signal.SIGINT, signal.SIG_IGN)
|
|
process_manager.stop()
|
|
|
|
while process_manager.is_processing():
|
|
sleep(0.5)
|
|
|
|
if state_manager.get_item('output_path'):
|
|
clear_temp_directory(state_manager.get_temp_path(), state_manager.get_item('output_path'))
|
|
|
|
hard_exit(error_code)
|