import asyncio
from typing import Optional

from pydantic import BaseModel
from beanie import PydanticObjectId, Document, init_beanie


class OutTestModel(BaseModel):
    id: Optional[PydanticObjectId]
    name: str


class TestModel(OutTestModel, Document):
    id: Optional[PydanticObjectId]
    secret_message: str


async def main():
    await init_beanie(
        connection_string="YOUR_CONNECTION_STRING",
        document_models=[TestModel])
    mymodel = TestModel(name="jeff", secret_message="this is gonna break")
    await mymodel.save()
    print(OutTestModel.parse_obj(mymodel))

asyncio.run(main())

import asyncio
from typing import Optional
from configfile import dburl
from pydantic import BaseModel
from beanie import PydanticObjectId, Document, init_beanie


class OutTestModel(BaseModel):
    id: Optional[PydanticObjectId]
    name: str


class TestModel(OutTestModel, Document):
    id: Optional[PydanticObjectId]
    secret_message: str



async def main():
    await init_beanie(
        connection_string=dburl,
        document_models=[TestModel])
    mymodel = TestModel(name="jeff", secret_message="this is gonna break")
    await mymodel.save()
    print(OutTestModel.parse_obj(mymodel))
    print(OutTestModel.parse_obj(mymodel.dict(by_alias=True)))

asyncio.run(main())

from typing import Optional

import motor.motor_asyncio
from beanie import PydanticObjectId, Document, init_beanie
from fastapi import FastAPI
from pydantic import BaseModel


DATABASE_URL = "mongodb://localhost:27017"
client = motor.motor_asyncio.AsyncIOMotorClient(
    DATABASE_URL, uuidRepresentation="standard"
)
db = client["database_name"]


class User(Document):
    email: str
    hashed_password: str


class UserRead(BaseModel):
    id: Optional[PydanticObjectId]
    email: str


app = FastAPI()


@app.get("/user", response_model=UserRead)
async def get_user():
    user = await User.all().first_or_none()
    return user


@app.on_event("startup")
async def on_startup():
    await init_beanie(
        database=db,
        document_models=[
            User,
        ],
    )
    user = User(email="[email protected]", hashed_password="abc")
    await user.save()

if isinstance(res, BaseModel):
        read_with_orm_mode = getattr(res.__config__, "read_with_orm_mode", None)
        if read_with_orm_mode:
            # Let from_orm extract the data from this model instead of converting
            # it now to a dict.
            # Otherwise there's no way to extract lazy data that requires attribute
            # access instead of dict iteration, e.g. lazy relationships.
            return res
        return res.dict(
            by_alias=True,
            exclude_unset=exclude_unset,
            exclude_defaults=exclude_defaults,
            exclude_none=exclude_none,
        )

class User(Document):
    email: str
    hashed_password: str

    class Config:
        read_with_orm_mode = True

class UserRead(BaseModel):
    id: Optional[PydanticObjectId]
    email: str

    @root_validator(pre=True)
    def fill_id(cls, values):
        id = values.get("_id")
        if id is not None:
            values["id"] = id
        return values

class UserRead(BaseModel):
    id: Optional[PydanticObjectId]
    email: str

    class Config:
        orm_mode = True

# ...

@app.get("/user")
async def get_user():
    user = await User.all().first_or_none()
    return UserRead.from_orm(user)