Skip to content

Commit

Permalink
fix(vega): Wrap vega module from pywebvue
Browse files Browse the repository at this point in the history
  • Loading branch information
DrewLazzeriKitware committed Oct 4, 2021
1 parent 7f97d49 commit 222ca66
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions trame/vegaEmbed/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
from trame import get_app_instance
from trame.html import AbstractElement

from pywebvue.modules import VegaEmbed

# Make sure used module is available
_app = get_app_instance()
_app.enableModule(VegaEmbed)


class VegaEmbed(AbstractElement):
@staticmethod
def altair_to_spec(chart):
return chart.to_dict()

def __init__(self, chart=None, name=None, **kwargs):
super().__init__("VegaEmbed", **kwargs)
self._chart = chart
self._name = name or f"chart_{self._id}"
self._attributes["spec"] = f':spec="{self._name}"'

def update(self, chart=None):
if chart:
self._chart = chart
if self._chart:
_app = get_app_instance()
_app.set(self._name, VegaEmbed.altair_to_spec(self._chart))

0 comments on commit 222ca66

Please sign in to comment.