b8b3f8910e
* add more type annotations through the code * add typing in reflex/utils * misc typing * more typings * state typing * keep typing * typing init and utils * more typing for components * fix attempt for 3.9 * need more __future * more typings * type event plz * type model * type vars/base.py * enable 'ANN001' for reflex folder (ignore tests and benchmarks) * fix pyi * add missing annotations * use more precise error when ignoring
524 lines
18 KiB
Python
524 lines
18 KiB
Python
"""A module that defines the chart components in Recharts."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, List, Union
|
|
|
|
from reflex.components.component import Component
|
|
from reflex.components.recharts.general import ResponsiveContainer
|
|
from reflex.constants import EventTriggers
|
|
from reflex.constants.colors import Color
|
|
from reflex.event import EventHandler, no_args_event_spec
|
|
from reflex.vars.base import Var
|
|
|
|
from .recharts import (
|
|
LiteralAnimationEasing,
|
|
LiteralComposedChartBaseValue,
|
|
LiteralLayout,
|
|
LiteralStackOffset,
|
|
LiteralSyncMethod,
|
|
RechartsCharts,
|
|
)
|
|
|
|
|
|
class ChartBase(RechartsCharts):
|
|
"""A component that wraps a Recharts charts."""
|
|
|
|
# The width of chart container. String or Integer
|
|
width: Var[Union[str, int]] = Var.create("100%")
|
|
|
|
# The height of chart container.
|
|
height: Var[Union[str, int]] = Var.create("100%")
|
|
|
|
# The customized event handler of click on the component in this chart
|
|
on_click: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of mouseenter on the component in this chart
|
|
on_mouse_enter: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of mousemove on the component in this chart
|
|
on_mouse_move: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of mouseleave on the component in this chart
|
|
on_mouse_leave: EventHandler[no_args_event_spec]
|
|
|
|
@staticmethod
|
|
def _ensure_valid_dimension(name: str, value: Any) -> None:
|
|
"""Ensure that the value is an int type or str percentage.
|
|
|
|
Unfortunately str Vars cannot be checked and are implicitly not allowed.
|
|
|
|
Args:
|
|
name: The name of the prop.
|
|
value: The value to check.
|
|
|
|
Raises:
|
|
ValueError: If the value is not an int type or str percentage.
|
|
"""
|
|
if value is None:
|
|
return
|
|
if isinstance(value, int):
|
|
return
|
|
if isinstance(value, str) and value.endswith("%"):
|
|
return
|
|
if isinstance(value, Var) and issubclass(value._var_type, int):
|
|
return
|
|
raise ValueError(
|
|
f"Chart {name} must be specified as int pixels or percentage, not {value!r}. "
|
|
"CSS unit dimensions are allowed on parent container."
|
|
)
|
|
|
|
@classmethod
|
|
def create(cls, *children: Any, **props: Any) -> Component:
|
|
"""Create a chart component.
|
|
|
|
Args:
|
|
*children: The children of the chart component.
|
|
**props: The properties of the chart component.
|
|
|
|
Returns:
|
|
The chart component wrapped in a responsive container.
|
|
"""
|
|
width = props.pop("width", None)
|
|
height = props.pop("height", None)
|
|
cls._ensure_valid_dimension("width", width)
|
|
cls._ensure_valid_dimension("height", height)
|
|
|
|
# Ensure that the min_height and min_width are set to prevent the chart from collapsing.
|
|
# We are using small values so that height and width can still be used over min_height and min_width.
|
|
# Without this, sometimes the chart will not be visible. Causing confusion to the user.
|
|
# With this, the user will see a small chart and can adjust the height and width and can figure out that the issue is with the size.
|
|
min_height = props.pop("min_height", 10)
|
|
min_width = props.pop("min_width", 10)
|
|
|
|
return ResponsiveContainer.create(
|
|
super().create(*children, **props),
|
|
width=width if width is not None else "100%",
|
|
height=height if height is not None else "100%",
|
|
min_width=min_width,
|
|
min_height=min_height,
|
|
)
|
|
|
|
|
|
class CategoricalChartBase(ChartBase):
|
|
"""A component that wraps a Categorical Recharts charts."""
|
|
|
|
# The source data, in which each element is an object.
|
|
data: Var[List[Dict[str, Any]]]
|
|
|
|
# The sizes of whitespace around the chart, i.e. {"top": 50, "right": 30, "left": 20, "bottom": 5}.
|
|
margin: Var[Dict[str, Any]]
|
|
|
|
# If any two categorical charts(rx.line_chart, rx.area_chart, rx.bar_chart, rx.composed_chart) have the same sync_id, these two charts can sync the position GraphingTooltip, and the start_index, end_index of Brush.
|
|
sync_id: Var[str]
|
|
|
|
# When sync_id is provided, allows customisation of how the charts will synchronize GraphingTooltips and brushes. Using 'index' (default setting), other charts will reuse current datum's index within the data array. In cases where data does not have the same length, this might yield unexpected results. In that case use 'value' which will try to match other charts values, or a fully custom function which will receive tick, data as argument and should return an index. 'index' | 'value' | function. Default: "index"
|
|
sync_method: Var[LiteralSyncMethod]
|
|
|
|
# The layout of area in the chart. 'horizontal' | 'vertical'. Default: "horizontal"
|
|
layout: Var[LiteralLayout]
|
|
|
|
# The type of offset function used to generate the lower and upper values in the series array. The four types are built-in offsets in d3-shape. 'expand' | 'none' | 'wiggle' | 'silhouette'
|
|
stack_offset: Var[LiteralStackOffset]
|
|
|
|
|
|
class AreaChart(CategoricalChartBase):
|
|
"""An Area chart component in Recharts."""
|
|
|
|
tag = "AreaChart"
|
|
|
|
alias = "RechartsAreaChart"
|
|
|
|
# The base value of area. Number | 'dataMin' | 'dataMax' | 'auto'. Default: "auto"
|
|
base_value: Var[Union[int, LiteralComposedChartBaseValue]]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"XAxis",
|
|
"YAxis",
|
|
"ReferenceArea",
|
|
"ReferenceDot",
|
|
"ReferenceLine",
|
|
"Brush",
|
|
"CartesianGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Area",
|
|
"Defs",
|
|
]
|
|
|
|
|
|
class BarChart(CategoricalChartBase):
|
|
"""A Bar chart component in Recharts."""
|
|
|
|
tag = "BarChart"
|
|
|
|
alias = "RechartsBarChart"
|
|
|
|
# The gap between two bar categories, which can be a percent value or a fixed value. Percentage | Number. Default: "10%"
|
|
bar_category_gap: Var[Union[str, int]]
|
|
|
|
# The gap between two bars in the same category, which can be a percent value or a fixed value. Percentage | Number. Default: 4
|
|
bar_gap: Var[Union[str, int]]
|
|
|
|
# The width of all the bars in the chart. Number
|
|
bar_size: Var[int]
|
|
|
|
# The maximum width of all the bars in a horizontal BarChart, or maximum height in a vertical BarChart.
|
|
max_bar_size: Var[int]
|
|
|
|
# The type of offset function used to generate the lower and upper values in the series array. The four types are built-in offsets in d3-shape. Default: "none"
|
|
stack_offset: Var[LiteralStackOffset]
|
|
|
|
# If false set, stacked items will be rendered left to right. If true set, stacked items will be rendered right to left. (Render direction affects SVG layering, not x position.) Default: False
|
|
reverse_stack_order: Var[bool]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"XAxis",
|
|
"YAxis",
|
|
"ReferenceArea",
|
|
"ReferenceDot",
|
|
"ReferenceLine",
|
|
"Brush",
|
|
"CartesianGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Bar",
|
|
]
|
|
|
|
|
|
class LineChart(CategoricalChartBase):
|
|
"""A Line chart component in Recharts."""
|
|
|
|
tag = "LineChart"
|
|
|
|
alias = "RechartsLineChart"
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"XAxis",
|
|
"YAxis",
|
|
"ReferenceArea",
|
|
"ReferenceDot",
|
|
"ReferenceLine",
|
|
"Brush",
|
|
"CartesianGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Line",
|
|
]
|
|
|
|
|
|
class ComposedChart(CategoricalChartBase):
|
|
"""A Composed chart component in Recharts."""
|
|
|
|
tag = "ComposedChart"
|
|
|
|
alias = "RechartsComposedChart"
|
|
|
|
# The base value of area. Number | 'dataMin' | 'dataMax' | 'auto'. Default: "auto"
|
|
base_value: Var[Union[int, LiteralComposedChartBaseValue]]
|
|
|
|
# The gap between two bar categories, which can be a percent value or a fixed value. Percentage | Number. Default: "10%"
|
|
bar_category_gap: Var[Union[str, int]]
|
|
|
|
# The gap between two bars in the same category. Default: 4
|
|
bar_gap: Var[int]
|
|
|
|
# The width or height of each bar. If the barSize is not specified, the size of the bar will be calculated by the barCategoryGap, barGap and the quantity of bar groups.
|
|
bar_size: Var[int]
|
|
|
|
# If false set, stacked items will be rendered left to right. If true set, stacked items will be rendered right to left. (Render direction affects SVG layering, not x position). Default: False
|
|
reverse_stack_order: Var[bool]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"XAxis",
|
|
"YAxis",
|
|
"ReferenceArea",
|
|
"ReferenceDot",
|
|
"ReferenceLine",
|
|
"Brush",
|
|
"CartesianGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Area",
|
|
"Line",
|
|
"Bar",
|
|
]
|
|
|
|
|
|
class PieChart(ChartBase):
|
|
"""A Pie chart component in Recharts."""
|
|
|
|
tag = "PieChart"
|
|
|
|
alias = "RechartsPieChart"
|
|
|
|
# The sizes of whitespace around the chart, i.e. {"top": 50, "right": 30, "left": 20, "bottom": 5}.
|
|
margin: Var[Dict[str, Any]]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"PolarAngleAxis",
|
|
"PolarRadiusAxis",
|
|
"PolarGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Pie",
|
|
]
|
|
|
|
# The customized event handler of mousedown on the sectors in this group
|
|
on_mouse_down: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of mouseup on the sectors in this group
|
|
on_mouse_up: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of mouseover on the sectors in this group
|
|
on_mouse_over: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of mouseout on the sectors in this group
|
|
on_mouse_out: EventHandler[no_args_event_spec]
|
|
|
|
|
|
class RadarChart(ChartBase):
|
|
"""A Radar chart component in Recharts."""
|
|
|
|
tag = "RadarChart"
|
|
|
|
alias = "RechartsRadarChart"
|
|
|
|
# The source data, in which each element is an object.
|
|
data: Var[List[Dict[str, Any]]]
|
|
|
|
# The sizes of whitespace around the chart, i.e. {"top": 50, "right": 30, "left": 20, "bottom": 5}. Default: {"top": 0, "right": 0, "left": 0, "bottom": 0}
|
|
margin: Var[Dict[str, Any]]
|
|
|
|
# The The x-coordinate of center. If set a percentage, the final value is obtained by multiplying the percentage of width. Number | Percentage. Default: "50%"
|
|
cx: Var[Union[int, str]]
|
|
|
|
# The The y-coordinate of center. If set a percentage, the final value is obtained by multiplying the percentage of height. Number | Percentage. Default: "50%"
|
|
cy: Var[Union[int, str]]
|
|
|
|
# The angle of first radial direction line. Default: 90
|
|
start_angle: Var[int]
|
|
|
|
# The angle of last point in the circle which should be startAngle - 360 or startAngle + 360. We'll calculate the direction of chart by 'startAngle' and 'endAngle'. Default: -270
|
|
end_angle: Var[int]
|
|
|
|
# The inner radius of first circle grid. If set a percentage, the final value is obtained by multiplying the percentage of maxRadius which is calculated by the width, height, cx, cy. Number | Percentage. Default: 0
|
|
inner_radius: Var[Union[int, str]]
|
|
|
|
# The outer radius of last circle grid. If set a percentage, the final value is obtained by multiplying the percentage of maxRadius which is calculated by the width, height, cx, cy. Number | Percentage. Default: "80%"
|
|
outer_radius: Var[Union[int, str]]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"PolarAngleAxis",
|
|
"PolarRadiusAxis",
|
|
"PolarGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Radar",
|
|
]
|
|
|
|
def get_event_triggers(self) -> dict[str, Union[Var, Any]]:
|
|
"""Get the event triggers that pass the component's value to the handler.
|
|
|
|
Returns:
|
|
A dict mapping the event trigger to the var that is passed to the handler.
|
|
"""
|
|
return {
|
|
EventTriggers.ON_CLICK: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_ENTER: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_LEAVE: no_args_event_spec,
|
|
}
|
|
|
|
|
|
class RadialBarChart(ChartBase):
|
|
"""A RadialBar chart component in Recharts."""
|
|
|
|
tag = "RadialBarChart"
|
|
|
|
alias = "RechartsRadialBarChart"
|
|
|
|
# The source data which each element is an object.
|
|
data: Var[List[Dict[str, Any]]]
|
|
|
|
# The sizes of whitespace around the chart. Default: {"top": 5, "right": 5, "left": 5 "bottom": 5}
|
|
margin: Var[Dict[str, Any]]
|
|
|
|
# The The x-coordinate of center. If set a percentage, the final value is obtained by multiplying the percentage of width. Number | Percentage. Default: "50%"
|
|
cx: Var[Union[int, str]]
|
|
|
|
# The The y-coordinate of center. If set a percentage, the final value is obtained by multiplying the percentage of height. Number | Percentage. Default: "50%"
|
|
cy: Var[Union[int, str]]
|
|
|
|
# The angle of first radial direction line. Default: 0
|
|
start_angle: Var[int]
|
|
|
|
# The angle of last point in the circle which should be startAngle - 360 or startAngle + 360. We'll calculate the direction of chart by 'startAngle' and 'endAngle'. Default: 360
|
|
end_angle: Var[int]
|
|
|
|
# The inner radius of first circle grid. If set a percentage, the final value is obtained by multiplying the percentage of maxRadius which is calculated by the width, height, cx, cy. Number | Percentage. Default: "30%"
|
|
inner_radius: Var[Union[int, str]]
|
|
|
|
# The outer radius of last circle grid. If set a percentage, the final value is obtained by multiplying the percentage of maxRadius which is calculated by the width, height, cx, cy. Number | Percentage. Default: "100%"
|
|
outer_radius: Var[Union[int, str]]
|
|
|
|
# The gap between two bar categories, which can be a percent value or a fixed value. Percentage | Number. Default: "10%"
|
|
bar_category_gap: Var[Union[int, str]]
|
|
|
|
# The gap between two bars in the same category, which can be a percent value or a fixed value. Percentage | Number. Default: 4
|
|
bar_gap: Var[str]
|
|
|
|
# The size of each bar. If the barSize is not specified, the size of bar will be calculated by the barCategoryGap, barGap and the quantity of bar groups.
|
|
bar_size: Var[int]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"PolarAngleAxis",
|
|
"PolarRadiusAxis",
|
|
"PolarGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"RadialBar",
|
|
]
|
|
|
|
|
|
class ScatterChart(ChartBase):
|
|
"""A Scatter chart component in Recharts."""
|
|
|
|
tag = "ScatterChart"
|
|
|
|
alias = "RechartsScatterChart"
|
|
|
|
# The sizes of whitespace around the chart. Default: {"top": 5, "right": 5, "bottom": 5, "left": 5}
|
|
margin: Var[Dict[str, Any]]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = [
|
|
"XAxis",
|
|
"YAxis",
|
|
"ZAxis",
|
|
"ReferenceArea",
|
|
"ReferenceDot",
|
|
"ReferenceLine",
|
|
"Brush",
|
|
"CartesianGrid",
|
|
"Legend",
|
|
"GraphingTooltip",
|
|
"Scatter",
|
|
]
|
|
|
|
def get_event_triggers(self) -> dict[str, Union[Var, Any]]:
|
|
"""Get the event triggers that pass the component's value to the handler.
|
|
|
|
Returns:
|
|
A dict mapping the event trigger to the var that is passed to the handler.
|
|
"""
|
|
return {
|
|
EventTriggers.ON_CLICK: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_DOWN: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_UP: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_MOVE: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_OVER: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_OUT: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_ENTER: no_args_event_spec,
|
|
EventTriggers.ON_MOUSE_LEAVE: no_args_event_spec,
|
|
}
|
|
|
|
|
|
class FunnelChart(ChartBase):
|
|
"""A Funnel chart component in Recharts."""
|
|
|
|
tag = "FunnelChart"
|
|
|
|
alias = "RechartsFunnelChart"
|
|
|
|
# The layout of bars in the chart. Default: "centric"
|
|
layout: Var[str]
|
|
|
|
# The sizes of whitespace around the chart. Default: {"top": 5, "right": 5, "bottom": 5, "left": 5}
|
|
margin: Var[Dict[str, Any]]
|
|
|
|
# The stroke color of each bar. String | Object
|
|
stroke: Var[Union[str, Color]]
|
|
|
|
# Valid children components
|
|
_valid_children: List[str] = ["Legend", "GraphingTooltip", "Funnel"]
|
|
|
|
|
|
class Treemap(RechartsCharts):
|
|
"""A Treemap chart component in Recharts."""
|
|
|
|
tag = "Treemap"
|
|
|
|
alias = "RechartsTreemap"
|
|
|
|
# The width of chart container. String or Integer. Default: "100%"
|
|
width: Var[Union[str, int]] = Var.create("100%")
|
|
|
|
# The height of chart container. String or Integer. Default: "100%"
|
|
height: Var[Union[str, int]] = Var.create("100%")
|
|
|
|
# data of treemap. Array
|
|
data: Var[List[Dict[str, Any]]]
|
|
|
|
# The key of a group of data which should be unique in a treemap. String | Number. Default: "value"
|
|
data_key: Var[Union[str, int]]
|
|
|
|
# The key of each sector's name. String. Default: "name"
|
|
name_key: Var[str]
|
|
|
|
# The treemap will try to keep every single rectangle's aspect ratio near the aspectRatio given. Number
|
|
aspect_ratio: Var[int]
|
|
|
|
# If set false, animation of area will be disabled. Default: True
|
|
is_animation_active: Var[bool]
|
|
|
|
# Specifies when the animation should begin, the unit of this option is ms. Default: 0
|
|
animation_begin: Var[int]
|
|
|
|
# Specifies the duration of animation, the unit of this option is ms. Default: 1500
|
|
animation_duration: Var[int]
|
|
|
|
# The type of easing function. 'ease' | 'ease-in' | 'ease-out' | 'ease-in-out' | 'linear'. Default: "ease"
|
|
animation_easing: Var[LiteralAnimationEasing]
|
|
|
|
# The customized event handler of animation start
|
|
on_animation_start: EventHandler[no_args_event_spec]
|
|
|
|
# The customized event handler of animation end
|
|
on_animation_end: EventHandler[no_args_event_spec]
|
|
|
|
@classmethod
|
|
def create(cls, *children, **props) -> Component:
|
|
"""Create a chart component.
|
|
|
|
Args:
|
|
*children: The children of the chart component.
|
|
**props: The properties of the chart component.
|
|
|
|
Returns:
|
|
The Treemap component wrapped in a responsive container.
|
|
"""
|
|
return ResponsiveContainer.create(
|
|
super().create(*children, **props),
|
|
width=props.pop("width", "100%"),
|
|
height=props.pop("height", "100%"),
|
|
)
|
|
|
|
|
|
area_chart = AreaChart.create
|
|
bar_chart = BarChart.create
|
|
line_chart = LineChart.create
|
|
composed_chart = ComposedChart.create
|
|
pie_chart = PieChart.create
|
|
radar_chart = RadarChart.create
|
|
radial_bar_chart = RadialBarChart.create
|
|
scatter_chart = ScatterChart.create
|
|
funnel_chart = FunnelChart.create
|
|
treemap = Treemap.create
|