Icons and empty labels support

README.md

@@ -63,7 +63,8 @@ See the [tests](/tests/test_py_d2) for more detailed usage examples.
 - [x] Shapes in shapes
 - [x] Arrow directions
 - [x] Markdown / block strings / code in shapes
-- [ ] Icons in shapes
+- [x] Icons in shapes
+- [x] Support for empty labels
 - [ ] SQL table shapes
 - [ ] Class shapes
 - [ ] Comments

src/py_d2/helpers.py

@@ -13,10 +13,13 @@ def add_label_and_properties(
     has_properties: bool = properties is not None and len(properties) > 0
 
     first_line = name
-    if label or has_properties:
+    if label is not None or has_properties:
         first_line += ":"
 
-    if label:
+    if label is not None and len(label) == 0:
+        first_line += ' ""'
+
+    if label and len(label) > 0:
         first_line += f" {label}"
 
     if has_properties:

src/py_d2/shape.py

@@ -72,6 +72,8 @@ def __init__(
         shapes: Optional[List[D2Shape]] = None,
         # The style of this shape
         style: Optional[D2Style] = None,
+        # An icon for this shape
+        icon: Optional[str] = None,
         # Connections for the child shapes (NOT the connections for this shape)
         connections: Optional[List[D2Connection]] = None,
         # A shape this is near
@@ -83,6 +85,7 @@ def __init__(
         self.shape = shape
         self.shapes = shapes or []
         self.style = style
+        self.icon = icon
         self.connections = connections or []
         self.near = near
         self.kwargs = kwargs
@@ -107,6 +110,9 @@ def lines(self) -> List[str]:
         if self.style:
             properties += self.style.lines()
 
+        if self.icon:
+            properties.append(f"icon: {self.icon}")
+
         for key, value in self.kwargs.items():
             other_property = value.lines()
             other_property_line_1 = other_property[0]

tests/test_py_d2/test_d2_shape.py

@@ -16,6 +16,11 @@ def test_d2_shape_label():
     assert str(shape) == "shape_name: shape_label"
 
 
+def test_d2_shape_empty_label():
+    shape = D2Shape(name="shape_name", label="")
+    assert str(shape) == 'shape_name: ""'
+
+
 def test_d2_shape_style():
     shape = D2Shape(name="shape_name", style=D2Style(fill="red"))
     assert str(shape) == "shape_name: {\n  style: {\n    fill: red\n  }\n}"
@@ -137,6 +142,11 @@ def test_d2_shape_shapes():
     assert str(shape_sequence_diagram) == "shape_name: {\n  shape: sequence_diagram\n}"
 
 
+def test_d2_shape_icon():
+    shape = D2Shape(name="shape_name", icon="https://icons.terrastruct.com/essentials%2F117-database.svg")
+    assert str(shape) == "shape_name: {\n  icon: https://icons.terrastruct.com/essentials%2F117-database.svg\n}"
+
+
 def test_d2_shape_near():
     shape = D2Shape(name="shape_name", near="some_other_shape")
     assert str(shape) == "shape_name: {\n  near: some_other_shape\n}"