refactor with strategy pattern

twf.py

@@ -9,26 +9,60 @@ import rich
 from rich.console import Console
 from rich.columns import Columns
 
-def group_by(tasks, field):
+# class make:
-    """Group tasks by field values."""
+#     def carder(card, **kwargs):
-    groups = {}
+#         def f(task):
-    for task in tasks:
+#             return card(task, **kwargs)
-        if field in task:
+#         return f
-            if task[field] in groups:
+
-                groups[ task[field] ].append(task)
+#     def stacker(stack, **kwargs):
-            else:
+#         def f(tasks):
-                groups[ task[field] ] = [task]
+#             return stack(tasks, **kwargs)
-    return groups
+#         return f
+
+#     def sectioner(section, **kwargs):
+#         def f(tasks, field, values):
+#             return section(tasks, field, values, **kwargs)
+#         return f
+
+class Widget:
+    def __init__(self, order, group = None):
+        self.order = order
+        self.group = group
+
+class Tasker(Widget):
+    def __init__(self, show_only, order, group = None):
+        super().__init__(order, group)
+        self.show_only = show_only
+
+    def __call__(self, task):
+        raise NotImplementedError
+
+class Stacker(Widget):
+    def __init__(self, tasker, order, group):
+        super().__init__(order, group)
+        self.tasker = tasker
+
+    def __call__(self, tasks):
+        raise NotImplementedError
+
+class Sectioner(Widget):
+    def __init__(self, stacker, order, group):
+        super().__init__(order, group)
+        self.stacker = stacker
+
+    def __call__(self, tasks):
+        raise NotImplementedError
 
 
-##### Widgets #####
+class Card(Tasker):
+    def __init__(self, show_only, order = None):
+        super().__init__(show_only, order, group = None)
 
-def card(task, in_widget = None, show_only = None):
+    def __call__(self, task):
-    """Widget being a single task."""
+        if not self.show_only:
-
-    if not show_only:
             # Show all existing fields.
-        show_only = task.keys()
+            self.show_only = task.keys()
 
         sid = str(task["id"])
         if ':' in task["description"]:
@@ -39,7 +73,7 @@ def card(task, in_widget = None, show_only = None):
             title = sid
 
         segments = []
-    for key in show_only:
+        for key in self.show_only:
             if key in task.keys() and key not in ["id", "description"]:
                 val = task[key]
                 segment = f"{key}: "
@@ -54,29 +88,144 @@ def card(task, in_widget = None, show_only = None):
         cols = Columns(segments)
         grp = rich.console.Group(desc.strip(), cols)
         panel = rich.panel.Panel(grp, title = title,
-        title_align="left", expand = True, padding = (0,1))
+            title_align="left", expand = False, padding = (0,1))
 
         return panel
 
+class VerticalStack(Stacker):
+    def __init__(self, tasker):
+        super().__init__(tasker, order = None, group = None)
 
-def stack(tasks, in_widget = None, show_only = None):
+    def __call__(self, tasks):
-    """Widget being a stack of tasks widgets."""
         stack = rich.table.Table()
         stack.add_column("Tasks")
         for task in tasks:
-       stack.add_row( card(task, in_widget, show_only) )
+           stack.add_row( self.tasker(task) )
         return stack
 
+class VerticalSections(Sectioner):
+    def __init__(self, stacker, order, group):
+        super().__init__(stacker, order, group)
 
-def sections(tasks, field, values, in_widget = None, show_only = None):
+    def __call__(self, tasks):
-    """Widget being a panel of stack widgets."""
         sections = []
-    groups = group_by(tasks, field)
+        groups = self.group(tasks)
-    for val in values:
+        for val in self.order():
             if val in groups:
-            sections.append( rich.panel.Panel(stack(groups[val], in_widget, show_only), title = val) )
+                sections.append( rich.panel.Panel(self.stacker(groups[val]), title = val) )
         return rich.console.Group(*sections)
 
