fix: Make day_of_week and day_of_year components be recognized by xar…

…ray. Refs #37 🔧
ZGIS · Jan 17, 2024 · 4676dd5 · 4676dd5
1 parent 0e51d98
commit 4676dd5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/semantique/components.py b/semantique/components.py
@@ -31,14 +31,14 @@ class time():
   DAY = "day"
   """:obj:`str` : Component storing the day number of a timestamp."""
 
-  DAY_OF_WEEK = "day_of_week"
+  DAY_OF_WEEK = "dayofweek"
   """:obj:`str` : Component storing the day of the week of a timestamp.
 
   The first day of the week is considered Monday, with an index of 0.
 
   """
 
-  DAY_OF_YEAR = "day_of_year"
+  DAY_OF_YEAR = "dayofyear"
   """:obj:`str` : Component storing the day of the year of a timestamp."""
 
   HOUR = "hour"

diff --git a/semantique/processor/arrays.py b/semantique/processor/arrays.py
@@ -254,6 +254,14 @@ def _extract_time(obj, component = None):
       try:
         out = out[component]
       except KeyError:
+        aliases = {
+          "day_of_week": "dayofweek",
+          "day_of_year": "dayofyear"
+        }
+        try:
+          component = aliases[component]
+        except KeyError:
+          pass
         try:
           out = getattr(out.dt, component)
         except AttributeError: