add product base type definition 🔧

This commit is contained in:
Ondrej Samohel 2025-05-16 18:13:33 +02:00
parent f3beea1ff8
commit b26bf73be0
No known key found for this signature in database
GPG key ID: 02376E18990A97C6

View file

@ -0,0 +1,47 @@
"""Base product types for the pipeline creation process."""
from dataclasses import dataclass
@dataclass
class BaseProductType:
"""Base class for product types."""
name: str
description: str
icon: str = "cube"
color: str = "#FFFFFF"
@dataclass
class Image(BaseProductType):
"""Image product type."""
name: str = "Image"
description: str = "An image product."
icon: str = "image"
color: str = "#FF0000"
@dataclass
class Video(BaseProductType):
"""Video product type."""
name: str = "Video"
description: str = "A video product."
icon: str = "video"
color: str = "#00FF00"
@dataclass
class Audio(BaseProductType):
"""Audio product type."""
name: str = "Audio"
description: str = "An audio product."
icon: str = "audio"
color: str = "#0000FF"
@dataclass
class Model(BaseProductType):
"""Document product type."""
name: str = "Model"
description: str = "A model product."
icon: str = "model"
color: str = "#FFFF00"