mgt c6 checkpoint 2.2

backend/app/routes.py

@@ -1,9 +1,11 @@
 from flask import render_template, flash, redirect, url_for, request
 from urllib.parse import urlsplit
+from datetime import datetime, timezone
+
 from app import app, db
-from app.forms import LoginForm, RegistrationForm
-from flask_login import current_user, login_user, logout_user, login_required
+from app.forms import LoginForm, RegistrationForm, EditProfileForm
 import sqlalchemy as sa
+from flask_login import current_user, login_user, logout_user, login_required
 from app.models import User
 
 @app.route('/')
@@ -25,6 +27,12 @@ def index():
     #return posts;
     return render_template('index.html', title='Home', posts=posts)
 
+@app.before_request
+def before_request():
+    if current_user.is_authenticated:
+        current_user.last_seen = datetime.now(timezone.utc)
+        db.session.commit()
+
 @app.route('/login', methods=['GET', 'POST'])
 def login():
     if current_user.is_authenticated:
@@ -57,7 +65,7 @@ def register():
         user.set_password(form.password.data)
         db.session.add(user)
         db.session.commit()
-        user.gen_avatar()
+        #user.gen_avatar()
         flash('User has been created.')
         return redirect(url_for('login'))
     return render_template('register.html', title='Register', form=form)
@@ -71,3 +79,19 @@ def user(username):
         {'author': user, 'body': 'Test2?'}
     ]
     return render_template('user.html', user=user, posts=posts)
+
+@app.route('/edit_profile', methods=['GET', 'POST'])
+@login_required
+def edit_profile():
+    form = EditProfileForm()
+    if form.validate_on_submit():
+        current_user.username = form.username.data
+        current_user.about_me = form.about_me.data
+        db.session.commit()
+        flash('Profile changes have been saved.')
+        return redirect(url_for('edit_profile'))
+    elif request.method == 'GET':
+        form.username.data = current_user.username
+        form.about_me.data = current_user.about_me()
+    return render_template('edit_profile.html', title='Edit Profile', form=form)
+