build.db.models

Complexity

Total Complexity

5

Size/Duplication

Total Lines	30
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	DocumentBaseModel.dict()	0	8	5

"""DB models."""
# pylint: disable=unused-argument,invalid-name

from datetime import datetime
from typing import Optional

from pydantic import BaseModel, Field


class DocumentBaseModel(BaseModel):
    """DocumentBaseModel."""

    id: str = Field(None, alias="_id")
    inserted_at: Optional[datetime]
    updated_at: Optional[datetime]

    def dict(self, **kwargs) -> dict:
        """Model to dict."""
        values = super().dict(**kwargs)
        if "id" in values and values["id"]:
            values["_id"] = values["id"]
        if "exclude" in kwargs and "_id" in kwargs["exclude"]:
            values.pop("_id")
        return values


class LivenessDoc(DocumentBaseModel):
    """Liveness document model."""
    enabled: bool