src/tt/controller/generators.py

@@ -40,7 +40,7 @@ def generate_needed_tasks():
             to_create.append(
                 {
                     "text": g.template.format(next=next),
-                    "project": "recurring",
+                    "category": "recurring",
                     "due": next,
                 }
             )

src/tt/controller/tasks.py

@@ -6,6 +6,11 @@ from ..db import db, Task, Category, SavedSearch
 from .. import config
 
 
+def category_lookup(category):
+    if category:
+        category, _ = Category.get_or_create(name=category)
+        return category.id
+
 
 def get_task(item_id: int) -> Task:
     return Task.get_by_id(item_id)
@@ -13,18 +18,19 @@ def get_task(item_id: int) -> Task:
 
 def add_task(
     text: str,
+    category: str,
     status: str,
     due: datetime | None = None,
     type: str = "",
-    project: str = "",
 ) -> Task:
     """
     Add a new task to the database.
     Returns the created task instance.
     """
     with db.atomic():
+        category_id = category_lookup(category)
         task = Task.create(
-            text=text, type=type, status=status, due=due, project=project
+            text=text, type=type, status=status, due=due, category_id=category_id
         )
     return task
 
@@ -34,6 +40,8 @@ def update_task(
     **kwargs,
 ) -> Task:
     with db.atomic():
+        if category := kwargs.pop("category", None):
+            kwargs["category_id"] = category_lookup(category)
         query = Task.update(kwargs).where(Task.id == item_id)
         query.execute()
         task = Task.get_by_id(item_id)
@@ -74,18 +82,18 @@ def _parse_sort_string(sort_string, status_order):
 
 def get_tasks(
     search_text: str | None = None,
+    category: int | None = None,
     statuses: tuple[str] | None = None,
-    projects: tuple[str] | None = None,
     sort: str = "",
 ) -> list[Task]:
     query = Task.select().where(~Task.deleted)
 
     if search_text:
         query = query.where(fn.Lower(Task.text).contains(search_text.lower()))
+    if category:
+        query = query.where(Task.category == Category.get(name=category))
     if statuses:
         query = query.where(Task.status.in_(statuses))
-    if projects:
-        query = query.where(Task.project.in_(projects))
 
     sort_expressions = _parse_sort_string(sort, statuses)
     query = query.order_by(*sort_expressions)

src/tt/db.py

@@ -26,13 +26,19 @@ class BaseModel(Model):
         database = db
 
 
+class Category(BaseModel):
+    name = CharField(unique=True)
+
+    def __str__(self):
+        return self.name
+
 
 class Task(BaseModel):
     text = TextField()
     status = CharField()
     due = DateTimeField(null=True)
+    category = ForeignKeyField(Category, backref="tasks", null=True)
     type = CharField()
-    project = CharField()
     created_at = DateTimeField(default=datetime.now)
     updated_at = DateTimeField(default=datetime.now)
     deleted = BooleanField(default=False)
@@ -105,5 +111,7 @@ class TaskGenerator(BaseModel):
 
 def initialize_db():
     db.connect()
-    db.create_tables([Task, SavedSearch, TaskGenerator])
+    db.create_tables([Category, Task, SavedSearch, TaskGenerator])
+    if not Category.select().exists():
+        Category.create(name="default")
     db.close()

src/tt/import_csv.py

@@ -0,0 +1,43 @@
+import csv
+from datetime import datetime
+from tt.db import initialize_db, Task, Category
+from tt.config import STATUSES
+
+
+def import_tasks_from_csv(filename: str):
+    initialize_db()
+
+    with open(filename, "r") as f:
+        reader = csv.DictReader(f)
+
+        for row in reader:
+            # Parse the due date if it exists and isn't empty
+            due_date = None
+            if row["due"] and row["due"].strip():
+                try:
+                    due_date = datetime.strptime(row["due"].strip(), "%Y-%m-%d")
+                except ValueError:
+                    print(f"Warning: Couldn't parse '{row['due']}', skipping due date")
+
+            # Validate status
+            status = row["status"].lower() if row["status"] else "zero"
+            if status not in STATUSES:
+                print(f"Warning: Invalid status '{status}', defaulting to 'zero'")
+                status = "zero"
+
+            category_id = None
+            if row["category"].strip():
+                category_name = row["category"].strip()
+                category, _ = Category.get_or_create(
+                    name=category_name, defaults={"description": None}
+                )
+                category_id = category.id
+
+            Task.create(
+                text=row["task"],
+                type=row["type"],
+                status=status,
+                due=due_date,
+                category_id=category_id,
+            )
+            print(f"Imported task: {row['task']}")

src/tt/tui/editor.py

@@ -13,8 +13,6 @@ from ..utils import (
     remove_rich_tag,
     filter_to_string,
     get_text_from_editor,
-    get_color_enum,
-    get_colored_date,
 )
 from .keymodal import KeyModal
 from .modals import ChoiceModal, DateModal, ConfirmModal
@@ -52,12 +50,6 @@ class TableColumnConfig:
             # default edit mode
             app._show_input("edit", current_value)
 
