New column types + example columns

Now supporting:
- Many enums
- date
- datetime
- bool

Author: joce

src/appui/formatting.py

@@ -2,9 +2,32 @@
 
 from __future__ import annotations
 
-from typing import Final
+from typing import TYPE_CHECKING, Final
+
+if TYPE_CHECKING:
+    from datetime import date, datetime
+    from enum import Enum
 
 _NO_VALUE: Final[str] = "N/A"
+_DEFAULT_DATE_FORMAT: Final[str] = "%Y-%m-%d"
+_DEFAULT_DATETIME_FORMAT: Final[str] = "%Y-%m-%d %H:%M"
+_CHECKED_VALUE: Final[str] = "☑"
+_UNCHECKED_VALUE: Final[str] = "☐"
+
+# TODO: Allow user-configurable date/time formats via the app configuration.
+
+
+def _as_title_case(value: str) -> str:
+    """Return a title-cased string with underscores treated as word separators.
+
+    Args:
+        value (str): The string to normalize.
+
+    Returns:
+        str: The normalized, title-cased label.
+    """
+
+    return " ".join(word.capitalize() for word in value.replace("_", " ").split())
 
 
 def as_percent(value: float | None) -> str:
@@ -72,3 +95,66 @@ def as_compact(value: int | None) -> str:
         return f"{value / 1000000000:.2f}B"
 
     return f"{value / 1000000000000:.2f}T"
+
+
+def as_date(value: date | None, fmt: str | None = None) -> str:
+    """Return the value formatted as a date string.
+
+    Args:
+        value (date | None): The date value to format.
+        fmt (str | None): Optional format string to override the default.
+
+    Returns:
+        str: The formatted date string or a placeholder if the value is None.
+    """
+
+    if value is None:
+        return _NO_VALUE
+    return value.strftime(fmt or _DEFAULT_DATE_FORMAT)
+
+
+def as_datetime(value: datetime | None, fmt: str | None = None) -> str:
+    """Return the value formatted as a datetime string.
+
+    Args:
+        value (datetime | None): The datetime value to format.
+        fmt (str | None): Optional format string to override the default.
+
+    Returns:
+        str: The formatted datetime string or a placeholder if the value is None.
+    """
+
+    if value is None:
+        return _NO_VALUE
+    return value.strftime(fmt or _DEFAULT_DATETIME_FORMAT)
+
+
+def as_enum(value: Enum | None) -> str:
+    """Return the value formatted as a title-cased enum label.
+
+    Args:
+        value (Enum | None): The enumeration value to format.
+
+    Returns:
+        str: The formatted enum label or a placeholder if the value is None.
+    """
+
+    if value is None:
+        return _NO_VALUE
+    raw_value = value.value if isinstance(value.value, str) else value.name
+    return _as_title_case(str(raw_value))
+
+
+def as_bool(*, value: bool | None) -> str:
+    """Return the value formatted as a checkbox string.
+
+    Args:
+        value (bool | None): The boolean value to format.
+
+    Returns:
+        str: A checked or unchecked box, or a placeholder if the value is None.
+    """
+
+    if value is None:
+        return _NO_VALUE
+    return _CHECKED_VALUE if value else _UNCHECKED_VALUE

src/appui/quote_column_definitions.py

@@ -6,10 +6,21 @@
 
 from .enhanced_data_table import EnhancedTableCell
 from .enums import Justify
-from .formatting import as_compact, as_float, as_percent
+from .formatting import (
+    as_bool,
+    as_compact,
+    as_date,
+    as_datetime,
+    as_enum,
+    as_float,
+    as_percent,
+)
 from .quote_table import quote_column
 
 if TYPE_CHECKING:
+    from datetime import date, datetime
+    from enum import Enum
+
     from .quote_table import QuoteColumn
 
 
@@ -184,6 +195,127 @@ def __init__(
         )
 
 
