🎨 support time string for TimeAttribute. #43

perillaroc · Sep 18, 2022 · 856c50b · 856c50b
1 parent 0500e9d
commit 856c50b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/takler/core/node.py b/takler/core/node.py
@@ -796,7 +796,7 @@ def add_repeat(self, r: RepeatBase):
 
     # Time Attribute -----------------------------------------------------------
 
-    def add_time(self, time: datetime.time) -> TimeAttribute:
+    def add_time(self, time: Union[datetime.time, str]) -> TimeAttribute:
         """
         Add a ``TimeAttribute`` to Node. Node can have multiply time attributes.
 

diff --git a/takler/core/time_attr.py b/takler/core/time_attr.py
@@ -1,4 +1,5 @@
 import datetime
+from typing import Union
 
 from .calendar import Calendar
 
@@ -16,7 +17,9 @@ class TimeAttribute:
     free
         if marked true, time attributes is ignored.
     """
-    def __init__(self, time: datetime.time):
+    def __init__(self, time: Union[datetime.time, str]):
+        if isinstance(time, str):
+            time = datetime.datetime.strptime(time, '%H:%M').time()
         self.time: datetime.time = time
         self.free: bool = False