Module ai_todo.models

Classes

class Task (title: str, description: str, done_status: bool, category: str, source_code_ref: str, assignee: Optional[str] = None)

Task(title: str, description: str, done_status: bool, category: str, source_code_ref: str, assignee: Optional[str] = None)

Expand source code

@dataclass
class Task:
    title: str
    description: str
    done_status: bool
    category: str
    source_code_ref: str
    assignee: Optional[str] = None

Class variables

var assignee : Optional[str]

var category : str

var description : str

var done_status : bool

var source_code_ref : str

var title : str

class Todos (tasks: dict[str, Task], file_path: str, valid_assignees: list[str])

Todos(tasks: dict[str, ai_todo.models.Task], file_path: str, valid_assignees: list[str])

Expand source code

@dataclass
class Todos:
    tasks: dict[str, Task]
    file_path: str
    valid_assignees: list[str]

    def __post_init__(self):
        self.tasks = {}  # Initialize the tasks dictionary

    def load_tasks(self):
        if os.path.exists(self.file_path):
            data = toml.load(self.file_path)
            for task_data in data.get("tasks", []):
                task = Task(**task_data)
                self.tasks[task.title] = task

    def save_tasks(self):
        with open(self.file_path, "w") as file:
            toml.dump({"tasks": [asdict(task) for task in self.tasks.values()]}, file)

    def add_task(self, new_task: Task):
        if new_task.title in self.tasks:
            raise ValueError(f"A task with the title '{new_task.title}' already exists.")
        self.tasks[new_task.title] = new_task
        self.save_tasks()

    def remove_task(self, title: str):
        if title in self.tasks:
            del self.tasks[title]
            self.save_tasks()
        else:
            raise KeyError(f"No task found with the title '{title}'.")

    def query_tasks_by_assignee(self, assignee_name: str):
        return [task for task in self.tasks.values() if task.assignee == assignee_name]

    def query_tasks(self, pattern: str):
        matching_tasks = []
        for _title, task in self.tasks.items():
            if re.search(pattern, task.title) or re.search(pattern, task.description):
                matching_tasks.append(task)
        return matching_tasks

Class variables

var file_path : str

var tasks : dict[str, Task]

var valid_assignees : list[str]

Methods

def add_task(self, new_task: Task)

Expand source code

def add_task(self, new_task: Task):
    if new_task.title in self.tasks:
        raise ValueError(f"A task with the title '{new_task.title}' already exists.")
    self.tasks[new_task.title] = new_task
    self.save_tasks()

def load_tasks(self)

Expand source code

def load_tasks(self):
    if os.path.exists(self.file_path):
        data = toml.load(self.file_path)
        for task_data in data.get("tasks", []):
            task = Task(**task_data)
            self.tasks[task.title] = task

def query_tasks(self, pattern: str)

Expand source code

def query_tasks(self, pattern: str):
    matching_tasks = []
    for _title, task in self.tasks.items():
        if re.search(pattern, task.title) or re.search(pattern, task.description):
            matching_tasks.append(task)
    return matching_tasks

def query_tasks_by_assignee(self, assignee_name: str)

Expand source code

def query_tasks_by_assignee(self, assignee_name: str):
    return [task for task in self.tasks.values() if task.assignee == assignee_name]

def remove_task(self, title: str)

Expand source code

def remove_task(self, title: str):
    if title in self.tasks:
        del self.tasks[title]
        self.save_tasks()
    else:
        raise KeyError(f"No task found with the title '{title}'.")

def save_tasks(self)

Expand source code

def save_tasks(self):
    with open(self.file_path, "w") as file:
        toml.dump({"tasks": [asdict(task) for task in self.tasks.values()]}, file)

class Work (completed_file: str, incomplete_file: str, valid_assignee: list[str])

Work(completed_file: str, incomplete_file: str, valid_assignee: list[str])

Expand source code

@dataclass
class Work:
    completed: Todos
    incomplete: Todos

    def __init__(self, completed_file: str, incomplete_file: str, valid_assignee: list[str]):
        self.completed = Todos({}, completed_file, valid_assignee)
        self.incomplete = Todos({}, incomplete_file, valid_assignee)
        self.completed.load_tasks()
        self.incomplete.load_tasks()

    def archive_completed_tasks(self):
        titles_to_archive = [title for title, task in self.incomplete.tasks.items() if task.done_status]

        for title in titles_to_archive:
            # Move the task from incomplete to completed
            self.completed.tasks[title] = self.incomplete.tasks[title]
            del self.incomplete.tasks[title]

        self.completed.save_tasks()
        self.incomplete.save_tasks()

Class variables

var completed : Todos

var incomplete : Todos

Methods

def archive_completed_tasks(self)

Expand source code

def archive_completed_tasks(self):
    titles_to_archive = [title for title, task in self.incomplete.tasks.items() if task.done_status]

    for title in titles_to_archive:
        # Move the task from incomplete to completed
        self.completed.tasks[title] = self.incomplete.tasks[title]
        del self.incomplete.tasks[title]

    self.completed.save_tasks()
    self.incomplete.save_tasks()