Skip to content

Commit

Permalink
Refactor: 이미지 조회 시에 태그를 요구하지 않게 수정정
Browse files Browse the repository at this point in the history
  • Loading branch information
joy-river committed Dec 29, 2024
1 parent 27d109c commit 33fecb3
Showing 1 changed file with 11 additions and 4 deletions.
15 changes: 11 additions & 4 deletions KUA/student/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -548,11 +548,18 @@ def get(self, request, *args, **kwargs):
if name:
name = re.sub(r'\d+$', '', name) # Remove trailing digits

if name and tag:
if name:
try:
image = models.Image.objects.get(name=name, tag=tag)
serializer = self.serializer_class(image)
return Response(serializer.data, status=200)
if tag:
images = models.Image.objects.filter(name=name, tag=tag)
else:
images = models.Image.objects.filter(name=name)

if images.exists():
serializer = self.serializer_class(images, many=True)
return Response(serializer.data, status=200)
else:
return Response({"error": "Image not found."}, status=404)
except models.Image.DoesNotExist:
return Response({"error": "Image not found."}, status=404)
else:
Expand Down

0 comments on commit 33fecb3

Please sign in to comment.