Implemented solution for GanttSeries.from_pandas() bug. Closes #46.

Author: hcpchris

highcharts_gantt/options/series/data/gantt.py

@@ -11,7 +11,7 @@
 from highcharts_gantt.metaclasses import HighchartsMeta
 from highcharts_gantt.options.series.data.connect import DataConnection
 from highcharts_gantt.options.series.data.collections import DataPointCollection
-from highcharts_gantt.utility_functions import validate_color, parse_jira_issue
+from highcharts_gantt.utility_functions import validate_color, parse_jira_issue, datetime64_to_datetime
 from highcharts_gantt.utility_classes.gradients import Gradient
 from highcharts_gantt.utility_classes.patterns import Pattern
 
@@ -218,7 +218,13 @@ def end(self, value):
         elif checkers.is_date(value):
             self._end = validators.date(value, allow_empty = True)
         else:
-            self._end = validators.datetime(value, allow_empty = True, coerce_value = True)
+            try:
+                self._end = validators.datetime(value, allow_empty = True, coerce_value = True)
+            except (ValueError, TypeError) as error:
+                if checkers.is_type(value, 'datetime64'):
+                    self._end = datetime64_to_datetime(value)
+                else:
+                    raise error
 
     @property
     def milestone(self) -> Optional[bool]:
@@ -279,7 +285,13 @@ def start(self, value):
         elif checkers.is_date(value):
             self._start = validators.date(value, allow_empty = True)
         else:
-            self._start = validators.datetime(value, allow_empty = True, coerce_value = True)
+            try:
+                self._start = validators.datetime(value, allow_empty = True, coerce_value = True)
+            except (ValueError, TypeError) as error:
+                if checkers.is_type(value, 'datetime64'):
+                    self._start = datetime64_to_datetime(value)
+                else:
+                    raise error
 
     @property
     def y(self) -> Optional[int | float | Decimal]:

highcharts_gantt/options/series/gantt.py

@@ -1129,13 +1129,13 @@ def _data_collection_class(cls):
         :rtype: :class:`DataPointCollection <highcharts_core.options.series.data.collections.DataPointCollection>`
           descendent
         """
-        return GanttData
-    
+        return GanttDataCollection
+
     @classmethod
     def _data_point_class(cls):
         """Returns the class object used for individual data points.
         
         :rtype: :class:`DataBase <highcharts_core.options.series.data.base.DataBase>` 
           descendent
         """
-        return GanttDataCollection
+        return GanttData

tests/input_files/series/base/bugfix162.csv

@@ -0,0 +1,5 @@
+task_id,name,start,end,parent,dependency
+1,DoThing1,2023-01-01,2023-02-01,,
+2,DoThing1-sub1,2023-01-11,2023-01-14,1,
+3,DoThing1-sub2,2023-01-01,2023-02-01,1,2
+4,DoThing2,2023-02-02,2023-03-01,,

tests/options/series/test_base.py

@@ -655,3 +655,25 @@ def test_to_dict(kwargs, error):
 ])
 def test_from_js_literal(input_files, filename, as_file, error):
     Class_from_js_literal(cls, input_files, filename, as_file, error)
+
+def test_bugfix162(input_files):
+    from highcharts_gantt.chart import Chart
+    from highcharts_gantt.options.series.gantt import GanttSeries
+
+    import pandas as pd
+    from collections import defaultdict
+
+    filename = "series/base/bugfix162.csv"
+    input_file = check_input_file(input_files, filename)
+
+    dtype = defaultdict(lambda: "str")
+
+    df = pd.read_csv(input_file, dtype=dtype, parse_dates=["start", "end"])
+
+    # Creating a Series from the DataFrame
+    my_series = GanttSeries.from_pandas(
+        df,
+        property_map={"id": "task_id", "name": "name", "start": "start", "end": "end"},
+    )
+
+    assert my_series is not None