-    def format_for_display(self, val):
-        val = str(val)
-        if "\n" in val:
-            val = val.split("\n")[0] + ELLIPSIS
-        return val
-
 
 class EnumColumnConfig(TableColumnConfig):
     def __init__(self, field: str, display_name: str, enum, **kwargs):
@@ -70,10 +62,6 @@ class EnumColumnConfig(TableColumnConfig):
         else:
             raise NotifyValidationError(f"Invalid value {val}. Use: {list(self.enum)}")
 
-    def format_for_display(self, val):
-        return get_color_enum(val, self.enum)
-
-
     def start_change(self, app, current_value):
         # a weird hack? pass app here and correct modal gets pushed
         app.push_screen(ChoiceModal(self.enum, current_value), app.apply_change)
@@ -86,9 +74,6 @@ class DateColumnConfig(TableColumnConfig):
         except ValueError:
             raise NotifyValidationError("Invalid date format. Use YYYY-MM-DD")
 
-    def format_for_display(self, val):
-        return get_colored_date(val)
-
     def start_change(self, app, current_value):
         app.push_screen(DateModal(current_value), app.apply_change)
 

src/tt/tui/modals.py

@@ -61,7 +61,7 @@ class ChoiceModal(ModalScreen):
         yield RadioSet(
             *[
                 RadioButton(
-                    get_color_enum(e.value, config.STATUSES), value=self.selected == str(e.value)
+                    get_color_enum(e.value, config.STATUSES, "red"), value=self.selected == str(e.value)
                 )
                 for e in self._enum
             ]

src/tt/tui/tasks.py

@@ -1,6 +1,7 @@
 import json
 from textual.widgets import Input
 
+from .. import config
 from ..controller.tasks import (
     get_task,
     get_tasks,
@@ -9,7 +10,14 @@ from ..controller.tasks import (
     save_view,
     get_saved_view,
 )
-from .editor import TableEditor
+from ..utils import (
+    get_color_enum,
+    get_colored_date,
+)
+from .editor import (
+    TableEditor,
+    ELLIPSIS,
+)
 
 
 class TT(TableEditor):
@@ -52,34 +60,39 @@ class TT(TableEditor):
             self._load_view(event.value)
             self.refresh_tasks(restore_cursor=False)
             self._hide_input()
-            # if event isn't handled here it will bubble to parent
             event.prevent_default()
-
-    def _db_item_to_row(self, item):
-        """
-            convert db item to an item  
-        """
-        row = []
-
-        for col in self.table_config:
-            val = getattr(item, col.field)
-            display_val = col.format_for_display(val)
-            row.append(display_val)
-
-        return row
+        # if event isn't handled here it will bubble to parent
 
     def refresh_items(self):
         items = get_tasks(
             self.search_query,
-            projects=self.filters.get("project", "").split(","),
+            category=self.filters.get("category"),
             statuses=self.filters.get("status", "").split(",")
             if "status" in self.filters
             else None,
             sort=self.sort_string,
         )
         for item in items:
+            category = get_color_enum(
+                item.category.name if item.category else " - ",
+                config.PROJECTS,
+                "grey"
+            )
+            status = get_color_enum(item.status, config.STATUSES, "red")
+            due = get_colored_date(item.due)
+
+            if "\n" in item.text:
+                text = item.text.split("\n")[0] + ELLIPSIS
+            else:
+                text = item.text
+
             self.table.add_row(
-                *self._db_item_to_row(item),
+                str(item.id),
+                text,
+                status,
+                item.type,
+                due,
+                category,
                 key=str(item.id),
             )
 
@@ -87,3 +100,7 @@ class TT(TableEditor):
 def run(default_view):
     app = TT(default_view)
     app.run()
+
+
+if __name__ == "__main__":
+    run()

src/tt/utils.py

@@ -7,10 +7,10 @@ import subprocess
 
 def filter_to_string(filters, search_query):
     pieces = []
-    project = filters.get("project")
+    category = filters.get("category")
     status = filters.get("status")
-    if project:
-        pieces += [f" project:{project}"]
+    if category:
+        pieces += [f" category:{category}"]
     if status:
         pieces += [f" status:{status}"]
     if search_query:
@@ -32,11 +32,18 @@ def advance_enum_val(enum_type, cur_val):
     return members[next_idx]
 
 
-def get_color_enum(value: str, enum: dict[str, dict]) -> str:
-    color = enum.get(value, {"color": "#ff0000"})["color"]
+def get_color_enum(value: str, enum: dict[str, dict], default: str) -> str:
+    color = enum.get(value, {"color": default})["color"]
     return f"[{color}]{value}[/]"
 
 
+def get_colored_category(category: str) -> str:
+    hash_val = sum(ord(c) for c in category)
+    hue = hash_val % 360
+    color = f"rgb({hue},200,200) on default"
+    return f"[{color}]{category}[/]"
+
+
 def get_colored_date(date: datetime.date) -> str:
     if not isinstance(date, datetime.date):
         return ""

tt.toml

@@ -55,11 +55,8 @@ field_type = "date"
 default = ""
 
 [[views.columns]]
-field_name = "project"
-display_name = "Project"
+field_name = "category"
 default = ""
-field_type = "enum"
-enum = "projects"
 
 [[views]]
 name = "recurring"