Passed
Pull Request — master (#56)
by Vinicius
03:32
created

build.db.models.DocumentBaseModel.dict()   A

Complexity

Conditions 5

Size

Total Lines 8
Code Lines 7

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 6
CRAP Score 5.0729

Importance

Changes 0
Metric Value
cc 5
eloc 7
nop 2
dl 0
loc 8
ccs 6
cts 7
cp 0.8571
crap 5.0729
rs 9.3333
c 0
b 0
f 0
1
"""DB models."""
2
# pylint: disable=unused-argument,invalid-name
3
4 1
from datetime import datetime
5 1
from typing import Optional
6
7 1
from pydantic import BaseModel, Field
8
9
10 1
class DocumentBaseModel(BaseModel):
11
    """DocumentBaseModel."""
12
13 1
    id: str = Field(None, alias="_id")
14 1
    inserted_at: Optional[datetime]
15 1
    updated_at: Optional[datetime]
16
17 1
    def dict(self, **kwargs) -> dict:
18
        """Model to dict."""
19 1
        values = super().dict(**kwargs)
20 1
        if "id" in values and values["id"]:
21 1
            values["_id"] = values["id"]
22 1
        if "exclude" in kwargs and "_id" in kwargs["exclude"]:
23
            values.pop("_id")
24 1
        return values
25
26
27 1
class LivenessDoc(DocumentBaseModel):
28
    """Liveness document model."""
29
    enabled: bool
30