Merge pull request #154 from equinor/feat-add-height-width-in-nodes

added width and heigh ion node style

Author: rupakkatwal

alembic/versions/4fcae4b62683_added_height_and_width_in_nodestyle.py

@@ -0,0 +1,34 @@
+"""Added height and width in nodestyle
+
+Revision ID: 4fcae4b62683
+Revises: e915e38cedc2
+Create Date: 2025-10-17 12:55:10.801087
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '4fcae4b62683'
+down_revision: Union[str, None] = 'e915e38cedc2'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('node_style', sa.Column('width', sa.INTEGER(), nullable=False,server_default=sa.text('200')))
+    op.add_column('node_style', sa.Column('height', sa.INTEGER(), nullable=False,server_default=sa.text('150')))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('node_style', 'height')
+    op.drop_column('node_style', 'width')
+    # ### end Alembic commands ###

src/dtos/node_style_dtos.py

@@ -9,6 +9,8 @@ class NodeStyleDto(BaseModel):
     node_id: uuid.UUID
     x_position: int = 0
     y_position: int = 0
+    width: int = 200
+    height: int = 150
 
 
 class NodeStyleIncomingDto(NodeStyleDto):
@@ -27,6 +29,8 @@ def to_outgoing_dto(entity: NodeStyle) -> NodeStyleOutgoingDto:
             node_id=entity.node_id,
             x_position=entity.x_position,
             y_position=entity.y_position,
+            width=entity.width,
+            height=entity.height,
         )
 
     @staticmethod
@@ -36,6 +40,8 @@ def to_entity(dto: NodeStyleIncomingDto) -> NodeStyle:
             node_id=dto.node_id if dto.node_id else None,
             x_position=dto.x_position,
             y_position=dto.y_position,
+            width=dto.width,
+            height=dto.height,
         )
 
     @staticmethod

src/models/node_style.py

@@ -21,6 +21,8 @@ class NodeStyle(Base, BaseEntity):
 
     x_position: Mapped[int] = mapped_column(INT)
     y_position: Mapped[int] = mapped_column(INT)
+    width: Mapped[int] = mapped_column(INT)
+    height: Mapped[int] = mapped_column(INT)
 
     node: Mapped["Node"] = relationship("Node", back_populates="node_style")
 
@@ -30,9 +32,13 @@ def __init__(
         node_id: Optional[uuid.UUID],
         x_position: int = 0,
         y_position: int = 0,
+        width: int = 200,
+        height: int = 150,
     ):
         self.id = id
         if node_id:
             self.node_id = node_id
         self.x_position = x_position
         self.y_position = y_position
+        self.width = width
+        self.height = height

src/repositories/node_style_repository.py

@@ -22,6 +22,8 @@ async def update(self, entities: list[NodeStyle]) -> list[NodeStyle]:
             entity = entities[n]
             entity_to_update.x_position = entity.x_position
             entity_to_update.y_position = entity.y_position
+            entity_to_update.width = entity.width
+            entity_to_update.height = entity.height
             if entity.node_id:
                 entity_to_update = entity.node_id