diff --git a/app.py b/app.py index 9bc770d2..f137ec7a 100755 --- a/app.py +++ b/app.py @@ -2832,6 +2832,16 @@ def register(): user.email_confirmed = True # 暂时默认已确认 db.session.add(user) + db.session.flush() # 获取 user.id + + # 自动创建积分账户,初始10000积分 + credit_account = UserCreditAccount( + user_id=user.id, + balance=10000, + frozen=0 + ) + db.session.add(credit_account) + db.session.commit() # 自动登录 @@ -2995,6 +3005,16 @@ def register_with_phone(): user.phone_confirmed = True db.session.add(user) + db.session.flush() # 获取 user.id + + # 自动创建积分账户,初始10000积分 + credit_account = UserCreditAccount( + user_id=user.id, + balance=10000, + frozen=0 + ) + db.session.add(credit_account) + db.session.commit() # 清除验证码 @@ -3244,6 +3264,16 @@ def register_with_email(): user.email_confirmed = True db.session.add(user) + db.session.flush() # 获取 user.id + + # 自动创建积分账户,初始10000积分 + credit_account = UserCreditAccount( + user_id=user.id, + balance=10000, + frozen=0 + ) + db.session.add(credit_account) + db.session.commit() # 清除验证码