장고로 쇼핑몰 만들기(코드 리펙토링)

• 기존에 있던 form에서 데이터를 저장하는 부분은 view로 옮기는 코드 리펙토링 작업을 했다.

• form에는 값의 유효성 검사정도 하고 데이터를 저장하는 부분은 view단에 하는 것을 권장한다고 한다.

User form 변화

from django import forms
from .models import Fcuser
from django.contrib.auth.hashers import check_password

class RegisterForm(forms.Form):
    email = forms.EmailField(
        error_messages={
            'required' : '이메일을 입력해주세요.'
        },
        max_length=64, label='이메일'
    )
    password = forms.CharField(
        error_messages={
            'required' : '비밀번호를 입력해주세요.'
        },
        widget=forms.PasswordInput, label='비밀번호'
    )
    re_password = forms.CharField(
        error_messages={
            'required' : '비밀번호를 입력해주세요.'
        },
        widget=forms.PasswordInput, label='비밀번호 확인'
    )

    def clean(self):
        # super()는 메소드 오버라이딩 
        cleaned_data = super().clean()  # 값이 들어있는지 검사한다. 들어잇으면 아래 변수를 할당, 없으면 실패하고 튕군다. 
        email = cleaned_data.get('email')
        password = cleaned_data.get('password')
        re_password = cleaned_data.get('re_password')

        if password and re_password:
            if password != re_password:
                self.add_error('password', '비밀번호가 서로 다릅니다.')
                self.add_error('re_password', '비밀번호가 서로 다릅니다.')
         
         #else:
         #       fcuser = Fcuser(
         #           email=email,
         #           password=make_password(password) 
         #       )
         #       fcuser.save()
         
         '기존 코드에서 회원가입한 회원을 저장하는 코드가 form에서 빠졌다.' 
           

class LoginForm(forms.Form):
    email = forms.EmailField(
        error_messages={
            'required' : '이메일을 입력해주세요.'
        },
        max_length=64, label='이메일'
    )
    password = forms.CharField(
        error_messages={
            'required' : '비밀번호를 입력해주세요.'
        },
        widget=forms.PasswordInput, label='비밀번호'
    )
    
    def clean(self):
        # super()는 메소드 오버라이딩 
        cleaned_data = super().clean()  # 값이 들어있는지 검사한다. 들어잇으면 아래 변수를 할당, 없으면 실패하고 튕군다. 
        email = cleaned_data.get('email')
        password = cleaned_data.get('password')

        if email and password:
            try:
                fcuser = Fcuser.objects.get(email=email)
            except Fcuser.DoesNotExist:
                self.add_error('email', '아이디가 없습니다.')
                return
            if not check_password(password, fcuser.password):
                self.add_error('password', '비밀번호를 틀렸습니다.')
           # self.email = fcuser.email
           '이 부분도 뷰단으로 넘어갔다.'
           '여기서는 email이 정의가 되어있다'

• 데이터를 저장하는 부분은 제거 되었다.
• self.email = fcuser.email사라진 코드이다. 여기서는 email이 변수에 할당 되어있지만 view단으로 옮기면 정의 되어있지 않아서 따로 받아줘야한다.

User view 변화

from django.shortcuts import render, redirect
from django.views.generic.edit import FormView 
from .forms import RegisterForm, LoginForm
from django.contrib.auth.hashers import make_password
from .models import Fcuser

# Create your views here.

def index(request):
    return render(request, 'index.html', {'email': request.session.get('user')})

class RegisterView(FormView):
    template_name = 'register.html'
    form_class = RegisterForm
    success_url = '/'  

# form_valid 함수가 추가 되었다. 
    def form_valid(self,form):     # form_valid 함수를 오버라이딩 							
        fcuser = Fcuser(
            email=form.data.get('email'),
            password=make_password(form.data.get('password')), 
            level = 'user'
        )
        fcuser.save()

        return super().form_valid(form)


class LoginView(FormView):
    template_name = 'login.html'
    form_class = LoginForm
    success_url = '/'  

# 여기도 마찬가지로 추가되었다.
    def form_valid(self, form):
        self.request.session['user'] = form.data.get('email')  # form_valid 함수를 통과하면 이 조건 실행 

        return super().form_valid(form)

def logout(request):
    if 'user' in request.session:
        del(request.session['user'])

    return redirect('/')

• 회원가입, 로그인 함수에 form_valid 함수를 오버라이딩 해서
  입력받은 값을 검사하고 데이터에 저장하는 로직을 넣었다.

