app/models/users.py

@@ -1,18 +1,22 @@
 from pydantic import BaseModel
-
+from pydantic_mongo import AbstractRepository, ObjectIdField
 
 class User(BaseModel):
-    id: int
+    id: ObjectIdField = None
     username: str
     password: str
     roles: str
     disabled: bool
 
 class UserOut(BaseModel):
-    id: int
+    id: ObjectIdField = None
     username: str
     roles: str
     disabled: bool
 
 class UserInDB(User):
     password: str
+
+class UserRepository(AbstractRepository[User]):
+    class Meta:
+        collection_name = "users"