wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
101 lines
4.1 KiB
Python
101 lines
4.1 KiB
Python
from fastapi import Body, HTTPException, Path, Query
|
|
from fastapi.routing import APIRouter
|
|
|
|
from invokeai.app.api.auth_dependencies import CurrentUserOrDefault
|
|
from invokeai.app.api.dependencies import ApiDependencies
|
|
from invokeai.backend.util.logging import logging
|
|
|
|
client_state_router = APIRouter(prefix="/v1/client_state", tags=["client_state"])
|
|
|
|
|
|
@client_state_router.get(
|
|
"/{queue_id}/get_by_key",
|
|
operation_id="get_client_state_by_key",
|
|
response_model=str | None,
|
|
)
|
|
async def get_client_state_by_key(
|
|
current_user: CurrentUserOrDefault,
|
|
queue_id: str = Path(description="The queue id (ignored, kept for backwards compatibility)"),
|
|
key: str = Query(..., description="Key to get"),
|
|
) -> str | None:
|
|
"""Gets the client state for the current user (or system user if not authenticated)"""
|
|
try:
|
|
return ApiDependencies.invoker.services.client_state_persistence.get_by_key(current_user.user_id, key)
|
|
except Exception as e:
|
|
logging.error(f"Error getting client state: {e}")
|
|
raise HTTPException(status_code=500, detail="Error getting client state")
|
|
|
|
|
|
@client_state_router.post(
|
|
"/{queue_id}/set_by_key",
|
|
operation_id="set_client_state",
|
|
response_model=str,
|
|
)
|
|
async def set_client_state(
|
|
current_user: CurrentUserOrDefault,
|
|
queue_id: str = Path(description="The queue id (ignored, kept for backwards compatibility)"),
|
|
key: str = Query(..., description="Key to set"),
|
|
value: str = Body(..., description="Stringified value to set"),
|
|
) -> str:
|
|
"""Sets the client state for the current user (or system user if not authenticated)"""
|
|
try:
|
|
return ApiDependencies.invoker.services.client_state_persistence.set_by_key(current_user.user_id, key, value)
|
|
except Exception as e:
|
|
logging.error(f"Error setting client state: {e}")
|
|
raise HTTPException(status_code=500, detail="Error setting client state")
|
|
|
|
|
|
@client_state_router.get(
|
|
"/{queue_id}/get_keys_by_prefix",
|
|
operation_id="get_client_state_keys_by_prefix",
|
|
response_model=list[str],
|
|
)
|
|
async def get_client_state_keys_by_prefix(
|
|
current_user: CurrentUserOrDefault,
|
|
queue_id: str = Path(description="The queue id (ignored, kept for backwards compatibility)"),
|
|
prefix: str = Query(..., description="Prefix to filter keys by"),
|
|
) -> list[str]:
|
|
"""Gets client state keys matching a prefix for the current user"""
|
|
try:
|
|
return ApiDependencies.invoker.services.client_state_persistence.get_keys_by_prefix(
|
|
current_user.user_id, prefix
|
|
)
|
|
except Exception as e:
|
|
logging.error(f"Error getting client state keys: {e}")
|
|
raise HTTPException(status_code=500, detail="Error getting client state keys")
|
|
|
|
|
|
@client_state_router.post(
|
|
"/{queue_id}/delete_by_key",
|
|
operation_id="delete_client_state_by_key",
|
|
responses={204: {"description": "Client state key deleted"}},
|
|
)
|
|
async def delete_client_state_by_key(
|
|
current_user: CurrentUserOrDefault,
|
|
queue_id: str = Path(description="The queue id (ignored, kept for backwards compatibility)"),
|
|
key: str = Query(..., description="Key to delete"),
|
|
) -> None:
|
|
"""Deletes a specific client state key for the current user"""
|
|
try:
|
|
ApiDependencies.invoker.services.client_state_persistence.delete_by_key(current_user.user_id, key)
|
|
except Exception as e:
|
|
logging.error(f"Error deleting client state key: {e}")
|
|
raise HTTPException(status_code=500, detail="Error deleting client state key")
|
|
|
|
|
|
@client_state_router.post(
|
|
"/{queue_id}/delete",
|
|
operation_id="delete_client_state",
|
|
responses={204: {"description": "Client state deleted"}},
|
|
)
|
|
async def delete_client_state(
|
|
current_user: CurrentUserOrDefault,
|
|
queue_id: str = Path(description="The queue id (ignored, kept for backwards compatibility)"),
|
|
) -> None:
|
|
"""Deletes the client state for the current user (or system user if not authenticated)"""
|
|
try:
|
|
ApiDependencies.invoker.services.client_state_persistence.delete(current_user.user_id)
|
|
except Exception as e:
|
|
logging.error(f"Error deleting client state: {e}")
|
|
raise HTTPException(status_code=500, detail="Error deleting client state")
|