simstack 0.2.0__py3-none-any.whl

simstack/models/base_types.py

@@ -0,0 +1,208 @@
+from odmantic import Model, Reference
+from pydantic import model_validator
+
+from simstack.models import simstack_model
+
+
+@simstack_model
+class IntData(Model):
+    field_name: str = "int"
+    value: int
+
+    @model_validator(mode="before")
+    @classmethod
+    def ensure_fieldname(cls, data):
+        """Ensure fieldname is set for existing documents"""
+        if isinstance(data, dict) and "field_name" not in data:
+            data["field_name"] = "int"
+        return data
+
+    def make_table_entries(
+        self,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return {self.field_name: self.value}
+
+    def make_column_defs_instance(
+        self,
+        table_name=None,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return [
+            {
+                "field": self.field_name,
+                "headerName": self.field_name,
+                "type": "numericColumn",
+            }
+        ]
+
+    @classmethod
+    def ui_schema(cls, **kwargs) -> dict:
+        return {
+            "ui:field": "IntDataField"
+        }  # This tells RJSF to use the custom component
+
+
+@simstack_model
+class FloatData(Model):
+    field_name: str = "float"
+    value: float
+
+    @model_validator(mode="before")
+    @classmethod
+    def ensure_fieldname(cls, data):
+        """Ensure fieldname is set for existing documents"""
+        if isinstance(data, dict) and "field_name" not in data:
+            data["field_name"] = "float"
+        return data
+
+    def make_table_entries(
+        self,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return {self.field_name: self.value}
+
+    def make_column_defs_instance(
+        self,
+        table_name=None,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return [
+            {
+                "field": self.field_name,
+                "headerName": self.field_name,
+                "type": "numericColumn",
+            }
+        ]
+
+    @classmethod
+    def ui_schema(cls, **kwargs) -> dict:
+        return {
+            "ui:field": "FloatDataField"
+        }  # This tells RJSF to use the custom component
+
+
+@simstack_model
+class StringData(Model):
+    field_name: str = "text"
+    value: str
+
+    @model_validator(mode="before")
+    @classmethod
+    def ensure_fieldname(cls, data):
+        """Ensure fieldname is set for existing documents"""
+        if isinstance(data, dict) and "field_name" not in data:
+            data["field_name"] = "text"
+        return data
+
+    def make_table_entries(
+        self,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return {self.field_name: self.value}
+
+    def make_column_defs_instance(
+        self,
+        table_name=None,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return [
+            {
+                "field": self.field_name,
+                "headerName": self.field_name,
+                "type": "textColumn",
+            }
+        ]
+
+    @classmethod
+    def ui_schema(cls, **kwargs) -> dict:
+        return {
+            "ui:field": "StringDataField"
+        }  # This tells RJSF to use the custom component
+
+
+@simstack_model
+class BooleanData(Model):
+    field_name: str = "boolean"
+    value: bool
+
+    @model_validator(mode="before")
+    @classmethod
+    def ensure_fieldname(cls, data):
+        """Ensure fieldname is set for existing documents"""
+        if isinstance(data, dict) and "field_name" not in data:
+            data["field_name"] = "boolean"
+        return data
+
+    def make_table_entries(
+        self,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return {self.field_name: self.value}
+
+    def make_column_defs_instance(
+        self,
+        table_name=None,
+        max_recursion_level=1,
+        drop_id=True,
+        current_level=0,
+        visited=None,
+        field_prefix="",
+    ):
+        return [
+            {
+                "field": self.field_name,
+                "headerName": self.field_name,
+                "type": "booleanColumn",
+            }
+        ]
+
+    @classmethod
+    def ui_schema(cls, **kwargs) -> dict:
+        return {
+            "ui:field": "BooleanDataField"
+        }  # This tells RJSF to use the custom component
+
+
+@simstack_model
+class BinaryOperationInput(Model):
+    arg1: FloatData = Reference()
+    arg2: FloatData = Reference()
+
+    def __str__(self):
+        return f"BinaryOperationInput(arg1={self.arg1}, arg2={self.arg2})"
+
+
+@simstack_model
+class IteratorInput(Model):
+    start: int
+    stop: int
+    generator: str = "range"

simstack/models/charts_artifact.py

@@ -0,0 +1,411 @@
+from typing import List, Union, Literal, Optional, Dict, Any
+
+from odmantic import Model, Field, EmbeddedModel, ObjectId
+
+
+# Chart Series Definitions
+class AGChartSeriesBase(EmbeddedModel):
+    """Base class for AG-Charts series configuration."""
+
+    type: str = Field(..., description="Chart series type")
+    xKey: str = Field(..., description="Key for x-axis data")
+    yKey: str = Field(..., description="Key for y-axis data")
+    visible: Optional[bool] = Field(
+        default=True, description="Whether series is visible"
+    )
+    showInLegend: Optional[bool] = Field(default=True, description="Show in legend")
+    title: Optional[str] = Field(default=None, description="Series title")
+    # Chart data
+    data: List[Dict[str, Any]] = Field(default_factory=list, description="Chart data")
+
+
+class AGLineSeriesConfig(AGChartSeriesBase):
+    """AG-Charts line series configuration."""
+
+    type: Literal["line"] = "line"
+    strokeWidth: Optional[float] = Field(default=2, description="Line stroke width")
+    strokeOpacity: Optional[float] = Field(default=1, description="Line stroke opacity")
+    lineDash: Optional[List[float]] = Field(
+        default=None, description="Line dash pattern"
+    )
+    marker: Optional[Dict[str, Any]] = Field(
+        default=None, description="Marker configuration"
+    )
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+class AGBarSeriesConfig(AGChartSeriesBase):
+    """AG-Charts bar series configuration."""
+
+    type: Literal["bar"] = "bar"
+    fillOpacity: Optional[float] = Field(default=1, description="Bar fill opacity")
+    strokeWidth: Optional[float] = Field(default=0, description="Bar stroke width")
+    cornerRadius: Optional[float] = Field(default=0, description="Bar corner radius")
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+class AGColumnSeriesConfig(AGChartSeriesBase):
+    """AG-Charts column series configuration."""
+
+    type: Literal["column"] = "column"
+    fillOpacity: Optional[float] = Field(default=1, description="Column fill opacity")
+    strokeWidth: Optional[float] = Field(default=0, description="Column stroke width")
+    cornerRadius: Optional[float] = Field(default=0, description="Column corner radius")
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+class AGAreaSeriesConfig(AGChartSeriesBase):
+    """AG-Charts area series configuration."""
+
+    type: Literal["area"] = "area"
+    fillOpacity: Optional[float] = Field(default=0.8, description="Area fill opacity")
+    strokeWidth: Optional[float] = Field(default=2, description="Area stroke width")
+    marker: Optional[Dict[str, Any]] = Field(
+        default=None, description="Marker configuration"
+    )
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+class AGScatterSeriesConfig(AGChartSeriesBase):
+    """AG-Charts scatter series configuration."""
+
+    type: Literal["scatter"] = "scatter"
+    marker: Optional[Dict[str, Any]] = Field(
+        default=None, description="Marker configuration"
+    )
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+class AGPieSeriesConfig(EmbeddedModel):
+    """AG-Charts pie series configuration."""
+
+    type: Literal["pie"] = "pie"
+    angleKey: str = Field(..., description="Key for pie slice angles")
+    radiusKey: Optional[str] = Field(
+        default=None, description="Key for pie slice radius"
+    )
+    labelKey: Optional[str] = Field(
+        default=None, description="Key for pie slice labels"
+    )
+    visible: Optional[bool] = Field(
+        default=True, description="Whether series is visible"
+    )
+    showInLegend: Optional[bool] = Field(default=True, description="Show in legend")
+    title: Optional[str] = Field(default=None, description="Series title")
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+class AGDonutSeriesConfig(EmbeddedModel):
+    """AG-Charts donut series configuration."""
+
+    type: Literal["donut"] = "donut"
+    angleKey: str = Field(..., description="Key for donut slice angles")
+    radiusKey: Optional[str] = Field(
+        default=None, description="Key for donut slice radius"
+    )
+    labelKey: Optional[str] = Field(
+        default=None, description="Key for donut slice labels"
+    )
+    innerRadiusRatio: Optional[float] = Field(
+        default=0.6, description="Inner radius ratio"
+    )
+    visible: Optional[bool] = Field(
+        default=True, description="Whether series is visible"
+    )
+    showInLegend: Optional[bool] = Field(default=True, description="Show in legend")
+    title: Optional[str] = Field(default=None, description="Series title")
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tooltip configuration"
+    )
+
+
+# Union type for all series configurations
+AGChartSeries = Union[
+    AGLineSeriesConfig,
+    AGBarSeriesConfig,
+    AGColumnSeriesConfig,
+    AGAreaSeriesConfig,
+    AGScatterSeriesConfig,
+    AGPieSeriesConfig,
+    AGDonutSeriesConfig,
+]
+
+
+# Axis Configurations
+class AGChartAxisConfig(EmbeddedModel):
+    """AG-Charts axis configuration."""
+
+    type: Literal["category", "number", "time", "log"] = Field(
+        ..., description="Axis type"
+    )
+    position: Literal["top", "right", "bottom", "left"] = Field(
+        ..., description="Axis position"
+    )
+    title: Optional[str] = Field(default=None, description="Axis title")
+    min: Optional[float] = Field(default=None, description="Minimum axis value")
+    max: Optional[float] = Field(default=None, description="Maximum axis value")
+    tick: Optional[Dict[str, Any]] = Field(
+        default=None, description="Tick configuration"
+    )
+    label: Optional[Dict[str, Any]] = Field(
+        default=None, description="Label configuration"
+    )
+    gridStyle: Optional[List[Dict[str, Any]]] = Field(
+        default=None, description="Grid line style"
+    )
+
+
+# Legend Configuration
+class AGChartLegendConfig(EmbeddedModel):
+    """AG-Charts legend configuration."""
+
+    enabled: bool = Field(default=True, description="Enable legend")
+    position: Literal["top", "right", "bottom", "left"] = Field(
+        default="right", description="Legend position"
+    )
+    spacing: float = Field(default=20, description="Legend spacing")
+    item: Dict[str, Any] = Field(
+        default_factory=dict, description="Legend item configuration"
+    )
+
+
+# Title Configuration
+class AGChartTitleConfig(EmbeddedModel):
+    """AG-Charts title configuration."""
+
+    text: str = Field("Chart Title", description="Title text")
+    # fontSize: Optional[int] = Field(default=16, description="Title font size")
+    # fontWeight: Optional[str] = Field(default="bold", description="Title font weight")
+    # color: Optional[str] = Field(default=None, description="Title color")
+
+
+# Subtitle Configuration
+class AGChartSubtitleConfig(EmbeddedModel):
+    """AG-Charts subtitle configuration."""
+
+    text: str = Field(..., description="Subtitle text")
+    fontSize: Optional[int] = Field(default=12, description="Subtitle font size")
+    color: Optional[str] = Field(default=None, description="Subtitle color")
+
+
+# Main Chart Model
+class ChartArtifactModel(Model):
+    """AG-Charts configuration model."""
+
+    parent_id: ObjectId = Field(default=None, description="ID of the node registry")
+
+    # Chart data
+    data: List[Dict[str, Any]] = Field(default_factory=list, description="Chart data")
+
+    # Chart configuration
+    title: AGChartTitleConfig = Field(..., description="Chart title configuration")
+    # subtitle: AGChartSubtitleConfig = Field(..., description="Chart subtitle configuration")
+
+    # Series configuration
+    series: List[AGChartSeries] = Field(
+        default_factory=list, description="Chart series configurations"
+    )
+
+    # Axes configuration
+    axes: List[AGChartAxisConfig] = Field(
+        default_factory=list, description="Chart axes configurations"
+    )
+
+    # # Legend configuration
+    legend: AGChartLegendConfig = Field(
+        default=AGChartLegendConfig(), description="Legend configuration"
+    )
+    #
+    # Chart styling and behavior
+    width: int = Field(default=800, description="Chart width in pixels")
+    height: int = Field(default=400, description="Chart height in pixels")
+    padding: Optional[Dict[str, int]] = Field(default=None, description="Chart padding")
+    background: Optional[Dict[str, Any]] = Field(
+        default=None, description="Background configuration"
+    )
+
+    # Animation
+    animation: Optional[Dict[str, Any]] = Field(
+        default=None, description="Animation configuration"
+    )
+
+    # Tooltip
+    tooltip: Optional[Dict[str, Any]] = Field(
+        default=None, description="Global tooltip configuration"
+    )
+
+    # Theme
+    theme: Optional[str] = Field(default="ag-default", description="Chart theme")
+
+    # Additional options
+    options: Optional[Dict[str, Any]] = Field(
+        default_factory=dict, description="Additional chart options"
+    )
+
+
+# Helper functions for creating specific chart types
+def create_simple_line_chart(
+    data: List[Dict[str, Any]],
+    x_key: str,
+    y_key: str,
+    title: Optional[str] = None,
+    parent_id: ObjectId = None,
+) -> ChartArtifactModel:
+    """Create a simple line chart."""
+    chart_title = (
+        AGChartTitleConfig(text=title) if title else AGChartTitleConfig(text="Chart")
+    )
+    # subtitle = AGChartSubtitleConfig(text="Subtitle")
+
+    series = [
+        AGLineSeriesConfig(
+            type="line", xKey=x_key, yKey=y_key, title=y_key.title(), data=data
+        )
+    ]
+
+    axes = [
+        AGChartAxisConfig(type="number", position="bottom", title=x_key.title()),
+        AGChartAxisConfig(type="number", position="left", title=y_key.title()),
+    ]
+
+    return ChartArtifactModel(
+        parent_id=parent_id,
+        data=data,
+        title=chart_title,
+        # subtitle=subtitle,
+        series=series,
+        axes=axes,
+    )
+
+
+def create_simple_bar_chart(
+    data: List[Dict[str, Any]],
+    x_key: str,
+    y_key: str,
+    title: Optional[str] = None,
+    parent_id: Optional[ObjectId] = None,
+) -> ChartArtifactModel:
+    """Create a simple bar chart."""
+    chart_title = AGChartTitleConfig(text=title) if title else None
+
+    series = [
+        AGColumnSeriesConfig(type="column", xKey=x_key, yKey=y_key, title=y_key.title())
+    ]
+
+    axes = [
+        AGChartAxisConfig(type="category", position="bottom", title=x_key.title()),
+        AGChartAxisConfig(type="number", position="left", title=y_key.title()),
+    ]
+
+    return ChartArtifactModel(
+        parent_id=parent_id, data=data, title=chart_title, series=series, axes=axes
+    )
+
+
+def create_simple_pie_chart(
+    data: List[Dict[str, Any]],
+    angle_key: str,
+    label_key: str,
+    title: Optional[str] = None,
+    parent_id: Optional[ObjectId] = None,
+) -> ChartArtifactModel:
+    """Create a simple pie chart."""
+    chart_title = AGChartTitleConfig(text=title) if title else None
+
+    series = [
+        AGPieSeriesConfig(
+            type="pie", angleKey=angle_key, labelKey=label_key, title="Distribution"
+        )
+    ]
+
+    return ChartArtifactModel(
+        parent_id=parent_id, data=data, title=chart_title, series=series
+    )
+
+
+def create_simple_area_chart(
+    data: List[Dict[str, Any]],
+    x_key: str,
+    y_key: str,
+    title: Optional[str] = None,
+    parent_id: Optional[ObjectId] = None,
+) -> ChartArtifactModel:
+    """Create a simple area chart."""
+    chart_title = AGChartTitleConfig(text=title) if title else None
+
+    series = [
+        AGAreaSeriesConfig(type="area", xKey=x_key, yKey=y_key, title=y_key.title())
+    ]
+
+    axes = [
+        AGChartAxisConfig(type="category", position="bottom", title=x_key.title()),
+        AGChartAxisConfig(type="number", position="left", title=y_key.title()),
+    ]
+
+    return ChartArtifactModel(
+        parent_id=parent_id, data=data, title=chart_title, series=series, axes=axes
+    )
+
+
+def create_simple_scatter_chart(
+    data: List[Dict[str, Any]],
+    x_key: str,
+    y_key: str,
+    title: Optional[str] = None,
+    parent_id: Optional[ObjectId] = None,
+) -> ChartArtifactModel:
+    """Create a simple scatter chart."""
+    chart_title = AGChartTitleConfig(text=title) if title else None
+
+    series = [
+        AGScatterSeriesConfig(
+            type="scatter", xKey=x_key, yKey=y_key, title=y_key.title()
+        )
+    ]
+
+    axes = [
+        AGChartAxisConfig(type="number", position="bottom", title=x_key.title()),
+        AGChartAxisConfig(type="number", position="left", title=y_key.title()),
+    ]
+
+    return ChartArtifactModel(
+        parent_id=parent_id, data=data, title=chart_title, series=series, axes=axes
+    )
+
+
+def create_simple_donut_chart(
+    data: List[Dict[str, Any]],
+    angle_key: str,
+    label_key: str,
+    title: Optional[str] = None,
+    inner_radius_ratio: float = 0.6,
+    parent_id: Optional[ObjectId] = None,
+) -> ChartArtifactModel:
+    """Create a simple donut chart."""
+    chart_title = AGChartTitleConfig(text=title) if title else None
+
+    series = [
+        AGDonutSeriesConfig(
+            type="donut",
+            angleKey=angle_key,
+            labelKey=label_key,
+            innerRadiusRatio=inner_radius_ratio,
+            title="Distribution",
+        )
+    ]
+
+    return ChartArtifactModel(
+        parent_id=parent_id, data=data, title=chart_title, series=series
+    )