build.db.models

Complexity

Total Complexity

5

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Test Coverage

Coverage

93.75%

Importance

Changes

0

1 Method

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

"""DB models."""
# pylint: disable=unused-argument,invalid-name,unsubscriptable-object
# pylint: disable=no-name-in-module

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] = None
    updated_at: Optional[datetime] = None

    def model_dump(self, **kwargs) -> dict:
        """Model to dict."""
        values = super().model_dump(**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