ComfyUI-WorkflowPublisher/app/database/__init__.py

49 lines
1.1 KiB
Python

"""
数据库模块
提供工作流和工作流运行的数据库操作接口
"""
from .models import Workflow, WorkflowRun, WorkflowRunNode, ComfyUIServer
from .api import (
init_db,
save_workflow,
get_all_workflows,
get_latest_workflow_by_base_name,
get_workflow_by_version,
get_workflow,
delete_workflow,
create_workflow_run,
update_workflow_run_status,
create_workflow_run_nodes,
update_workflow_run_node_status,
get_workflow_run,
get_workflow_run_nodes,
get_pending_workflow_runs,
get_running_workflow_runs,
)
__all__ = [
# 模型
"Workflow",
"WorkflowRun",
"WorkflowRunNode",
"ComfyUIServer",
# API函数
"init_db",
"save_workflow",
"get_all_workflows",
"get_latest_workflow_by_base_name",
"get_workflow_by_version",
"get_workflow",
"delete_workflow",
"create_workflow_run",
"update_workflow_run_status",
"create_workflow_run_nodes",
"update_workflow_run_node_status",
"get_workflow_run",
"get_workflow_run_nodes",
"get_pending_workflow_runs",
"get_running_workflow_runs",
]