task_manager.py

import os
import json
from datetime import datetime
import shortuuid

from logger import Logger, INFO, WARNING, ERROR

class TaskManager:
    def __init__(self, task_file="data/tasks.json"):
        self.task_file = task_file
        self.tasks = {}
        self.logger = Logger(True, "logs/task_manager_logs.txt")

        os.makedirs("data", exist_ok=True)
        os.makedirs("logs", exist_ok=True)

        self._load_tasks()
    
    def _load_tasks(self):
        if os.path.exists(self.task_file):
            try:
                with open(self.task_file, "r") as f:
                    self.tasks = json.load(f)
                    self.logger.log("Loaded tasks from file", INFO)
                    return True
            
            except Exception as e:
                self.logger.log(f"Error loading tasks: {e}", ERROR)
                self.tasks = {}
                return False
        
        else:
            self.tasks = {}
            return False
    
    def _save_tasks(self):
        try:
            with open(self.task_file, "w") as f:
                json.dump(self.tasks, f, indent=4)
            self.logger.log("Saved tasks to file", INFO)
            return True
        
        except Exception as e:
            self.logger.log(f"Error saving tasks: {e}", ERROR)
            return False
    
    def _ensure_project_exists(self, username, project_id):
        return self._username_in_tasks(username) and self._project_id_assigned_to_user(project_id, username)
    
    def _username_in_tasks(self, username):
        return username in self.tasks
    
    def _project_id_assigned_to_user(self, project_id, username):
        return project_id in self.tasks[username]
    
    def _task_id_in_project(self, username, project_id, task_id):
        return task_id in self.tasks[username][project_id]
    
    def _project_exists_and_task_id_in_project(self, username, project_id, task_id):
        return (self._ensure_project_exists(username, project_id) and 
                self._task_id_in_project(username, project_id, task_id))
    
    def add_task(self, username, project_id, name, description="", due=None):
        def create_task():
            self.tasks[username][project_id][task_id] = {
            "name": name,
            "description": description,
            "created": timestamp,
            "due": due if due else None,
            "done": False
        }

        if not name.strip():
            self.logger.log("Task name cannot be empty", WARNING)
            return False, "Invalid task name"
        
        if not self._username_in_tasks(username):
            self.tasks[username] = {}
        
        if not self._project_id_assigned_to_user(project_id, username):
            self.tasks[username][project_id] = {}
        
        task_id = shortuuid.uuid()[:8]
        timestamp = datetime.now().isoformat()

        create_task()

        self.logger.log(f"Added task '{name}' ({task_id}) for user {username} in project {project_id}", INFO)
        self._save_tasks()
        return task_id, None
    
    def list_tasks(self, username, project_id):
        if self._ensure_project_exists(username, project_id):
            return self.tasks[username][project_id]
        else:
            self.logger.log(f"No tasks found for user {username} in project {project_id}", WARNING)
            return {}
    
    def delete_task(self, username, project_id, task_id):
        if self._project_exists_and_task_id_in_project(username, project_id, task_id):
            del self.tasks[username][project_id][task_id]
            self.logger.log(f"Deleted task '{task_id}' for user '{username}'", INFO)
            self._save_tasks()
            return True
        
        self.logger.log(f"Attempt to delete non-existent task {task_id}", WARNING)
        return False, "Task not found"
    
    def edit_task(self, username, project_id, task_id, name=None, description=None, due=None):
        if self._project_exists_and_task_id_in_project(username, project_id, task_id):
            task = self.tasks[username][project_id][task_id]
            if name: task['name'] = name
            if description: task['description'] = description
            if due: task['due'] = due
            self._save_tasks()
            self.logger.log(f"Edited task '{task_id}' for user '{username}'", INFO)
            return True, None
        
        self.logger.log(f"Attempt to edit non-existent task '{task_id}'", WARNING)
        return False, "Task not found"
    
    def mark_task(self, username, project_id, task_id, done=True):
        if self._project_exists_and_task_id_in_project(username, project_id, task_id):
            self.tasks[username][project_id][task_id]["done"] = done
            state = "complete" if done else "incomplete"
            self.logger.log(f"Marked task '{task_id}' as {state}", INFO)
            self._save_tasks()
            return True, None
        self.logger.log(f"Attempt to mark non-existent task '{task_id}'", WARNING)
        return False, "Task not found"