auths.py 3.5 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
from fastapi import Response
from fastapi import Depends, FastAPI, HTTPException, status
from datetime import datetime, timedelta
from typing import List, Union

from fastapi import APIRouter
from pydantic import BaseModel
import time
import uuid

from apps.web.models.auths import (
    SigninForm,
    SignupForm,
14
    UpdatePasswordForm,
15
16
17
18
19
20
21
    UserResponse,
    SigninResponse,
    Auths,
)
from apps.web.models.users import Users


22
from utils.utils import get_password_hash, get_current_user, create_token
Timothy J. Baek's avatar
Timothy J. Baek committed
23
24
from utils.misc import get_gravatar_url
from constants import ERROR_MESSAGES
25
26


Timothy J. Baek's avatar
Timothy J. Baek committed
27
28
router = APIRouter()

29
30
31
32
33
############################
# GetSessionUser
############################


34
@router.get("/", response_model=UserResponse)
35
36
37
38
39
40
41
42
async def get_session_user(user=Depends(get_current_user)):
    return {
        "id": user.id,
        "email": user.email,
        "name": user.name,
        "role": user.role,
        "profile_image_url": user.profile_image_url,
    }
43
44


45
46
47
48
49
############################
# Update Password
############################


50
@router.post("/update/password", response_model=bool)
Anuraag Jain's avatar
Anuraag Jain committed
51
52
53
async def update_password(
    form_data: UpdatePasswordForm, session_user=Depends(get_current_user)
):
54
55
    if session_user:
        user = Auths.authenticate_user(session_user.email, form_data.password)
56

57
58
        if user:
            hashed = get_password_hash(form_data.new_password)
Timothy J. Baek's avatar
Timothy J. Baek committed
59
            return Auths.update_user_password_by_id(user.id, hashed)
60
61
        else:
            raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_PASSWORD)
62
63
64
65
    else:
        raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)


66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
############################
# SignIn
############################


@router.post("/signin", response_model=SigninResponse)
async def signin(form_data: SigninForm):
    user = Auths.authenticate_user(form_data.email.lower(), form_data.password)
    if user:
        token = create_token(data={"email": user.email})

        return {
            "token": token,
            "token_type": "Bearer",
            "id": user.id,
            "email": user.email,
            "name": user.name,
            "role": user.role,
Timothy J. Baek's avatar
Timothy J. Baek committed
84
            "profile_image_url": user.profile_image_url,
85
86
        }
    else:
Timothy J. Baek's avatar
Timothy J. Baek committed
87
        raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)
88
89
90
91
92
93
94
95
96
97
98


############################
# SignUp
############################


@router.post("/signup", response_model=SigninResponse)
async def signup(form_data: SignupForm):
    if not Users.get_user_by_email(form_data.email.lower()):
        try:
99
            role = "admin" if Users.get_num_users() == 0 else "pending"
100
            hashed = get_password_hash(form_data.password)
Timothy J. Baek's avatar
Timothy J. Baek committed
101
102
103
            user = Auths.insert_new_auth(
                form_data.email.lower(), hashed, form_data.name, role
            )
104
105
106
107
108
109
110
111
112
113
114
115

            if user:
                token = create_token(data={"email": user.email})
                # response.set_cookie(key='token', value=token, httponly=True)

                return {
                    "token": token,
                    "token_type": "Bearer",
                    "id": user.id,
                    "email": user.email,
                    "name": user.name,
                    "role": user.role,
Timothy J. Baek's avatar
Timothy J. Baek committed
116
                    "profile_image_url": user.profile_image_url,
117
118
                }
            else:
119
                raise HTTPException(500, detail=ERROR_MESSAGES.CREATE_USER_ERROR)
120
121
122
        except Exception as err:
            raise HTTPException(500, detail=ERROR_MESSAGES.DEFAULT(err))
    else:
Timothy J. Baek's avatar
Timothy J. Baek committed
123
        raise HTTPException(400, detail=ERROR_MESSAGES.EMAIL_TAKEN)