django - 인증, 인가 실습

password = 'user1234'
encoded_password = password.encode('utf-8')

b'user1234'

import bcrypt

hashed_password = bcrypt.hashpw(encoded_password, bcrypt.gensalt())

b'$2b$12$PS77USI4OnMydAuPoPCoK.9rCCOghSmMrR82Xn6L3x2xRYaeA7i8m'

decoded_hashed_password = hash_password.decode('utf-8')

$2b$12$Q.bIZfHuJPgPICdJbcO8r.6xHBhXuhNXYSuSfQuOQXPV/c.5hF38K

import json
from django.views import View
from django.http import JsonResponse, HttpResponse
from .models import Users

import bcrypt
import jwt

class MainView(View):
  def get(self, request):
    return JsonResponse({'Welcome to':'Westagram', 'Sign-up':'/users/sign-up', 'Log-in':'/users/log-in'}, status=200)

class SignUpView(View):
  def post(self, request):
    # 가입 시 유저가 입력한 아이디, 비밀번호 값을 request body를 통해 받는다.
    data = json.loads(request.body)
    
    # 유저가 입력한 비밀번호값 data['password']을 byte 타입으로 인코딩 한 후
    # bcrypt를 이용해 hasing과 salting을 해준다.
    hased_pw = bcrypt.hashpw(data['password'].encode('utf-8'), bcrypt.gensalt())
    # 해싱한 암호를 다시 디코딩 해서 DB에 저장 가능한 형태로 만들어준다.
    decoded_hashed_pw = hased_pw.decode('utf-8')

    try:
      if data['username'] == '':
        return JsonResponse({'message':'ID_IS_REQUIRED'}, status=401)
            
      elif data['password'] == '':
        return JsonResponse({'message':'PASSWORD_IS_REQUIRED'}, status=401)
            
      elif Users.objects.filter(username=data['username']).exists():
        return JsonResponse({'message':'ID_EXISTS'}, status=409)

      # 유저가 입력한 아이디가 기존 아이디와 겹치지 않는다면 if문 안 코드를 실행한다.
      elif not Users.objects.filter(username=data['username']).exists():
        # 유저가 입력한 아이디는 DB에 그대로 저장해주고, 비밀번호는 위에서 암호화 한 형태로 저장한다.
        Users(
          username = data['username'],
          password = decoded_hashed_pw
        ).save()
        return JsonResponse({'message':'WELCOME'}, status=200)

    except:
      return JsonResponse({'message':'INVALID_ID'}, status=401)

  # get 메서드 생략

class LogInView(View):
  def post(self, request):
    # 로그인 시 유저가 입력한 아이디, 비밀번호 값을 request body를 통해 받는다.
    data = json.loads(request.body)

    try:
      # 로그인 시 입력한 아이디 값 data['username']에 맞는 유저 정보가 DB에 존재한다면 if문 안 코드를 실행한다.
      if Users.objects.filter(username=data['username']).exists():

        # 객체 형태로 바로 리턴되는 get을 이용해 유저가 입력한 아이디 값에 해당하는 객체를 구해준다.
        user_id = Users.objects.get(username=data['username'])

        # 유저가 입력한 비밀번호와 DB에 저장된 비밀번호가 일치하는지 비교해준다.
        # 비교는 둘 다 byte 타입인 상태에서 해야한다.
        if bcrypt.checkpw(data['password'].encode('utf-8'), user_id.password.encode('utf-8')) == True:
          # user_id 객체의 PK값인 id를 특정 알고리즘을 사용해 암호화하여 토큰을 생성한다.
          # 여기서의 id는 유저 아이디가 아니라 DB 테이블 상 고유값 id다.
          access_token = jwt.encode({'id':user_id.id}, 'secret', algorithm='HS256')
          # 정상적으로 동작하면 토큰과 함께 status code 200을 리턴한다.
          return JsonResponse({'token': access_token.decode('utf-8')}, status=200)

        else:
          return JsonResponse({'message':'비밀번호가 틀립니다.'}, status=401)
          
      else:
        return JsonResponse({'message':'아이디가 없습니다.'}, status=401)
        
    except:
      return JsonResponse({'message':'INVALID_USER'}, status=401)
  
  # get 메서드 생략