79 lines
3.2 KiB
Python
79 lines
3.2 KiB
Python
import io
|
|
from flask import Blueprint, make_response, render_template, redirect, url_for, flash
|
|
from werkzeug.http import http_date
|
|
from app.models import Profile, db
|
|
from app.forms import ProfileForm
|
|
from flask_login import login_required, current_user
|
|
import base64
|
|
from datetime import datetime
|
|
from PIL import Image
|
|
|
|
user = Blueprint('user', __name__)
|
|
|
|
@user.route('/profile', methods=['GET', 'POST'])
|
|
@login_required
|
|
def profile():
|
|
profile = current_user.profile or Profile(user_id=current_user.id)
|
|
form = ProfileForm(obj=profile)
|
|
|
|
if form.validate_on_submit():
|
|
# Update profile fields
|
|
profile.name = form.name.data
|
|
profile.email = form.email.data
|
|
profile.systolic_threshold = form.systolic_threshold.data or profile.systolic_threshold
|
|
profile.diastolic_threshold = form.diastolic_threshold.data or profile.diastolic_threshold
|
|
profile.dark_mode = form.dark_mode.data
|
|
profile.timezone = form.timezone.data
|
|
|
|
# Handle profile picture upload
|
|
if form.profile_pic.data:
|
|
file_data = form.profile_pic.data.read()
|
|
|
|
# Resize and compress the image
|
|
try:
|
|
image = Image.open(io.BytesIO(file_data))
|
|
image = image.convert("RGB") # Ensure it's in RGB format
|
|
image.thumbnail((300, 300)) # Resize to a maximum of 300x300 pixels
|
|
|
|
# Save the resized image to a buffer
|
|
buffer = io.BytesIO()
|
|
image.save(buffer, format="JPEG", quality=80) # Compress with quality=80
|
|
buffer.seek(0)
|
|
|
|
# Encode the compressed image as base64
|
|
profile.profile_pic = base64.b64encode(buffer.read()).decode('utf-8')
|
|
except Exception as e:
|
|
flash(f"Error processing profile picture: {e}", 'danger')
|
|
|
|
|
|
db.session.add(profile)
|
|
db.session.commit()
|
|
flash('Profile updated successfully!', 'success')
|
|
return redirect(url_for('user.profile'))
|
|
|
|
return render_template('profile.html', form=form, profile=profile)
|
|
|
|
@user.route('/profile/image/<int:user_id>')
|
|
def profile_image(user_id):
|
|
# Ensure the reading belongs to the logged-in user
|
|
if user_id != current_user.id:
|
|
flash('You are not authorized to delete this reading.', 'danger')
|
|
return redirect(url_for('main.dashboard'))
|
|
|
|
profile = Profile.query.filter_by(user_id=user_id).first()
|
|
if profile and profile.profile_pic:
|
|
image_data = base64.b64decode(profile.profile_pic)
|
|
response = make_response(image_data)
|
|
response.headers.set('Content-Type', 'image/jpeg')
|
|
response.headers.set('Cache-Control', 'public, max-age=86400') # Cache for 1 day
|
|
response.headers.set('ETag', str(hash(profile.profile_pic))) # Unique ETag for the image
|
|
response.headers.set('Last-Modified', http_date(datetime.utcnow().timestamp()))
|
|
|
|
return response
|
|
else:
|
|
# Serve the default SVG if no profile picture is found
|
|
with open('app/static/images/default-profile.svg', 'r') as f:
|
|
default_image = f.read()
|
|
|
|
response = make_response(default_image)
|
|
response.headers.set('Content-Type', 'image/svg+xml') |