RobertRosca
commented
2 years ago Document objects are used to define the mapping between Beanie and a collection in the database, so the error about the collection not being initialized (not existing) is correct, since all Documents should be mapped to the db.
The problem is that you define UserBase as:
class UserBase(Document):
...And then all of your other classes inherit from it, so all of the other classes are also Documents.
If you want to use the other classes just for validation, then they shouldn't inherit Document, only a single class should be a Document in this case, so you can do something like:
from beanie import Document
from pydantic import BaseModel
# Now inherits from pydantic `BaseModel`, not document
class UserBase(BaseModel):
username: str | None
parent_id: str | None
role_id: str | None
payment_type: str | None
disabled: bool | None
note: str | None
access_token: str | None
# User inherits `UserBase` and `Document`
class User(UserBase, Document):
username: str
password: str
disabled: bool = False
salt: str
class Settings:
name = "users"
# Rest of the classes inherit `UserBase` so they're
# only pydantic models, not documents
class UserIn(UserBase):
username: str
password: str
disabled: bool = False
class UserOut(UserBase):
note: str | None
access_token: str | NoneThis should solve the issue you're having.
Also, you're repeating some of the fields where the field is already inherited, for example UserIn has username and disabled which are both in UserBase, so password is needed there. And by marking all of the fields in UserBase as Optional you end up losing the validation of required fields being present where you'd probably like to have them.
This page from the FastAPI docs shows a user model as an example: https://fastapi.tiangolo.com/tutorial/extra-models/?h=userout#reduce-duplication
nghianv19940
Hi I'm using beanie with fastapi. this is my model:
my routes:
my beane init function:
if i'm not put UserIn to the init beanie function, then call /users, i will get this error:
Then i put UserIn in the beanie init function. Everything is working.
I just want the UserIn model to validate input data and beanie try to find it in the db. Models not mapped to db should not have anything to do with the db. Right?