Merge pull request #86 from resum-ai/feat/resume

Feat/resume

accounts/migrations/0002_customuser_available_chat_count_and_more.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.0.3 on 2024-04-13 08:32
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("accounts", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="customuser",
+            name="available_chat_count",
+            field=models.IntegerField(default=5),
+        ),
+        migrations.AddField(
+            model_name="customuser",
+            name="reset_chat_date",
+            field=models.DateField(blank=True, null=True),
+        ),
+    ]

accounts/models.py

@@ -25,6 +25,9 @@ class CustomUser(AbstractBaseUser, PermissionsMixin):
     is_active = models.BooleanField(default=True)  # 계정 활성화 상태
     created_at = models.DateTimeField(auto_now_add=True)
 
+    available_chat_count = models.IntegerField(default=5)
+    reset_chat_date = models.DateField(null=True, blank=True)
+
     USERNAME_FIELD = "email"
     REQUIRED_FIELDS = []
 

resume/views.py

@@ -1,5 +1,6 @@
 import json
 import logging
+from datetime import datetime
 
 from django.http import Http404
 from django.db.models import Q
@@ -386,6 +387,13 @@ class ChatView(APIView):
         },
     )
     def post(self, request, id):
+        user = request.user
+        today = datetime.now().date()
+
+        # 채팅 횟수 count
+        if user.chat_count <= 0:
+            return JsonResponse({"error": "채팅 횟수가 모두 소진되었습니다."}, status=status.HTTP_403_FORBIDDEN)
+
         query = request.data.get("query", "")
 
         resume = get_object_or_404(Resume, pk=id)
@@ -408,6 +416,8 @@ def post(self, request, id):
             resume=resume, query=query, response=chatbot_response
         )
         new_chat_history.save()
+        user.available_chat_count -= 1
+        user.save()
 
         # 챗봇의 응답을 반환
         return JsonResponse({"answer": chatbot_response}, status=status.HTTP_200_OK)