mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 01:06:59 +08:00
fix: User email check
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from typing import Optional
|
||||
from fastapi import APIRouter, HTTPException, Query
|
||||
from sqlmodel import func, or_, select, delete as sqlmodel_delete
|
||||
from apps.system.crud.user import check_account_exists, get_db_user, single_delete, user_ws_options
|
||||
from apps.system.crud.user import check_account_exists, check_email_exists, get_db_user, single_delete, user_ws_options
|
||||
from apps.system.models.system_model import UserWsModel
|
||||
from apps.system.models.user import UserModel
|
||||
from apps.system.schemas.auth import CacheName, CacheNamespace
|
||||
@@ -116,9 +116,10 @@ async def query(session: SessionDep, trans: Trans, id: int) -> UserEditor:
|
||||
|
||||
@router.post("")
|
||||
async def create(session: SessionDep, creator: UserCreator):
|
||||
count = check_account_exists(session=session, account=creator.account)
|
||||
if count > 0:
|
||||
if check_account_exists(session=session, account=creator.account):
|
||||
raise Exception(f"Account [{creator.account}] already exists!")
|
||||
if check_email_exists(session=session, email=creator.email):
|
||||
raise Exception(f"Email [{creator.email}] already exists!")
|
||||
data = creator.model_dump(exclude_unset=True)
|
||||
user_model = UserModel.model_validate(data)
|
||||
#user_model.create_time = get_timestamp()
|
||||
@@ -147,6 +148,8 @@ async def update(session: SessionDep, editor: UserEditor):
|
||||
raise Exception(f"User with id [{editor.id}] not found!")
|
||||
if editor.account != user_model.account:
|
||||
raise Exception(f"account cannot be changed!")
|
||||
if editor.email != user_model.email and check_email_exists(session=session, account=editor.email):
|
||||
raise Exception(f"Email [{editor.email}] already exists!")
|
||||
origin_oid: int = user_model.oid
|
||||
del_stmt = sqlmodel_delete(UserWsModel).where(UserWsModel.uid == editor.id)
|
||||
session.exec(del_stmt)
|
||||
|
||||
@@ -72,4 +72,6 @@ async def clean_user_cache(id: int):
|
||||
|
||||
|
||||
def check_account_exists(*, session: Session, account: str) -> bool:
|
||||
return session.exec(select(func.count()).select_from(UserModel).where(UserModel.account == account)).one()
|
||||
session.exec(select(func.count()).select_from(UserModel).where(UserModel.account == account)).one() > 0
|
||||
def check_email_exists(*, session: Session, email: str) -> bool:
|
||||
return session.exec(select(func.count()).select_from(UserModel).where(UserModel.email == email)).one() > 0
|
||||
Reference in New Issue
Block a user