models.py

from dataclasses import dataclass
from datetime import datetime
from typing import Callable

from pathier import Pathier

root = Pathier(__file__).parent


@dataclass
class Company:
    """
    Fields:
    * id: int
    * name: str
    * date_added: datetime
    """

    id: int = -1
    name: str = ""
    date_added: datetime = datetime.now()


@dataclass
class Board:
    """
    Fields:
    * company: models.Company
    * id: int
    * url: str
    * active: bool
    * date_added: datetime
    """

    company: Company
    id: int = -1
    url: str = ""
    active: bool = True
    date_added: datetime = datetime.now()


@dataclass
class Listing:
    """
    Fields:
    * company: models.Company
    * id: int
    * position: str
    * location: str
    * url: str
    * alive: bool
    * date_added: datetime
    * date_removed: datetime | None
    """

    company: Company
    id: int = -1
    position: str = ""
    location: str = ""
    url: str = ""
    alive: bool = True
    date_added: datetime = datetime.now()
    date_removed: datetime | None = None

    def __post_init__(self):
        self.alive = bool(self.alive)

    def prune_strings(self):
        """Remove extra whitespaces from `self.position` and `self.location` strings."""
        prune: Callable[[str], str] = lambda s: " ".join(s.strip().split())
        self.position = prune(self.position)
        self.location = prune(self.location)


@dataclass
class Application:
    """
    Fields:
    * listing: models.Listing
    * id: int
    * date_applied: datetime
    """

    listing: Listing
    id: int = -1
    date_applied: datetime = datetime.now()


@dataclass
class Rejection:
    """
    Fields:
    * application: Application
    * id: int
    * date_rejected: datetime
    """

    application: Application
    id: int = -1
    date_rejected: datetime = datetime.now()


@dataclass
class Scraper:
    """
    Fields:
    * company: models.Company
    * board: models.Board
    """

    company: Company
    board: Board