Changeset f0e64eb in OpenWorkouts-current for ow/views/user.py

Timestamp:

Feb 15, 2019, 11:00:35 AM (5 years ago)

Author:

Borja Lopez <borja@…>

Branches:

current, feature/docs, master

Children:

0fa21cd

Parents:

eb20dc8

Message:

(#51) Fixed profile images were too big:

• Added new dependency (Pillow)

• Modified the user profile picture view. Now it accepts a GET parameter (size) telling the size of the image we want. Only one value is needed, the code will scale the image appropiately.

• Modified the dashboard, profile and edit_profile templates to ask for a smaller version (200px) of the user profile picture.

IMPORTANT: Ensure you install Pillow in any existing environments after pulling:

pip install Pillow

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())