Merge pull request #106 from snuhcs-course/feat/initialize-weights-ba…

…ckend

Initialize weight table when signing up (Backend)

backend/setup/tests.py

@@ -55,7 +55,7 @@ def test_get_settings_success(self):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         data = response.json()
         self.assertEqual(data['display_mode'], 0)  # 0 is the default value
-        self.assertEqual(data['default_menu'], 0)  # 0 is the default value
+        self.assertEqual(data['default_menu'], 1)  # 1 is the default value
 
 
 

backend/setup/views.py

@@ -33,7 +33,7 @@ def get(self, request):
         if not Setting.objects.filter(user=user).exists():
             response_data = {
                 "display_mode": 0,
-                "default_menu": 0,
+                "default_menu": 1,
                 "updated_at": ""
             }
         else:

backend/user/serializers.py

@@ -5,6 +5,7 @@
 
 from user.models import EmailVerification
 from user.utils import normalize_email
+from weight_table.models import WeightTable
 
 User = get_user_model()
 
@@ -30,6 +31,12 @@ def create(self, validated_data):
         password = validated_data.get('password')
         nickname = validated_data.get('nickname')
         user = User.objects.create_user(nickname, email, password)
+
+        with open('./weight_table.txt', 'r') as file:
+            for symbol_id, line in enumerate(file, start=1):
+                weight = f"[{line}]"
+                WeightTable.objects.create(symbol_id=symbol_id, weight=weight, user=user)
+
         return user
 
 

backend/user/views.py

@@ -1,6 +1,7 @@
 import boto3
 from django.contrib.auth import get_user_model
 from django.core.mail import EmailMessage
+from django.db import transaction
 from rest_framework import status, permissions
 from rest_framework.views import APIView
 from rest_framework.response import Response
@@ -22,6 +23,7 @@
 class UserSignUpView(APIView):
     permission_classes = (permissions.AllowAny,)
 
+    @transaction.atomic
     def post(self, request):
         serializer = UserSignUpSerializer(data=request.data)
         serializer.is_valid(raise_exception=True)