taskai-cli 1.1.4__tar.gz → 1.2.0__tar.gz

{taskai_cli-1.1.4 → taskai_cli-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taskai-cli
-Version: 1.1.4
+Version: 1.2.0
 Author-email: Alex Paskal <alexcpaskal@gmail.com>
 Requires-Python: >=3.12
 Requires-Dist: google-genai>=2.8

{taskai_cli-1.1.4 → taskai_cli-1.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "taskai-cli"
-version = "1.1.4"
+version = "1.2.0"
 description = ""
 authors = [{name = "Alex Paskal", email = "alexcpaskal@gmail.com"}]
 readme = "README.md"

{taskai_cli-1.1.4 → taskai_cli-1.2.0}/taskai/cli.py

@@ -112,11 +112,17 @@ class Controller:
         print(f"Added comment {comment_id} to item {item_id} - '{content}'")
         db.commit()
             
-    def update_item(item_id: int|str, **kwargs):
+    def update_item(item_id: int|str, recursive=False, **kwargs):
         if not _is_int(item_id):
             item_id = Controller._find_model_by_stringmatch("name", item_id)
         db.update_item(item_id, **kwargs)
         print(f"Updated item {item_id}")
+
+        if recursive:
+            children = db.get_item_attr(item_id, "child_ids")
+            for child_id in children:
+                Controller.update_item(child_id, recursive=True, **kwargs) 
+
         db.commit()
 
     def delete_item(id_: int|str):
@@ -134,7 +140,10 @@ class Controller:
     
     def delete_completed():
         for item_id in db.get_item_ids():
-            item: TodoItem = db.get_item(item_id)
+            try:
+                item: TodoItem = db.get_item(item_id)
+            except:
+                continue
             if item.completed:
                 db.delete_item(item_id)
         db.commit()
@@ -370,7 +379,7 @@ def execute_commands(*args, **kwargs) -> int:
 
             case "complete" | "done":
                 match args[1]:
-                    case _ if _is_int(args[1]): Controller.update_item(args[1], completed=True)
+                    case _ if _is_int(args[1]): Controller.update_item(args[1], completed=True, recursive=True)
                     case _: Controller.throw_error("unrecognized complete command", *args, **kwargs)
 
             case "examples":

{taskai_cli-1.1.4 → taskai_cli-1.2.0}/taskai/models.py

@@ -61,5 +61,10 @@ class CLIConfig(Base):
     GEMINI_MODEL: str = None
     GEMINI_API_KEY: str = None
 
+    # VIEW
+    DISPLAY_STRING: str = "id name status due_by"
+    DISPLAY_COLORS: str = "_ white dark_orange _"
+
+
 class LLMConfig(Base):
     pass

{taskai_cli-1.1.4 → taskai_cli-1.2.0}/taskai/views.py

@@ -1,6 +1,7 @@
 # local
 from taskai.json_dir_database import JsonDirectoryDatabase
 from taskai.models import TodoItem, Comment
+from taskai.config import config
 
 # external
 from rich import print
@@ -8,19 +9,68 @@ from rich.console import Console
 import rich
 
 
+"""
+Lets give the lists a display string
+
+
+set of valid attributes
+- id
+- name
+- status
+- 
+
+
+"""
+
+
+VALID_ATTRS = {
+    "id",
+    "name",
+    "created_on",
+    "due_by",
+    "priority",
+    "status"
+}
+
+
+@config("DISPLAY_STRING", "display_str")
+@config("DISPLAY_COLORS", "display_colors")
 def view_lists(
         db: JsonDirectoryDatabase, 
         roots: list[int], 
-        show_done=True, 
+        show_done=True,
+        display_str: str = "id name status", 
+        display_colors: str = None
 ):
     """Shows all the lists"""
 
+    attrs = display_str.lower().split(" ")
+
+    if display_colors:
+        colors = display_colors.lower().split(" ")
+        assert len(colors) == len(attrs), "must have a color for every attr"
+
+    for attr in attrs:
+        assert attr in VALID_ATTRS, "invalid attribute in display string {}".format(attr)
+
+    def _render_display_string(item: TodoItem):
+        display_string = ""
+        for i, attr in enumerate(attrs):
+            part = getattr(item, attr)
+            if not part:
+                continue
+            part = str(part)
+            if display_colors and colors[i] != "_":
+                part = _wrap_string(part, f"[{colors[i]}]", f"[/{colors[i]}]")  
+            display_string += f" {part}"
+        display_string = _wrap_string(display_string, "[strike]", "[/strike]", condition=item.completed)
+        return display_string
 
     def _print_item(item: TodoItem, level: int):
         if not show_done and item.completed:
             return
-        display_string = f"{item.id} {item.name}"
-        display_string = _wrap_string(display_string, "[strike]", "[/strike]", condition=item.completed)
+        
+        display_string = _render_display_string(item)
         indent = "\t" * level
         print(indent + display_string)
 
@@ -101,4 +151,4 @@ def _wrap_string(string_, before, after=None, condition=True):
     if condition:
         return f"{before}{string_}{after}"
     else:
-        return string_
+        return string_