From 32ae6fe0efc4f1d9ac1681b33b86616c78d756a3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 6 Jun 2023 16:24:20 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=89=20feat(flow=5Fstyle.py):=20add=20F?= =?UTF-8?q?lowStyle=20model=20and=20its=20CRUD=20classes=20The=20FlowStyle?= =?UTF-8?q?=20model=20is=20added=20to=20the=20project,=20which=20represent?= =?UTF-8?q?s=20the=20style=20of=20a=20flow.=20It=20has=20a=20color=20and?= =?UTF-8?q?=20an=20emoji=20field,=20and=20a=20foreign=20key=20to=20the=20F?= =?UTF-8?q?low=20model.=20The=20CRUD=20classes=20are=20also=20added=20to?= =?UTF-8?q?=20the=20file,=20which=20are=20FlowStyleCreate,=20FlowStyleRead?= =?UTF-8?q?,=20and=20FlowStyleUpdate.=20These=20classes=20are=20used=20to?= =?UTF-8?q?=20create,=20read,=20and=20update=20FlowStyle=20instances=20res?= =?UTF-8?q?pectively.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/database/models/flow_style.py | 33 +++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 src/backend/langflow/database/models/flow_style.py diff --git a/src/backend/langflow/database/models/flow_style.py b/src/backend/langflow/database/models/flow_style.py new file mode 100644 index 000000000..fe53799fe --- /dev/null +++ b/src/backend/langflow/database/models/flow_style.py @@ -0,0 +1,33 @@ +# Path: src/backend/langflow/database/models/flowstyle.py + +from langflow.database.models.base import SQLModelSerializable +from sqlmodel import Field, Relationship +from uuid import UUID, uuid4 +from typing import TYPE_CHECKING, Optional + +if TYPE_CHECKING: + from langflow.database.models.flow import Flow + + +class FlowStyleBase(SQLModelSerializable): + color: str + emoji: str + flow_id: UUID = Field(default=None, foreign_key="flow.id") + + +class FlowStyle(FlowStyleBase, table=True): + id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True) + flow: "Flow" = Relationship(back_populates="style") + + +class FlowStyleUpdate(SQLModelSerializable): + color: Optional[str] = None + emoji: Optional[str] = None + + +class FlowStyleCreate(FlowStyleBase): + pass + + +class FlowStyleRead(FlowStyleBase): + id: UUID