• form.data.get('email') 이렇듯 form에 저장했던 변수가 없어졌으므로 입력받은 값에서 email을 뽑아온다.

• 대부분의 값들을 data.get으로 뽑아온다.

product form 변화

rom django import forms
from .models import Product 

class RegisterForm(forms.Form):
    name = forms.CharField(
        error_messages={
            'required' : '상품명을 입력해주세요.'
        },
        max_length=64, label='상품명'
    )
    price = forms.IntegerField(
        error_messages={
            'required' : '상품가격을 입력해주세요.'
        },
        label='상품가격'
    )
    description = forms.CharField(
        error_messages={
            'required' : '상품설명을 입력해주세요.'
        },
        label='상품설명'
    )
    stock = forms.IntegerField(
        error_messages={
            'required' : '재고를 입력해주세요.'
        },
        label='재고'
    )

    def clean(self):
        cleaned_data = super().clean()
        name = cleaned_data.get('name')
        price = cleaned_data.get('price')
        description = cleaned_data.get('description')
        stock = cleaned_data.get('stock')

       # if name and price and description and stock:
       #     product = Product(
       #         name=name,
       #         price=price,
       #         description=description,
       #         stock=stock
       #     )
       #     product.save()

• User 동일하게 상품 데이터를 저장하는 코드는 사라졌다.

product view 변화

@method_decorator(admin_required, name='dispatch')
class ProductCreate(FormView):
    template_name = 'register_product.html'
    form_class = RegisterForm
    success_url = '/product/'
	
    # 추가된 부분 
    def form_valid(self, form):
        product = Product(
                name=form.data.get('name'),
                price=form.data.get('price'),
                description=form.data.get('description'),
                stock=form.data.get('stock')
            )
        product.save()
        
        return super().form_valid(form)

• User와 동일하다.

• return super().form_valid(form) 오버라이딩 했기 때문에 super()함수를 사용한다.

• 저장하기 때문에 return값을 넣어준다.

Order form 변화

from django import forms
from .models import Order 
from fcuser.models import Fcuser
from product.models import Product
from django.db import transaction

class RegisterForm(forms.Form):
    def __init__(self, request, *args, **kwargs):
        super().__init__(*args, **kwargs)           # form에 request를 전달 받기 위해 만들었다. 
        self.request = request
        

    quantity = forms.IntegerField(
        error_messages={
            'required' : '수량을 입력해주세요.'
        }, label='수량'
    )
    product = forms.IntegerField(
        error_messages={
            'required' : '상품가격을 입력해주세요.'
        }, label='상품가격', widget=forms.HiddenInput # 화면에는 보이지 않는 인풋
    )
   

    def clean(self):
        cleaned_data = super().clean()
        quantity = cleaned_data.get('quantity')
        product = cleaned_data.get('product')
        fcuser = self.request.session.get('user')

        #if quantity and product and fcuser:
        #    with transaction.atomic():
        #        prod = Product.objects.get(pk=product)
        #        order = Order(
        #            quantity=quantity,
        #            product=prod,               
                   # fcuser=Fcuser.objects.get(email=fcuser)
        #        )
        #        order.save()
        #        prod.stock -= quantity
        #        prod.save()

        #else: 
         #   self.product = product
         #   self.add_error('quantity', '수량을 입력해주세요')
         #   self.add_error('product', '값을 입력해주세요.')
         if not (quantity and product):                        
            self.add_error('quantity', '수량을 입력해주세요')
            self.add_error('product', '값을 입력해주세요.')
    

• 저장하는 부분이 잘렸다.
• 값이 없는 조건문은 if not (quantity and product):로 표현했다.

order view 변화

@method_decorator(login_required, name='dispatch')
class OrderCreate(FormView): 
    form_class = RegisterForm
    success_url = '/product/'

# 추가된 함수 
    def form_valid(self, form):
        with transaction.atomic():
            prod = Product.objects.get(pk=form.data.get('product'))
            order = Order(
                quantity=form.data.get('quantity'),
                product=prod,               # 주문한 상품의 객체를 주문모델의 상품필드에 저장         
                fcuser=Fcuser.objects.get(email=self.request.session.get('user'))
            )
            order.save()
            prod.stock -= int(form.data.get('quantity'))
            prod.save()

        return super().form_valid(form)

• 동일하게 form_valid함수를 오버라이딩 했다.
• prod.stock -= int(form.data.get('quantity')) 정수형으로 바꾸어 주었다.