注册功能成功。

This commit is contained in:
高子兴 2024-11-19 21:23:59 +08:00
parent 077ecb8d5e
commit f84a035640
4 changed files with 17 additions and 11 deletions

View File

@ -37,7 +37,7 @@ async def login(response: Response, user_data: dict, session: SessionDep):
# 验证用户名和密码 # 验证用户名和密码
if not user or user.password != user_data['password']: if not user or user.password != user_data['password']:
raise HTTPException(status_code=401, detail="Login failed") raise HTTPException(status_code=401, detail="登录失败,用户名或密码错误")
# 生成JWT token # 生成JWT token
token = create_access_token(data={"id": user.id, "role": user.role, "tanant_id": user.tenant.id}) token = create_access_token(data={"id": user.id, "role": user.role, "tanant_id": user.tenant.id})
@ -49,3 +49,18 @@ async def login(response: Response, user_data: dict, session: SessionDep):
session.close() session.close()
return {"message": f"Login successful"} return {"message": f"Login successful"}
@router.post("/api/s1/register")
async def register(data: dict, session: SessionDep):
if session.exec(select(Tenant).where(Tenant.name == data['name'])):
raise HTTPException(status_code=409, detail="租户名已存在")
if session.exec(select(User).where(User.username == data['username'])):
raise HTTPException(status_code=409, detail="用户名已存在")
tenant = Tenant(name=data['name'])
user = User(username=data['username'], password=data['password'], role=1, tenant=tenant)
session.add(tenant)
session.add(user)
session.close()
return {"detail": "注册成功"}

View File

@ -15,6 +15,4 @@ from dependencies import *
router = APIRouter() router = APIRouter()
@router.get(...)
def example():
return "hello"

View File

@ -14,7 +14,3 @@ from dependencies import *
router = APIRouter() router = APIRouter()
@router.get(...)
def example():
return "hello"

View File

@ -16,6 +16,3 @@ from dependencies import *
router = APIRouter() router = APIRouter()
@router.get(...)
def example():
return "hello"