altair theme helpers

Author: gordonwoodhull

quarto/__init__.py

@@ -6,4 +6,4 @@
 from quarto.quarto import path
 from quarto.render import render
 from quarto.metadata import metadata
-from quarto.theme import plotly_theme_colors, plotly_theme_brand
+from quarto.theme import plotly_theme_colors, plotly_theme_brand, altair_theme_colors, altair_theme_brand

quarto/theme.py

@@ -1,8 +1,8 @@
 import yaml
-import plotly.graph_objects as go
 from functools import partial
 
 def plotly_theme_colors(bg, fg):
+  import plotly.graph_objects as go
   return go.layout.Template({'layout': {
     'paper_bgcolor': bg,
     'plot_bgcolor': bg,
@@ -12,3 +12,26 @@ def plotly_theme_colors(bg, fg):
 def plotly_theme_brand(brand_yml):
   brand = yaml.safe_load(open(brand_yml).read())
   return plotly_theme_colors(brand["color"]["background"], brand["color"]["foreground"])
+
+def altair_theme_colors(bg, fg):
+  return {
+    'config': {
+      'view': {'stroke': 'transparent'},
+      'axis': {
+        'domainColor': fg,
+        'labelColor': fg,
+        'titleColor': fg,
+      },
+      'legend': {
+        'labelColor': fg,
+        'titleColor': fg,
+      },
+      'background': bg,  # Background color
+    }
+  }
+
+def altair_theme_brand(brand_yml):
+    brand = yaml.safe_load(open(brand_yml).read())
+    bg = brand["color"]["background"]
+    fg = brand["color"]["foreground"]
+    return partial(altair_theme_colors, bg, fg)