mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
32 lines
788 B
Python
32 lines
788 B
Python
from abc import ABC, abstractmethod
|
|
from enum import Enum, EnumMeta
|
|
|
|
from invokeai.app.util.misc import uuid_string
|
|
|
|
|
|
class ResourceType(str, Enum, metaclass=EnumMeta):
|
|
"""Enum for resource types."""
|
|
|
|
IMAGE = "image"
|
|
LATENT = "latent"
|
|
|
|
|
|
class NameServiceBase(ABC):
|
|
"""Low-level service responsible for naming resources (images, latents, etc)."""
|
|
|
|
# TODO: Add customizable naming schemes
|
|
@abstractmethod
|
|
def create_image_name(self) -> str:
|
|
"""Creates a name for an image."""
|
|
pass
|
|
|
|
|
|
class SimpleNameService(NameServiceBase):
|
|
"""Creates image names from UUIDs."""
|
|
|
|
# TODO: Add customizable naming schemes
|
|
def create_image_name(self) -> str:
|
|
uuid_str = uuid_string()
|
|
filename = f"{uuid_str}.png"
|
|
return filename
|