coach-scraper/app/exporter.py

from typing import Union

from typing_extensions import TypedDict

from app.repo import AnsiColor, Repo


class Export(TypedDict, total=False):
    # The coach's rapid rating relative to the site they were sourced from.
    rapid: int
    # The coach's blitz rating relative to the site they were sourced from.
    blitz: int
    # The coach's bullet rating relative to the site they were sourced from.
    bullet: int


def _insert(export: Export, key: str, value: any):
    if value is None:
        return
    export[key] = value


class BaseExporter(Repo):
    def __init__(self, site: str, username: str):
        super().__init__(site)
        self.username = username

    def export_rapid(self) -> Union[int, None]:
        raise NotImplementedError()

    def export_blitz(self) -> Union[int, None]:
        raise NotImplementedError()

    def export_bullet(self) -> Union[int, None]:
        raise NotImplementedError()

    def export(self) -> Export:
        """Transform coach-specific data into uniform format."""
        export: Export = {}

        _insert(export, "site", self.site)
        _insert(export, "username", self.username)
        _insert(export, "rapid", self.export_rapid())
        _insert(export, "blitz", self.export_blitz())
        _insert(export, "bullet", self.export_bullet())

        self.log(
            [
                (AnsiColor.INFO, "[INFO]"),
                (None, ": Exported "),
                (AnsiColor.DATA, self.username),
            ]
        )

        return export