#71 fix : modify get location function (#89)

0321minji · Aug 21, 2024 · 800f91d · 800f91d
1 parent cf390ae
commit 800f91d
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/lands/views.py b/lands/views.py
@@ -47,8 +47,21 @@ def get_item_image(self, obj):
 
         def get_locations(self, obj):
             user_email = self.context.get('user_email')
-            locations = obj.locations.get(land__user__email=user_email)
-            return LandApi.LocationSerializer(locations).data
+            try:
+                location, created = obj.locations.get_or_create(
+                    land__user__email=user_email,
+                    defaults={
+                        'x': 0,
+                        'y': 0,
+                        'z': 0,
+                        'show': False
+                    }
+                )
+            except Exception as e:
+                # 예외가 발생하면 적절한 에러 메시지를 반환합니다.
+                return {'error': str(e)}
+
+            return LandApi.LocationSerializer(location).data
 
     class LandItemOutputSerializer(serializers.Serializer):
         land = serializers.SerializerMethodField()