+class DateCell(EnhancedTableCell):
+    """Cell that renders date values."""
+
+    def __init__(
+        self,
+        value: date | None,
+        *,
+        date_format: str | None = None,
+        justification: Justify = Justify.LEFT,
+        style: str = "",
+        secondary_key: str | None = None,
+    ) -> None:
+        """Initialize the date cell.
+
+        Args:
+            value (date | None): The date value to display.
+            date_format (str | None): Optional format override for display.
+            justification (Justify): The text justification.
+            style (str): The style string for the cell.
+            secondary_key (str | None): An optional secondary string key to use for
+                tie-breaking during sorting.
+        """
+
+        safe_value = float("-inf") if value is None else value.toordinal()
+        super().__init__(
+            _with_secondary_key(safe_value, secondary_key),
+            as_date(value, date_format),
+            justification,
+            style,
+        )
+
+
+class DateTimeCell(EnhancedTableCell):
+    """Cell that renders datetime values."""
+
+    def __init__(
+        self,
+        value: datetime | None,
+        *,
+        datetime_format: str | None = None,
+        justification: Justify = Justify.LEFT,
+        style: str = "",
+        secondary_key: str | None = None,
+    ) -> None:
+        """Initialize the datetime cell.
+
+        Args:
+            value (datetime | None): The datetime value to display.
+            datetime_format (str | None): Optional format override for display.
+            justification (Justify): The text justification.
+            style (str): The style string for the cell.
+            secondary_key (str | None): An optional secondary string key to use for
+                tie-breaking during sorting.
+        """
+
+        safe_value = float("-inf") if value is None else value.timestamp()
+        super().__init__(
+            _with_secondary_key(safe_value, secondary_key),
+            as_datetime(value, datetime_format),
+            justification,
+            style,
+        )
+
+
+class EnumCell(EnhancedTableCell):
+    """Cell that renders enum values in title case."""
+
+    def __init__(
+        self,
+        value: Enum | None,
+        *,
+        justification: Justify = Justify.LEFT,
+        style: str = "",
+        secondary_key: str | None = None,
+    ) -> None:
+        """Initialize the enum cell.
+
+        Args:
+            value (Enum | None): The enum value to display.
+            justification (Justify): The text justification.
+            style (str): The style string for the cell.
+            secondary_key (str | None): An optional secondary string key to use for
+                tie-breaking during sorting.
+        """
+
+        display_value = as_enum(value)
+        primary = display_value.lower() if value is not None else ""
+        sort_key = (primary, secondary_key.lower()) if secondary_key else (primary,)
+        super().__init__(sort_key, display_value, justification, style)
+
+
+class BooleanCell(EnhancedTableCell):
+    """Cell that renders boolean values as checkboxes."""
+
+    def __init__(
+        self,
+        *,
+        value: bool | None,
+        justification: Justify = Justify.CENTER,
+        style: str = "",
+        secondary_key: str | None = None,
+    ) -> None:
+        """Initialize the boolean cell.
+
+        Args:
+            value (bool | None): The boolean value to display.
+            justification (Justify): The text justification.
+            style (str): The style string for the cell.
+            secondary_key (str | None): An optional secondary string key to use for
+                tie-breaking during sorting.
+        """
+
+        safe_value = float("-inf") if value is None else float(value)
+        super().__init__(
+            _with_secondary_key(safe_value, secondary_key),
+            as_bool(value=value),
+            justification,
+            style,
+        )
+
+
 def _get_style_for_value(value: float) -> str:
     """Get the style string based on the sign of a value.
 
@@ -391,6 +523,90 @@ def _get_style_for_value(value: float) -> str:
             ),
         )
     ),
+    "dividend_date": (
+        quote_column(
+            "Div Date",
+            full_name="Dividend Date",
+            width=10,
+            key="dividend_date",
+            justification=Justify.LEFT,
+            cell_factory=lambda q: DateCell(
+                q.dividend_date,
+                justification=Justify.LEFT,
+                secondary_key=q.symbol or "",
+            ),
+        )
+    ),
+    "market_state": (
+        quote_column(
+            "Mkt State",
+            full_name="Market State",
+            width=10,
+            key="market_state",
+            justification=Justify.LEFT,
+            cell_factory=lambda q: EnumCell(
+                q.market_state,
+                justification=Justify.LEFT,
+                secondary_key=q.symbol or "",
+            ),
+        )
+    ),
+    "option_type": (
+        quote_column(
+            "Opt Type",
+            full_name="Option Type",
+            width=8,
+            key="option_type",
+            justification=Justify.LEFT,
+            cell_factory=lambda q: EnumCell(
+                q.option_type,
+                justification=Justify.LEFT,
+                secondary_key=q.symbol or "",
+            ),
+        )
+    ),
+    "quote_type": (
+        quote_column(
+            "Type",
+            full_name="Quote Type",
+            width=15,
+            key="quote_type",
+            justification=Justify.LEFT,
+            cell_factory=lambda q: EnumCell(
+                q.quote_type,
+                justification=Justify.LEFT,
+                secondary_key=q.symbol or "",
+            ),
+        )
+    ),
+    "tradeable": (
+        quote_column(
+            "Tradeable",
+            full_name="Tradeable",
+            width=9,
+            key="tradeable",
+            justification=Justify.CENTER,
+            cell_factory=lambda q: BooleanCell(
+                value=q.tradeable,
+                justification=Justify.CENTER,
+                secondary_key=q.symbol or "",
+            ),
+        )
+    ),
+    "post_market_datetime": (
+        quote_column(
+            "Post Mkt",
+            full_name="Post-Market Datetime",
+            width=16,
+            key="post_market_datetime",
+            justification=Justify.LEFT,
+            cell_factory=lambda q: DateTimeCell(
+                q.post_market_datetime,
+                justification=Justify.LEFT,
+                secondary_key=q.symbol or "",
+            ),
+        )
+    ),
 }
 """
 A dictionary that contains QuoteColumns available for the quote table.

src/calahan/yquote.py

@@ -214,6 +214,7 @@ class YQuote(BaseModel):
 
     Applies to ALL quotes.
     """
+
     exchange: str
     """
     Securities exchange on which the security is traded.