Project.py

from enum import Enum
from dataclasses import dataclass
from datetime import datetime, timedelta
from typing import Optional, List
from User import User
from Task import Task
from Status import Status

@dataclass
class DateRange:
    end_date: datetime
    start_date: Optional[datetime]

    def __init__(self, end_date: datetime, start_date: Optional[datetime] = None):
        self.start_date = start_date
        self.end_date = end_date
    
    def is_overdue(self) -> bool:
        return (datetime.now() - self.end_date).days > 0

    def days_overdue(self) -> int:
        return (datetime.now() - self.end_date).days


class Project:
    project_name: str
    project_id: str
    status: Status
    timeline: DateRange
    dri_emails: List[str]
    task_ids: List[str]
    project_url: str
    tasks: List[Task]

    def __init__(self, name: str, prj_id: str, status: Status, dri_emails: List[User], task_ids: List[str], project_url: str, timeline: Optional[DateRange] = None):
        self.project_name = name
        self.project_id = prj_id
        self.status = status
        self.timeline = timeline
        self.dri_emails = dri_emails
        self.task_ids = task_ids
        self.project_url = project_url
        self.tasks = []
    
    def is_finished(self) -> bool:
        return self.status == Status.DONE

    def info(self) -> List[str]:
        ret_val = []

        return ret_val

    def warnings(self) -> List[str]:
        ret_val = []

        if self.timeline != None and self.timeline.is_overdue() and self.status != Status.DONE:
            ret_val.append(f"Project is {self.timeline.days_overdue()} day(s) overdue.")
        
        return ret_val
    
    def errors(self) -> List[str]:
        ret_val = []

        if len(self.dri_emails) == 0:
            ret_val.append("No Project DRI assigned.")

        if self.timeline == None:
            ret_val.append("No existing timeline for project.")

        if all(task is None for task in self.tasks):
            ret_val.append("No tasks created for project.")
        
        return ret_val
    
    def has_issues(self) -> bool:
        return len(self.info()) != 0 or len(self.warnings()) != 0 or len(self.errors()) != 0 
    
    def has_sub_issues(self) -> bool:
        ret_val = False

        for task in self.tasks:
            if task is None:
                continue

            ret_val = ret_val or task.has_issues()
        
        return ret_val
    
    def sub_tasks_are_done(self) -> bool:
        ret_val = True

        for task in self.tasks:
            if task is None:
                continue
                
            ret_val = ret_val and task.status == Status.DONE
        
        return ret_val