mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
2a35d93a4d
**Service Dependencies** Services that depend on other services now access those services via the `Invoker` object. This object is provided to the service as a kwarg to its `start()` method. Until now, most services did not utilize this feature, and several services required their dependencies to be initialized and passed in on init. Additionally, _all_ services are now registered as invocation services - including the low-level services. This obviates issues with inter-dependent services we would otherwise experience as we add workflow storage. **Database Access** Previously, we were passing in a separate sqlite connection and corresponding lock as args to services in their init. A good amount of posturing was done in each service that uses the db. These objects, along with the sqlite startup and cleanup logic, is now abstracted into a simple `SqliteDatabase` class. This creates the shared connection and lock objects, enables foreign keys, and provides a `clean()` method to do startup db maintenance. This is not a service as it's only used by sqlite services.
133 lines
4.3 KiB
Python
133 lines
4.3 KiB
Python
from abc import ABC, abstractmethod
|
|
|
|
from invokeai.app.services.board_images import board_record_to_dto
|
|
from invokeai.app.services.board_record_storage import BoardChanges
|
|
from invokeai.app.services.image_record_storage import OffsetPaginatedResults
|
|
from invokeai.app.services.invoker import Invoker
|
|
from invokeai.app.services.models.board_record import BoardDTO
|
|
|
|
|
|
class BoardServiceABC(ABC):
|
|
"""High-level service for board management."""
|
|
|
|
@abstractmethod
|
|
def create(
|
|
self,
|
|
board_name: str,
|
|
) -> BoardDTO:
|
|
"""Creates a board."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_dto(
|
|
self,
|
|
board_id: str,
|
|
) -> BoardDTO:
|
|
"""Gets a board."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def update(
|
|
self,
|
|
board_id: str,
|
|
changes: BoardChanges,
|
|
) -> BoardDTO:
|
|
"""Updates a board."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def delete(
|
|
self,
|
|
board_id: str,
|
|
) -> None:
|
|
"""Deletes a board."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_many(
|
|
self,
|
|
offset: int = 0,
|
|
limit: int = 10,
|
|
) -> OffsetPaginatedResults[BoardDTO]:
|
|
"""Gets many boards."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_all(
|
|
self,
|
|
) -> list[BoardDTO]:
|
|
"""Gets all boards."""
|
|
pass
|
|
|
|
|
|
class BoardService(BoardServiceABC):
|
|
__invoker: Invoker
|
|
|
|
def start(self, invoker: Invoker) -> None:
|
|
self.__invoker = invoker
|
|
|
|
def create(
|
|
self,
|
|
board_name: str,
|
|
) -> BoardDTO:
|
|
board_record = self.__invoker.services.board_records.save(board_name)
|
|
return board_record_to_dto(board_record, None, 0)
|
|
|
|
def get_dto(self, board_id: str) -> BoardDTO:
|
|
board_record = self.__invoker.services.board_records.get(board_id)
|
|
cover_image = self.__invoker.services.image_records.get_most_recent_image_for_board(board_record.board_id)
|
|
if cover_image:
|
|
cover_image_name = cover_image.image_name
|
|
else:
|
|
cover_image_name = None
|
|
image_count = self.__invoker.services.board_image_records.get_image_count_for_board(board_id)
|
|
return board_record_to_dto(board_record, cover_image_name, image_count)
|
|
|
|
def update(
|
|
self,
|
|
board_id: str,
|
|
changes: BoardChanges,
|
|
) -> BoardDTO:
|
|
board_record = self.__invoker.services.board_records.update(board_id, changes)
|
|
cover_image = self.__invoker.services.image_records.get_most_recent_image_for_board(board_record.board_id)
|
|
if cover_image:
|
|
cover_image_name = cover_image.image_name
|
|
else:
|
|
cover_image_name = None
|
|
|
|
image_count = self.__invoker.services.board_image_records.get_image_count_for_board(board_id)
|
|
return board_record_to_dto(board_record, cover_image_name, image_count)
|
|
|
|
def delete(self, board_id: str) -> None:
|
|
self.__invoker.services.board_records.delete(board_id)
|
|
|
|
def get_many(self, offset: int = 0, limit: int = 10) -> OffsetPaginatedResults[BoardDTO]:
|
|
board_records = self.__invoker.services.board_records.get_many(offset, limit)
|
|
board_dtos = []
|
|
for r in board_records.items:
|
|
cover_image = self.__invoker.services.image_records.get_most_recent_image_for_board(r.board_id)
|
|
if cover_image:
|
|
cover_image_name = cover_image.image_name
|
|
else:
|
|
cover_image_name = None
|
|
|
|
image_count = self.__invoker.services.board_image_records.get_image_count_for_board(r.board_id)
|
|
board_dtos.append(board_record_to_dto(r, cover_image_name, image_count))
|
|
|
|
return OffsetPaginatedResults[BoardDTO](items=board_dtos, offset=offset, limit=limit, total=len(board_dtos))
|
|
|
|
def get_all(self) -> list[BoardDTO]:
|
|
board_records = self.__invoker.services.board_records.get_all()
|
|
board_dtos = []
|
|
for r in board_records:
|
|
cover_image = self.__invoker.services.image_records.get_most_recent_image_for_board(r.board_id)
|
|
if cover_image:
|
|
cover_image_name = cover_image.image_name
|
|
else:
|
|
cover_image_name = None
|
|
|
|
image_count = self.__invoker.services.board_image_records.get_image_count_for_board(r.board_id)
|
|
board_dtos.append(board_record_to_dto(r, cover_image_name, image_count))
|
|
|
|
return board_dtos
|