Add input validation to register
This commit is contained in:
@@ -49,8 +49,12 @@ def authenticate_user(db: Session, username: str, password: str):
|
||||
|
||||
def register(db: Session, username: str, password: str, avatar: str):
|
||||
"""Register a new user"""
|
||||
if avatar == "":
|
||||
if len(avatar) == 0:
|
||||
raise HTTPException(status_code=400, detail="No avatar was provided")
|
||||
if len(username) == 0:
|
||||
raise HTTPException(status_code=400, detail="No username was provided")
|
||||
if len(password) == 0:
|
||||
raise HTTPException(status_code=400, detail="No password was provided")
|
||||
db_user = get_user_by_username(db, username)
|
||||
if db_user:
|
||||
raise HTTPException(status_code=400, detail="Username already registered")
|
||||
@@ -64,6 +68,7 @@ def register(db: Session, username: str, password: str, avatar: str):
|
||||
|
||||
|
||||
def login(db: Session, username: str, password: str):
|
||||
"""Log in based on username and password; supply access token if succeeded"""
|
||||
user = authenticate_user(db, username, password)
|
||||
if not user:
|
||||
raise HTTPException(status_code=401, detail="Invalid username or password")
|
||||
|
||||
Reference in New Issue
Block a user