task_types.py

from dataclasses import dataclass
from datetime import datetime
from enum import StrEnum

@dataclass
class VerificationData:
    method: str

# TODO: achievement diary and skill verification methods
@dataclass
class CollectionLogVerificationData(VerificationData):
    item_ids: list[int]
    count: int

@dataclass
class AchievementDiaryVerificationData(VerificationData):
    region: str
    difficulty: str

@dataclass
class SkillVerificationData(VerificationData):
    experience: dict
    count: int


@dataclass
class TaskData:
    id: str
    name: str
    short_name: str
    tip: str
    wiki_link: str
    image_link: str
    display_item_id: int
    tags: list["TaskTag"]
    verification: VerificationData

class TaskTag(StrEnum):
    LMS = "lms"


@dataclass
class UserCurrentTask:
    id: str
    assigned_date: datetime = None

@dataclass
class CompletionMethod(StrEnum):
    MANUAL = 'manual'
    SYNC = 'sync'

@dataclass
class UserCompletedTask:
    id: str
    assigned_date: datetime = None
    completed_date: datetime = None
    completed_item_ids: list[int] = None
    was_active: bool | None = None
    method: CompletionMethod | None = None
    play_time: int | None = None

@dataclass
class UserTaskList:
    current_task: UserCurrentTask
    completed_tasks: list[UserCompletedTask]
    recorded_item_ids_by_task: dict[str, list[int]] = None

@dataclass
class TierProgress:
    percent_complete: int
    total: int
    total_complete: int

@dataclass
class UserTierProgressCache:
    easy: TierProgress
    medium: TierProgress
    hard: TierProgress
    elite: TierProgress
    master: TierProgress
    passive: TierProgress
    extra: TierProgress
    pets: TierProgress

class PageTask:
    def __init__(self, is_completed: bool, is_current: bool, task_data: TaskData,
                 completed_date = None, completed_date_iso: str = None,
                 verification_item_ids: list[int] = None,
                 verification_required_count: int = None,
                 related_item_ids: list[int] = None,
                 completed_item_ids: list[int] = None):
        self.name = task_data.name
        self.is_completed = is_completed
        self.id = task_data.id
        self.is_current = is_current
        self.wiki_link = task_data.wiki_link
        self.image_link = task_data.image_link
        self.tip = task_data.tip
        self.completed_date = completed_date
        self.completed_date_iso = completed_date_iso
        self.verification_item_ids = verification_item_ids or []
        self.verification_required_count = verification_required_count
        self.related_item_ids = related_item_ids or []
        self.completed_item_ids = completed_item_ids or []

@dataclass
class LeaderboardEntry:
    username: str
    easy_progress: TierProgress
    medium_progress: TierProgress
    hard_progress: TierProgress
    elite_progress: TierProgress
    master_progress : TierProgress

    # TODO Have different weightings per tier
    def points(self) -> int:
        return self.easy_progress.percent_complete + self.medium_progress.percent_complete + \
            self.hard_progress.percent_complete + self.elite_progress.percent_complete