infra.py

import dataclasses
import datetime
import os
import pickle

from typing import Any
from typing import Dict
from typing import List
from typing import Optional
from typing import Union

from models import Feedback
from models import User


@dataclasses.dataclass()
class Store:
    feedbacks: List[Feedback] = dataclasses.field(default_factory=list)
    users: Dict[int, User] = dataclasses.field(default_factory=dict)
    next_user_id: int = 1
    created_at: datetime.datetime = dataclasses.field(default_factory=datetime.datetime.utcnow)


@dataclasses.dataclass()
class Repository:
    store: Store = dataclasses.field(default_factory=Store)
    filename: Optional[str] = None

    def reset(self):
        self.store = Store()

    @classmethod
    def load(cls, filename: str) -> "Repository":
        if os.path.exists(filename):
            with open(filename, "rb") as file:
                store = pickle.load(file)
        else:
            store = Store()
        return cls(
            store=store,
            filename=filename
        )

    def persistent(self):
        if self.filename is None:
            return

        with open(self.filename, "wb") as file:
            pickle.dump(self.store, file)


@dataclasses.dataclass()
class Response:
    mode: str
    status: int
    message: Union[List[Any], Dict[Any, Any]]

    def pretty(self) -> str:
        result = [
            f"mode = {self.mode}",
            f"status = {self.status}",
            "-" * 40
        ]
        if isinstance(self.message, dict):
            for k, v in self.message.items():
                result.append(k)
                result.append(v)
        else:
            for i in self.message:
                result.append(i)

        return "\n".join([o if isinstance(o, str) else repr(o) for o in result])