diff --git a/KUA/student/views.py b/KUA/student/views.py index 05fc521..303873b 100644 --- a/KUA/student/views.py +++ b/KUA/student/views.py @@ -8,6 +8,7 @@ from . import serializers import string import random +import re from django.core.mail import EmailMessage from django.utils import timezone from rest_framework.authentication import TokenAuthentication @@ -18,6 +19,7 @@ from drf_yasg import openapi from rest_framework.parsers import MultiPartParser, FormParser + # 이메일 코드 전송 기능 @@ -543,6 +545,9 @@ def get(self, request, *args, **kwargs): name = request.query_params.get('name') tag = request.query_params.get('tag') + if name: + name = re.sub(r'\d+$', '', name) # Remove trailing digits + if name and tag: try: image = models.Image.objects.get(name=name, tag=tag)