Skip to content

Commit 9c2c7c8

Browse files
fix(nextPlan): crash when creating user
1 parent fd21d7b commit 9c2c7c8

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

app/db/crud/user.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -459,11 +459,17 @@ async def create_user(db: AsyncSession, new_user: UserCreate, groups: list[Group
459459
db_user.groups = groups
460460
db_user.expire = new_user.expire or None
461461
db_user.proxy_settings = new_user.proxy_settings.dict()
462-
db_user.next_plan = NextPlan(**new_user.next_plan.model_dump()) if new_user.next_plan else None
463462

464463
db.add(db_user)
465464
await db.commit()
466465
await db.refresh(db_user)
466+
467+
if new_user.next_plan:
468+
db_user.next_plan = NextPlan(user_id=db_user.id, **new_user.next_plan.model_dump())
469+
db.add(db_user.next_plan)
470+
await db.commit()
471+
await db.refresh(db_user)
472+
467473
await load_user_attrs(db_user)
468474
return db_user
469475

@@ -778,7 +784,9 @@ async def get_user_sub_update_list(
778784
return result, count
779785

780786

781-
async def autodelete_expired_users(db: AsyncSession, include_limited_users: bool = False) -> list[UserNotificationResponse]:
787+
async def autodelete_expired_users(
788+
db: AsyncSession, include_limited_users: bool = False
789+
) -> list[UserNotificationResponse]:
782790
"""
783791
Deletes expired (optionally also limited) users whose auto-delete time has passed.
784792

0 commit comments

Comments
 (0)