mongodb/models/users.py

import uuid
from typing import List, Optional
from pydantic import BaseModel, Field, SecretStr, PrivateAttr
from pydantic.networks import EmailStr

from pydantic_mongo import ObjectIdField

'''Class for user model used to relate users to past calls'''
class User(BaseModel):
    _id: uuid.UUID = PrivateAttr(default_factory=uuid.uuid4) # private attr not included in http calls
    user_id: str
    name: str
    email: EmailStr = Field(unique=True, index=True)
    password: SecretStr
    call_ids: Optional[List[str]] = None

    class Config:
        populate_by_name = True
        json_schema_extra = {
            "example": {
                "user_id": "65ede65b6d246e52aaba9d4f",
                "name": "benjolo",
                "email": "[email protected]",
                "password": "therealbenjolo",
                "call_ids": ["65e205ced1be3a22854ff300", "65df8c3eba9c7c2ed1b20e85"]
            }
        }

'''Class for updating user records'''
class UpdateUser(BaseModel):
    user_id: Optional[str] = None
    name: Optional[str] = None
    email: Optional[EmailStr] = None
    ''' To decode use -> SecretStr("abc").get_secret_value()'''
    # password: Optional[SecretStr]
    call_ids: Optional[List[str]] = None

    class Config:
        populate_by_name = True
        json_schema_extra = {
            "example": {
                "email": "[email protected]",
                "call_ids": ["65e205ced1be3a22854ff300", "65df8c3eba9c7c2ed1b20e85", "65eef930e9abd3b1e3506906"]
            }
        }