beanie_mongodb_odm_quickstart_with_motor_crud_operations.py

1import asyncio
2from typing import Optional
3from motor.motor_asyncio import AsyncIOMotorClient
4from pydantic import BaseModel
5from beanie import Document, Indexed, init_beanie
6
7
8class Category(BaseModel):
9    name: str
10    description: str
11
12
13class Product(Document):
14    name: str  # You can use normal Python types
15    description: Optional[str] = None
16    price: Indexed(float)  # You can use Beanie's Indexed type
17    category: Category  # You can include Pydantic models as fields
18
19
20async def example():
21    # Beanie uses Motor under the hood
22    client = AsyncIOMotorClient("mongodb://localhost:27017")
23
24    # Initialize beanie with the Product document class
25    await init_beanie(database=client.db_name, document_models=[Product])
26
27    # Create a new document
28    chocolate = Category(name="Chocolate", description="A delicious treat")
29    tony = Product(name="Tony's Chocolonely", price=3.95, category=chocolate)
30    await tony.insert()
31
32    # Find documents
33    product = await Product.find_one(Product.name == "Tony's Chocolonely")
34    print(product)
35
36
37if __name__ == "__main__":
38    asyncio.run(example())