async def register(user: UserInput, background: BackgroundTasks):
if User.find_user_with_email(email=user.email):
return Fortauto.response({"message": "Account already exist"},
status_code=status.HTTP_400_BAD_REQUEST)
if user.password.strip() == user.confirm_password.strip():
try:
password = UserMixin.hash_password(
user.password) if UserMixin.hash_password(
user.password) else None
newUser = User(email=user.email.lower(),
first_name=user.first_name.lower(),
last_name=user.last_name.lower(),
phone_number=user.phone_number,
city=user.city.lower(),
state=user.state.lower(),
address=user.address,
password=password)
newUser.save(clean=True)
if newUser:
background.add_task(
Fortauto.mailUsr,
email_title=f"Account activation",
user_email=user.email,
email_message=f"Welcome to {WEBSITE_NAME}, "
f"kindly click on the link below to activate your "
f"account\n\n {user.website_url}/{newUser.id}/activate")
return Fortauto.response(
{
"message":
"Account created successfully,"
" an verification link has been sent to your email"
},
status_code=status.HTTP_201_CREATED)
return Fortauto.response({
"message":
"Error Creating account, please check your details or try"
})
except errors.NotUniqueError:
return Fortauto.response(
{"message": "User with email already exist"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.ValidationError:
return Fortauto.response(
{
"message":
"Error validating details, please check your details and try again"
},
status_code=status.HTTP_400_BAD_REQUEST)
except Exception.__base__:
return Fortauto.response(
{
"message":
"Error Creating account, please check your details or try"
},
status_code=status.HTTP_400_BAD_REQUEST)
return Fortauto.response({"message": "Password is does not match"},
status_code=status.HTTP_400_BAD_REQUEST)
async def get_all_user_car_details(carVinNumber: str,
current_user: dict = Depends(
UserMixin.authenticate_user)):
try:
user = User.find_user_with_Id(current_user["id"])
if user:
user_car = user.car_details.filter(vin_number=carVinNumber).first()
if user_car:
user.car_details.remove(user_car)
user.save()
return Fortauto.response({"car_details": carVinNumber},
status_code=status.HTTP_200_OK)
return Fortauto.response(
{"message": "car with this vin doest not exists"},
status_code=status.HTTP_400_BAD_REQUEST)
return Fortauto.response({"message": "user does not exist"},
status_code=status.HTTP_401_UNAUTHORIZED)
except errors.NotUniqueError:
return Fortauto.response(
{"message": "car with this vin number already exists"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.DoesNotExist:
return Fortauto.response({"message": "car does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.ValidationError:
return Fortauto.response({"message": "Error validating user"},
status_code=status.HTTP_400_BAD_REQUEST)
except Exception.__base__:
return Fortauto.response(
{"message": "Error adding car, try again"},
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
async def get_all_user_car_details(current_user: dict = Depends(
UserMixin.authenticate_user)):
try:
user = User.find_user_with_Id(current_user["id"])
if user:
if user.car_details:
user_car = [userCar.to_json() for userCar in user.car_details]
return Fortauto.response({"car_details": user_car},
status_code=status.HTTP_200_OK)
return Fortauto.response(
{"message": "User does not have any car details"},
status_code=status.HTTP_404_NOT_FOUND)
return Fortauto.response({"message": "user does not exist"},
status_code=status.HTTP_401_UNAUTHORIZED)
except errors.NotUniqueError:
return Fortauto.response(
{"message": "car with this vin number already exists"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.DoesNotExist:
return Fortauto.response({"message": "car does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.ValidationError:
return Fortauto.response({"message": "Error validating user"},
status_code=status.HTTP_400_BAD_REQUEST)
except Exception.__base__:
return Fortauto.response(
{"message": "Error getting car car, try again"},
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
async def activate_user(user: UpdateUserInput):
update_user = User.find_user_with_Id(userId=user.userId)
if update_user["active"]:
return Fortauto.response(
{"message": "Account had already been activated"},
status_code=status.HTTP_400_BAD_REQUEST)
if update_user:
update_user.update(active=True)
update_user.save(clean=True)
return Fortauto.response(
{"message": "Account was activated successfully"},
status_code=status.HTTP_200_OK)
return Fortauto.response({"message": "Account does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
def authenticate_user(token: str = Depends(Oauth_schema)):
credentials_exception = HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Could not validate credentials",
headers={"WWW-Authenticate": "Bearer"},
)
try:
payload = jwt.decode(token=token, key=SECRET_KEY)
user: dict = payload.get("user")
if user is None:
raise credentials_exception
currentUser = User.find_user_with_email(user["email"])
if currentUser is None:
raise credentials_exception
return currentUser.to_json()
except JWTError:
raise credentials_exception
async def reset_password(user: PasswordResetInput):
if user.password.strip() == user.confirm_password.strip():
check_user = User.find_user_with_Id(user.userId)
if check_user:
password = UserMixin.hash_password(
plaintext_password=user.password)
if password:
check_user.update(password=password)
check_user.save(clean=True)
return Fortauto.response(
{"message": "password changed successfully"},
status_code=status.HTTP_200_OK)
return Fortauto.response({"message": "Error updating password"},
status_code=status.HTTP_400_BAD_REQUEST)
return Fortauto.response({"message": "Account does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
return Fortauto.response({"message": "Password does not match"},
status_code=status.HTTP_400_BAD_REQUEST)
def getUserAccount(userId: str,
user: dict = Depends(UserMixin.authenticate_user)):
try:
if user["super_admin"]:
user = User.find_user_with_Id(userId)
if user:
return userId
return Fortauto.response({"message": "User does not exist"},
status_code=status.HTTP_401_UNAUTHORIZED)
except errors.DoesNotExist:
return Fortauto.response({"message": "User does not exist"},
status_code=status.HTTP_401_UNAUTHORIZED)
except errors.ValidationError:
return Fortauto.response({"message": "error removing user"},
status_code=status.HTTP_401_UNAUTHORIZED)
except Exception.__base__:
return Fortauto.response(
{"message": "Error removing user"},
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
async def get_user_account_activation_link(background: BackgroundTasks,
user: GetPasswordResetLink):
current_user = User.find_user_with_email(email=user.email)
if current_user:
background.add_task(
Fortauto.mailUsr,
email_title=f"Account activation",
user_email=user.email,
email_message=f"Welcome to {WEBSITE_NAME}, "
f"kindly click on the link below to activate your "
f"account\n\n {user.website_url}/{current_user.id}/activate")
return Fortauto.response(
{
"message":
"Account created successfully,"
"an verification link has been sent to your email"
},
status_code=status.HTTP_201_CREATED)
return Fortauto.response({"message": "Account does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
async def get_password_reset_link(background: BackgroundTasks,
user: GetPasswordResetLink):
current_user = User.find_user_with_email(email=user.email)
if current_user:
background.add_task(
Fortauto.mailUsr,
user_email=user.email,
email_message=
f" Good day {current_user.first_name} {current_user.last_name}, You requested for password reset, "
f"please click\n\n "
f"on to link below to continue \n\n {user.website_url}/{current_user.id}/password/reset",
email_title="Password reset link")
return Fortauto.response(
{
"message":
"password reset link has been sent your email address, click it to continue."
},
status_code=status.HTTP_200_OK)
return Fortauto.response({"message": "Account does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
def loginUserAccount(userIn: UserLoginInput, response: Response) -> Response:
user = User.find_user_with_email(email=userIn.email)
if user:
if user.active:
if UserMixin.check_password(userIn.password, user.password):
encode_jwt_access, encode_jwt_refresh = UserMixin.JwtEncoder(
user=user.to_json())
if encode_jwt_access and encode_jwt_refresh:
response.set_cookie(
key="refresh_token",
value=encode_jwt_refresh,
httponly=True,
max_age=172800,
expires=172800,
)
SuccessResponseData = {
"user": user.to_json(indent=4),
"message": "logged in successfully",
"access_token": encode_jwt_access,
"access_token_type": "Bearer",
"expires": "2 days",
}
return Fortauto.response(SuccessResponseData,
status_code=status.HTTP_200_OK)
ErrorResponseData = {"message": "Password does not match"}
return Fortauto.response(ErrorResponseData,
status_code=status.HTTP_401_UNAUTHORIZED)
ErrorResponseData = {
"message": "Email was sent to you, please verify your email"
}
return Fortauto.response(ErrorResponseData,
status_code=status.HTTP_401_UNAUTHORIZED)
ErrorResponseData = {"message": "Account does not exist"}
return Fortauto.response(ErrorResponseData,
status_code=status.HTTP_401_UNAUTHORIZED)
async def add_user_car_details(car: CarDetail,
current_user: dict = Depends(
UserMixin.authenticate_user)):
try:
user = User.find_user_with_Id(current_user["id"])
if user:
if not user.car_details:
user.car_details.create(**car.dict())
user.save(clean=True)
return Fortauto.response(
{"message": "car details added successfully"},
status_code=status.HTTP_201_CREATED)
user_car = user.car_details.filter(vin_number=car.vin_number)
if not user_car:
user.car_details.create(**car.dict())
user.save(clean=True)
return Fortauto.response(
{"message": "car details added successfully"},
status_code=status.HTTP_201_CREATED)
return Fortauto.response({"message": "car already exists"},
status_code=status.HTTP_400_BAD_REQUEST)
return Fortauto.response({"message": "user does not exist"},
status_code=status.HTTP_401_UNAUTHORIZED)
except errors.NotUniqueError:
return Fortauto.response(
{"message": "car with this vin number already exists"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.DoesNotExist:
return Fortauto.response({"message": "car does not exist"},
status_code=status.HTTP_400_BAD_REQUEST)
except errors.ValidationError:
return Fortauto.response({"message": "Error validating user"},
status_code=status.HTTP_400_BAD_REQUEST)
except Exception.__base__:
return Fortauto.response(
{"message": "Error adding car, try again"},
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)