app/tracker/tasks/models.py

"""
Database models for tasks.
"""
from datetime import datetime
from typing import Optional
from uuid import UUID


class ScmTask:
    """Model representing a task in the database"""
    
    def __init__(
        self,
        id: UUID,
        merchant_id: UUID,
        assigned_to: UUID,
        title: str,
        description: Optional[str],
        status: str,
        latitude: Optional[float],
        longitude: Optional[float],
        address: Optional[str],
        scheduled_at: Optional[datetime],
        started_at: Optional[int],
        completed_at: Optional[int],
        created_at: datetime,
        updated_at: datetime
    ):
        self.id = id
        self.merchant_id = merchant_id
        self.assigned_to = assigned_to
        self.title = title
        self.description = description
        self.status = status
        self.latitude = latitude
        self.longitude = longitude
        self.address = address
        self.scheduled_at = scheduled_at
        self.started_at = started_at
        self.completed_at = completed_at
        self.created_at = created_at
        self.updated_at = updated_at