+class SectionSorter:
+    def __call__(self):
+        raise NotImplementedError
+
+class OnValues(SectionSorter):
+    def __init__(self, values):
+        self.values = values
+
+    def __call__(self):
+        return self.values
+
+class Grouper:
+    """Group tasks by field values."""
+    def __init__(self, field):
+        self.field = field
+
+    def __call__(self, tasks):
+        groups = {}
+        for task in tasks:
+            if self.field in task:
+                if task[self.field] in groups:
+                    groups[ task[self.field] ].append(task)
+                else:
+                    groups[ task[self.field] ] = [task]
+        return groups
+
+class TasksSorter:
+    def make(self, tasks, field, reverse = False):
+        return sorted(tasks, key = lambda t: t[field], reverse = reverse)
+
+
+# class layout:
+#     def __init__(self, carder, stacker, sectioner)
+#         self.carder = carder
+#         self.stacker = stacker
+#         self.sectioner = sectioner
+
+#     def __call__(self, sec_group, sec_sort, sub_group, sub_sort):
+#         self.section(self.stack(), 
+
+# def group_by(tasks, field):
+#     """Group tasks by field values."""
+#     groups = {}
+#     for task in tasks:
+#         if field in task:
+#             if task[field] in groups:
+#                 groups[ task[field] ].append(task)
+#             else:
+#                 groups[ task[field] ] = [task]
+#     return groups
+
+# def sort_by(tasks, field = "urgency", reverse = False):
+#     return sorted(tasks, key = lambda t: t[field], reverse = reverse)
+
+##### Widgets #####
+
+# class packed:
+#     def card(task, in_widget = None, show_only = None):
+#         """Widget being a single task."""
+
+#         if not show_only:
+#             # Show all existing fields.
+#             show_only = task.keys()
+
+#         sid = str(task["id"])
+#         if ':' in task["description"]:
+#             short, desc = task["description"].split(":")
+#             title = ":".join([sid,short.strip()])
+#         else:
+#             desc = task["description"]
+#             title = sid
+
+#         segments = []
+#         for key in show_only:
+#             if key in task.keys() and key not in ["id", "description"]:
+#                 val = task[key]
+#                 segment = f"{key}: "
+#                 if type(val) == str:
+#                     segments.append(segment+t)
+#                 elif type(val) == list:
+#                     g = Columns([f"+{t}" for t in val])
+#                     segments.append(g)
+#                 else:
+#                     segments.append(segment+str(val))
+
+#         cols = Columns(segments)
+#         grp = rich.console.Group(desc.strip(), cols)
+#         panel = rich.panel.Panel(grp, title = title,
+#             title_align="left", expand = False, padding = (0,1))
+
+#         return panel
+
+
+#     def stack(tasks, in_widget = None, show_only = None):
+#         """Widget being a stack of tasks widgets."""
+#         stack = rich.table.Table()
+#         stack.add_column("Tasks")
+#         for task in tasks:
+#            stack.add_row( card(task, in_widget, show_only) )
+#         return stack
+
+
+#     def sections(tasks, field, values, in_widget = None, show_only = None):
+#         """Widget being a panel of stack widgets."""
+#         sections = []
+#         groups = group_by(tasks, field)
+#         for val in values:
+#             if val in groups:
+#                 sections.append( rich.panel.Panel(stack(groups[val], in_widget, show_only), title = val) )
+#         return rich.console.Group(*sections)
+
 
 if __name__ == "__main__":
 
@@ -125,7 +274,13 @@ if __name__ == "__main__":
         jdata = json.loads(out)
         print(json.dumps(jdata, indent=4))
 
+        tasker = Card(show_only)
+        stacker = VerticalStack(tasker)
+        group_by_status = Grouper("status")
+        sort_on_values = OnValues(["pending","completed"])
+        sectioner = VerticalSections(stacker, sort_on_values, group_by_status)
+
         console = Console()
         console.rule("taskwarrior-fancy")
-        console.print(sections(jdata, "status", ["pending","completed"], None, show_only))
+        console.print(sectioner(jdata))