Revision 75768d0e api/util.py
b/api/util.py | ||
---|---|---|
121 | 121 |
|
122 | 122 |
try: |
123 | 123 |
image_id = int(image_id) |
124 |
return Image.objects.get(id=image_id, owner=owner) |
|
124 |
image = Image.objects.get(id=image_id) |
|
125 |
if not image.public and image.owner != owner: |
|
126 |
raise ItemNotFound('Image not found.') |
|
127 |
return image |
|
125 | 128 |
except ValueError: |
126 | 129 |
raise BadRequest('Invalid image ID.') |
127 | 130 |
except Image.DoesNotExist: |
Also available in: Unified diff