Refactored chart.Chart.get_required_modules()

Author: hcpchris

highcharts_maps/chart.py

@@ -1,10 +1,10 @@
-from typing import Optional
+from typing import Optional, List
 
 from validator_collection import validators, checkers
 
 from highcharts_core.chart import Chart as ChartBase
 
-from highcharts_maps import constants, errors, utility_functions
+from highcharts_maps import errors, utility_functions
 from highcharts_maps.options import HighchartsOptions, HighchartsMapsOptions
 from highcharts_maps.decorators import validate_types
 from highcharts_maps.js_literal_functions import serialize_to_js_literal
@@ -29,36 +29,6 @@ def __init__(self, **kwargs):
 
         super().__init__(**kwargs)
 
-    def _jupyter_include_scripts(self):
-        """Return the JavaScript code that is used to load the Highcharts JS libraries.
-
-        .. note::
-
-          Currently includes *all* `Highcharts JS <https://www.highcharts.com/>`_ modules
-          in the HTML. This issue will be addressed when roadmap issue :issue:`2` is
-          released.
-
-        :rtype: :class:`str <python:str>`
-        """
-        js_str = ''
-        if self.is_maps_chart:
-            for item in constants.MAPS_INCLUDE_LIBS:
-                js_str += utility_functions.jupyter_add_script(item)
-                js_str += """.then(() => {"""
-
-            for item in constants.MAPS_INCLUDE_LIBS:
-                js_str += """});"""
-
-        else:
-            for item in constants.INCLUDE_LIBS:
-                js_str += utility_functions.jupyter_add_script(item)
-                js_str += """.then(() => {"""
-
-            for item in constants.INCLUDE_LIBS:
-                js_str += """});"""
-
-        return js_str
-
     def _jupyter_javascript(self, 
                             global_options = None, 
                             container = None,
@@ -118,6 +88,25 @@ def _jupyter_javascript(self,
 
         return js_str
 
+    def get_required_modules(self, include_extension = False) -> List[str]:
+        """Return the list of URLs from which the Highcharts JavaScript modules
+        needed to render the chart can be retrieved.
+        
+        :param include_extension: if ``True``, will return script names with the 
+          ``'.js'`` extension included. Defaults to ``False``.
+        :type include_extension: :class:`bool <python:bool>`
+
+        :rtype: :class:`list <python:list>`
+        """
+        initial_scripts = ['highcharts']
+        if self.is_maps_chart:
+            initial_scripts.extend(['maps/modules/map',
+                                    'modules/exporting'])
+
+        scripts = self._process_required_modules(initial_scripts, include_extension)
+
+        return scripts
+
     @property
     def is_maps_chart(self) -> bool:
         """If ``True``, indicates that the chart should be rendered as a