Skip to content

Commit 1ad071f

Browse files
M03EDImMohammad20000
authored andcommitted
fix(user): handle expire on modify with template
1 parent 68f7d1a commit 1ad071f

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

app/db/crud/user.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -454,10 +454,13 @@ async def create_user(db: AsyncSession, new_user: UserCreate, groups: list[Group
454454
Returns:
455455
User: Created user object.
456456
"""
457-
db_user = User(**new_user.model_dump(exclude={"group_ids", "expire", "proxy_settings", "next_plan"}))
457+
db_user = User(
458+
**new_user.model_dump(exclude={"group_ids", "expire", "proxy_settings", "next_plan", "on_hold_timeout"})
459+
)
458460
db_user.admin = admin
459461
db_user.groups = groups
460462
db_user.expire = new_user.expire or None
463+
db_user.on_hold_timeout = new_user.on_hold_timeout or None
461464
db_user.proxy_settings = new_user.proxy_settings.dict()
462465

463466
db.add(db_user)

app/operation/user.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -400,14 +400,14 @@ def load_base_user_args(template: UserTemplate) -> dict:
400400
if template.expire_duration:
401401
user_args["expire"] = dt.now(tz.utc) + td(seconds=template.expire_duration)
402402
else:
403-
user_args["expire"] = None
403+
user_args["expire"] = 0
404404
else:
405405
user_args["expire"] = 0
406406
user_args["on_hold_expire_duration"] = template.expire_duration
407407
if template.on_hold_timeout:
408408
user_args["on_hold_timeout"] = dt.now(tz.utc) + td(seconds=template.on_hold_timeout)
409409
else:
410-
user_args["on_hold_timeout"] = None
410+
user_args["on_hold_timeout"] = 0
411411

412412
return user_args
413413

0 commit comments

Comments
 (0)