Skip to content

Commit 77ce4cb

Browse files
committed
fix: 修复UserInfoFromAdmin自定义用户模型更新错误
1 parent 4060fed commit 77ce4cb

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

fastapi_user_auth/admin.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,8 @@ async def get_form(self, request: Request) -> Form:
233233
async def handle(self, request: Request, data: BaseModel, **kwargs) -> BaseApiOut[Any]:
234234
stmt = update(self.user_model).where(self.user_model.username == request.user.username).values(data.dict())
235235
await self.site.db.async_execute(stmt)
236-
return BaseApiOut(data={**request.user.dict(), **data.dict()})
236+
await self.site.db.async_refresh(request.user)
237+
return BaseApiOut(data=self.schema_submit_out.parse_obj(request.user))
237238

238239
async def has_page_permission(self, request: Request) -> bool:
239240
return await self.site.auth.requires(response=False)(request)

fastapi_user_auth/app.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def __init__(self, app: "AdminApp"):
4141
self.auth.user_model, "UserRegIn", include={"username", "password", "email"}
4242
)
4343
self.UserRegFormAdmin.schema_submit_out = self.UserRegFormAdmin.schema_submit_out or self.schema_user_login_out
44+
self.UserInfoFormAdmin.user_model = self.auth.user_model
4445
self.UserInfoFormAdmin.schema = self.UserInfoFormAdmin.schema or schema_create_by_schema(
4546
self.auth.user_model,
4647
"UserInfoForm",

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ dependencies = [
4242
"passlib>=1.7.4",
4343
"bcrypt>=4.0.0",
4444
"sqlmodelx>=0.0.2",
45+
"sqlalchemy-database>=0.0.10",
4546
]
4647

4748

0 commit comments

Comments
 (0)