PyPI - datajunction-server - Versions diffs - 0.0.1a91.dev0__tar.gz → 0.0.1a91.dev1__tar.gz - Mend

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: datajunction-server
-Version: 0.0.1a91.dev0
+Version: 0.0.1a91.dev1
 Summary: DataJunction server library for running to a DataJunction server
 Project-URL: Homepage, https://datajunction.io
 Project-URL: Repository, https://github.com/DataJunction/dj

@@ -0,0 +1,5 @@
+"""
+Version for Hatch
+"""
+__version__ = "0.0.1a91.dev1"

@@ -0,0 +1,28 @@
+"""Added column description
+Revision ID: 135fa5833fed
+Revises: ae9eba981a2d
+Create Date: 2025-03-14 23:04:39.248333+00:00
+"""
+# pylint: disable=no-member, invalid-name, missing-function-docstring, unused-import, no-name-in-module
+import sqlalchemy as sa
+from alembic import op
+# revision identifiers, used by Alembic.
+revision = "135fa5833fed"
+down_revision = "ae9eba981a2d"
+branch_labels = None
+depends_on = None
+def upgrade():
+    with op.batch_alter_table("column", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("description", sa.String(), nullable=True))
+def downgrade():
+    with op.batch_alter_table("column", schema=None) as batch_op:
+        batch_op.drop_column("description")

@@ -1628,6 +1628,7 @@ async def set_column_display_name(
     column = await get_column(session, node.current, column_name)  # type: ignore
     column.display_name = display_name
     session.add(column)
+    await session.commit()
     await save_history(
         event=History(
             entity_type=EntityType.COLUMN_ATTRIBUTE,
@@ -1641,7 +1642,48 @@ async def set_column_display_name(
         ),
         session=session,
     )
+    return column
+@router.patch(
+    "/nodes/{node_name}/columns/{column_name}/description",
+    response_model=ColumnOutput,
+    status_code=201,
+)
+async def set_column_description(
+    node_name: str,
+    column_name: str,
+    description: str,
+    current_user: User = Depends(get_and_update_current_user),
+    save_history: Callable = Depends(get_save_history),
+    *,
+    session: AsyncSession = Depends(get_session),
+) -> ColumnOutput:
+    """
+    Set column description for the node
+    """
+    node = await Node.get_by_name(
+        session,
+        node_name,
+        options=[joinedload(Node.current)],
+    )
+    column = await get_column(session, node.current, column_name)  # type: ignore
+    column.description = description
+    session.add(column)
     await session.commit()
+    await save_history(
+        event=History(
+            entity_type=EntityType.COLUMN_ATTRIBUTE,
+            node=node.name,  # type: ignore
+            activity_type=ActivityType.UPDATE,
+            details={
+                "column": column.name,
+                "description": description,
+            },
+            user=current_user.username,
+        ),
+        session=session,
+    )
     return column

@@ -72,7 +72,7 @@ def rename_columns(
             and not (hasattr(expression, "alias") and expression.alias)  # type: ignore
         ):
             alias_name = expression.alias_or_name.identifier(False)  # type: ignore
-            if expression.alias_or_name.name in node_columns:  # type: ignore  # pragma: no cover
+            if expression.alias_or_name.name.lower() in node_columns:  # type: ignore  # pragma: no cover
                 alias_name = node.name + SEPARATOR + expression.alias_or_name.name  # type: ignore
             expression = expression.copy()
             expression.set_semantic_entity(alias_name)  # type: ignore

@@ -39,6 +39,7 @@ class Column(Base):  # type: ignore
         insert_default=lambda context: labelize(context.current_parameters.get("name")),
     )
     type: Mapped[Optional[ColumnType]] = mapped_column(ColumnTypeDecorator)
+    description: Mapped[Optional[str]] = mapped_column()
     dimension_id: Mapped[Optional[int]] = mapped_column(
         ForeignKey("node.id", ondelete="SET NULL", name="fk_column_dimension_id_node"),
@@ -156,6 +157,7 @@ class Column(Base):  # type: ignore
             name=self.name,
             display_name=self.display_name,
             type=self.type,
+            description=self.description,
             dimension_id=self.dimension_id,
             dimension_column=self.dimension_column,
             attributes=[

@@ -19,6 +19,7 @@ class ColumnYAML(TypedDict, total=False):
     type: str
     dimension: str
+    description: str
 class ColumnTypeDecorator(TypeDecorator):

@@ -590,6 +590,7 @@ class ColumnOutput(BaseModel):
     name: str
     display_name: Optional[str]
     type: str
+    description: Optional[str]
     attributes: Optional[List[AttributeOutput]]
     dimension: Optional[NodeNameOutput]
     partition: Optional[PartitionOutput]

@@ -1541,6 +1541,7 @@ async def test_updating_cube(
                 "display_name": "Discounted Orders Rate",
                 "type": "double",
                 "attributes": [],
+                "description": None,
                 "dimension": None,
                 "partition": None,
             },
@@ -1549,6 +1550,7 @@ async def test_updating_cube(
                 "display_name": "City",
                 "type": "string",
                 "attributes": [],
+                "description": None,
                 "dimension": None,
                 "partition": None,
             },
@@ -1557,6 +1559,7 @@ async def test_updating_cube(
                 "display_name": "Hire Date",
                 "type": "timestamp",
                 "attributes": [],
+                "description": None,
                 "dimension": None,
                 "partition": {
                     "type_": "temporal",

@@ -1087,6 +1087,7 @@ async def test_metric_expression_auto_aliased(module__client_with_roads: AsyncCl
     assert data["columns"] == [
         {
             "attributes": [],
+            "description": None,
             "dimension": None,
             "display_name": "Dream Count 4",
             "name": "basic_DOT_dream_count_4",

datajunction-server 0.0.1a91.dev0__tar.gz → 0.0.1a91.dev1__tar.gz

datajunction-server 0.0.1a91.dev0tar.gz → 0.0.1a91.dev1tar.gz