Files
facefusion/facefusion/process_manager.py
T
Henry Ruhs d96580b34c convert content store (#1236)
* convert content store

* add local_id variables

* add local_id variables

* fix misconcept of clear() in stores, pass session id to threads
2026-09-11 21:17:22 +02:00

59 lines
1.1 KiB
Python

from facefusion import store_creator
from facefusion.session_context import get_session_id
from facefusion.types import ProcessState, Store
PROCESS_STORE : Store = store_creator.create_store('pending')
def init() -> None:
session_id = get_session_id()
store_creator.init_content(PROCESS_STORE, session_id)
def get_state() -> ProcessState:
session_id = get_session_id()
return store_creator.get_content(PROCESS_STORE, session_id)
def is_checking() -> bool:
return get_state() == 'checking'
def is_processing() -> bool:
return get_state() == 'processing'
def is_stopping() -> bool:
return get_state() == 'stopping'
def is_pending() -> bool:
return get_state() == 'pending'
def set_state(process_state : ProcessState) -> None:
session_id = get_session_id()
store_creator.set_content(PROCESS_STORE, session_id, process_state)
def check() -> None:
set_state('checking')
def start() -> None:
set_state('processing')
def stop() -> None:
set_state('stopping')
def end() -> None:
set_state('pending')
def clear() -> None:
session_id = get_session_id()
store_creator.init_content(PROCESS_STORE, session_id)