Fixed missing TreegraphOptions.levels property.

Author: hcpchris

highcharts_core/options/plot_options/levels.py

@@ -10,6 +10,7 @@
 from highcharts_core.utility_classes.gradients import Gradient
 from highcharts_core.utility_classes.patterns import Pattern
 from highcharts_core.utility_classes.states import States
+from highcharts_core.utility_classes.markers import Marker
 
 
 class BaseLevelOptions(HighchartsMeta):
@@ -617,3 +618,29 @@ def _to_untrimmed_dict(self, in_cls = None) -> dict:
             untrimmed[key] = parent_as_dict[key]
 
         return untrimmed
+
+
+class TreegraphLevelOptions(TreemapLevelOptions):
+    """Set options on specific levels for Treegraph Charts. Takes precedence over series
+    options, but not points."""
+    
+    def __init__(self, **kwargs):
+        self._marker = None
+        
+        self.marker = kwargs.get('marker', None)
+        
+        super().__init__(**kwargs)
+        
+    @property
+    def marker(self) -> Optional[Marker]:
+        """Set marker options for nodes at the level.
+        
+        :rtype: :class:`Marker <highcharts_core.utility_classes.markers.Marker>` or 
+          :obj:`None <python:None>`
+        """
+        return self._marker
+    
+    @marker.setter
+    @class_sensitive(Marker)
+    def marker(self, value):
+        self._marker = value

highcharts_core/options/plot_options/treegraph.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Optional
+from typing import Optional, List
 from decimal import Decimal
 
 from validator_collection import validators
@@ -11,6 +11,7 @@
 from highcharts_core.utility_classes.events import SeriesEvents
 from highcharts_core.utility_classes.javascript_functions import CallbackFunction
 from highcharts_core.options.plot_options.link import LinkOptions
+from highcharts_core.options.plot_options.levels import TreegraphLevelOptions
 
 
 class TreegraphEvents(SeriesEvents):
@@ -120,6 +121,8 @@ def __init__(self, **kwargs):
         self._reversed = None        
         self._traverse_up_button = None
 
+        self._levels = None
+        
         self.animation_limit = kwargs.get('animation_limit', None)
         self.boost_blending = kwargs.get('boost_blending', None)
         self.boost_threshold = kwargs.get('boost_threshold', None)
@@ -144,6 +147,8 @@ def __init__(self, **kwargs):
         self.link = kwargs.get('link', None)
         self.reversed = kwargs.get('reversed', None)
 
+        self.levels = kwargs.get('levels', None)
+        
         super().__init__(**kwargs)
 
     @property
@@ -626,7 +631,25 @@ def reversed(self, value):
             self._reversed = None
         else:
             self._reversed = bool(value)
-            
+
+    @property
+    def levels(self) -> Optional[List[TreegraphLevelOptions]]:
+        """Set options on specific levels. 
+        
+        .. note::
+        
+          Takes precedence over series options, but not point options.
+          
+        :rtype: :class:`TreemapLevelOptions <highcharts_core.options.plot_options.levels.TreemapLevelOptions>` 
+          or :obj:`None <python:None>`
+        """
+        return self._levels
+    
+    @levels.setter
+    @class_sensitive(TreegraphLevelOptions)
+    def levels(self, value):
+        self._levels = value
+        
     @classmethod
     def _get_kwargs_from_dict(cls, as_dict):
         """Convenience method which returns the keyword arguments used to initialize the
@@ -699,6 +722,7 @@ class from a Highcharts Javascript-compatible :class:`dict <python:dict>` object
             'fill_space': as_dict.get('fillSpace', None),
             'link': as_dict.get('link', None),
             'reversed': as_dict.get('reversed', None),
+            'levels': as_dict.get('levels', None),
         }
 
         return kwargs
@@ -762,6 +786,7 @@ def _to_untrimmed_dict(self, in_cls = None) -> dict:
             'fillSpace': self.fill_space,
             'link': self.link,
             'reversed': self.reversed,
+            'levels': self.levels,
         }
 
         return untrimmed

highcharts_core/options/series/treegraph.py

@@ -142,6 +142,7 @@ def _get_kwargs_from_dict(cls, as_dict):
             'fill_space': as_dict.get('fillSpace', None),
             'link': as_dict.get('link', None),
             'reversed': as_dict.get('reversed', None),
+            'levels': as_dict.get('levels', None),
 
             'data': as_dict.get('data', None),
             'id': as_dict.get('id', None),