Changes in ow/views/user.py [bddf042:f0e64eb] in OpenWorkouts-current

File:

• ow/views/user.py (modified) (3 diffs)

ow/views/user.py

@@ -3 +3 @@
 from datetime import datetime, timezone, timedelta
 from decimal import Decimal
+from io import BytesIO
 
 from pyramid.httpexceptions import HTTPFound, HTTPNotFound
@@ -11 +12 @@
 from pyramid_simpleform import Form, State
 from pytz import common_timezones
+from PIL import Image
 
 from ..models.user import User
@@ -212 +214 @@
     permission='view')
 def profile_picture(context, request):
-    return Response(
-        content_type='image',
-        body_file=context.picture.open())
+    if context.picture is None:
+        return HTTPNotFound()
+
+    size = request.GET.get('size', 0)
+    # we will need a tuple, it does not matter if both values are the same,
+    # Pillow will keep aspect ratio
+    size = (int(size), int(size))
+
+    image = Image.open(context.picture.open())
+
+    if size > (0, 0) and size < image.size:
+        # resize only if they are asking for smaller size, prevent
+        # someone asking for a "too big" image
+        image.thumbnail(size)
+
+    body_file = BytesIO()
+    image.save(body_file, format=image.format)
+    return Response(content_type='image', body=body_file.getvalue())