mirror of
https://github.com/facefusion/facefusion.git
synced 2026-04-29 04:55:57 +02:00
Henry Ruhs
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
import os
|
|
from typing import Any, Union
|
|
|
|
from facefusion.app_context import detect_app_context
|
|
from facefusion.processors.types import ProcessorState, ProcessorStateKey, ProcessorStateSet
|
|
from facefusion.session_context import get_session_id
|
|
from facefusion.types import Args, State, StateKey, StateSet
|
|
|
|
STATE_SET : Union[StateSet, ProcessorStateSet] =\
|
|
{
|
|
'api': {}, #type:ignore[assignment]
|
|
'cli': {} #type:ignore[assignment]
|
|
}
|
|
|
|
|
|
def get_state() -> Union[State, ProcessorState]:
|
|
app_context = detect_app_context()
|
|
return STATE_SET.get(app_context) #type:ignore[return-value]
|
|
|
|
|
|
def collect_state(args : Args) -> Union[State, ProcessorState]:
|
|
state =\
|
|
{
|
|
key: get_item(key) for key in args #type:ignore[arg-type]
|
|
}
|
|
return state #type:ignore[return-value]
|
|
|
|
|
|
def init_item(key : Union[StateKey, ProcessorStateKey], value : Any) -> None:
|
|
STATE_SET['api'][key] = value #type:ignore[literal-required]
|
|
STATE_SET['cli'][key] = value #type:ignore[literal-required]
|
|
|
|
|
|
def get_item(key : Union[StateKey, ProcessorStateKey]) -> Any:
|
|
return get_state().get(key) #type:ignore[literal-required]
|
|
|
|
|
|
def set_item(key : Union[StateKey, ProcessorStateKey], value : Any) -> None:
|
|
app_context = detect_app_context()
|
|
STATE_SET[app_context][key] = value #type:ignore[literal-required]
|
|
|
|
|
|
def clear_item(key : Union[StateKey, ProcessorStateKey]) -> None:
|
|
set_item(key, None)
|
|
|
|
|
|
def get_jobs_path() -> str:
|
|
jobs_path = get_item('jobs_path')
|
|
session_id = get_session_id()
|
|
|
|
if session_id:
|
|
return os.path.join(jobs_path, session_id)
|
|
return jobs_path
|
|
|
|
|
|
def get_temp_path() -> str:
|
|
temp_path = get_item('temp_path')
|
|
session_id = get_session_id()
|
|
|
|
if session_id:
|
|
return os.path.join(temp_path, session_id)
|
|
return temp_path
|