import asyncio

from mango import Document, EmbeddedDocument, Field, Mango


# 嵌入式文档
class Author(EmbeddedDocument):
    name: str
    profile: str | None = None


# Mango 文档模型
class Book(Document):
    name: str = Field(primary_key=True)  # 将字段设置为主键，如果不显式指定主键，则会自动创建 id 字段作为主键
    summary: str | None = None
    author: Author  # 嵌入文档
    price: int = Field(index=True)  # 为字段添加索引


async def main():
    # 初始化 Mango，它会创建连接并初始化文档模型，你可以传入 db 或者 uri 参数来指定连接
    await Mango.init()

    # 像 pydantic 的模型一样使用
    book = Book(name="book", author=Author(name="author"), price=10)
    # 将它插入到数据库中
    await book.save()

    # Mango 提供了丰富的查询语言，允许您使用 Python 表达式来查询
    if book := await Book.find(Book.price <= 20, Book.author.name == "author").get():
        # 更新文档的 summary 字段
        book.summary = "summary"
        await book.update()


if __name__ == "__main__":
    asyncio.run(main())