Avatars management with Flask-Avatars

albumy/__init__.py

@@ -14,7 +14,7 @@
 from albumy.blueprints.auth import auth_bp
 from albumy.blueprints.main import main_bp
 from albumy.blueprints.user import user_bp
-from albumy.extensions import bootstrap, db, login_manager, mail, dropzone, moment, csrf
+from albumy.extensions import bootstrap, db, login_manager, mail, dropzone, moment, avatars, csrf
 from albumy.models import Role, User, Photo, Permission
 from albumy.settings import config
 
@@ -44,6 +44,7 @@ def register_extensions(app):
     mail.init_app(app)
     dropzone.init_app(app)
     moment.init_app(app)
+    avatars.init_app(app)
     csrf.init_app(app)
 
 

albumy/blueprints/main.py

@@ -7,7 +7,7 @@
 """
 import os
 
-from flask import render_template, current_app, request, Blueprint
+from flask import render_template, current_app, send_from_directory, request, Blueprint
 from flask_login import login_required, current_user
 
 from albumy.decorators import confirm_required, permission_required
@@ -28,6 +28,11 @@ def explore():
     return render_template('main/explore.html')
 
 
+@main_bp.route('/avatars/<path:filename>')
+def get_avatar(filename):
+    return send_from_directory(current_app.config['AVATARS_SAVE_PATH'], filename)
+
+
 @main_bp.route('/upload', methods=['GET', 'POST'])
 @login_required
 @confirm_required

albumy/extensions.py

@@ -5,6 +5,7 @@
     :copyright: © 2018 Grey Li <withlihui@gmail.com>
     :license: MIT, see LICENSE for more details.
 """
+from flask_avatars import Avatars
 from flask_bootstrap import Bootstrap
 from flask_dropzone import Dropzone
 from flask_login import LoginManager, AnonymousUserMixin
@@ -20,6 +21,7 @@
 mail = Mail()
 dropzone = Dropzone()
 moment = Moment()
+avatars = Avatars()
 csrf = CSRFProtect()
 
 

albumy/models.py

@@ -8,6 +8,7 @@
 from datetime import datetime
 
 from flask import current_app
+from flask_avatars import Identicon
 from flask_login import UserMixin
 from werkzeug.security import generate_password_hash, check_password_hash
 
@@ -66,6 +67,9 @@ class User(db.Model, UserMixin):
     bio = db.Column(db.String(120))
     location = db.Column(db.String(50))
     member_since = db.Column(db.DateTime, default=datetime.utcnow)
+    avatar_s = db.Column(db.String(64))
+    avatar_m = db.Column(db.String(64))
+    avatar_l = db.Column(db.String(64))
 
     confirmed = db.Column(db.Boolean, default=False)
 
@@ -76,6 +80,7 @@ class User(db.Model, UserMixin):
 
     def __init__(self, **kwargs):
         super(User, self).__init__(**kwargs)
+        self.generate_avatar()
         self.set_role()
 
     def set_password(self, password):
@@ -92,6 +97,14 @@ def set_role(self):
     def validate_password(self, password):
         return check_password_hash(self.password_hash, password)
 
+    def generate_avatar(self):
+        avatar = Identicon()
+        filenames = avatar.generate(text=self.username)
+        self.avatar_s = filenames[0]
+        self.avatar_m = filenames[1]
+        self.avatar_l = filenames[2]
+        db.session.commit()
+
     @property
     def is_admin(self):
         return self.role.name == 'Administrator'

albumy/settings.py

@@ -49,6 +49,9 @@ class BaseConfig:
 
     SQLALCHEMY_TRACK_MODIFICATIONS = False
 
+    AVATARS_SAVE_PATH = os.path.join(ALBUMY_UPLOAD_PATH, 'avatars')
+    AVATARS_SIZE_TUPLE = (30, 100, 200)
+
     MAIL_SERVER = os.getenv('MAIL_SERVER')
     MAIL_PORT = 465
     MAIL_USE_SSL = True

albumy/templates/base.html

@@ -48,7 +48,9 @@
                     <div class="dropdown nav-item">
                         <a href="#" class="nav-link dropdown-toggle" data-toggle="dropdown" role="button"
                            aria-expanded="false">
-                            {{ current_user.name }}<span class="caret"></span>
+                            <img class="avatar-xs"
+                                 src="{{ url_for('main.get_avatar', filename=current_user.avatar_s) }}">
+                            <span class="caret"></span>
                         </a>
                         <div class="dropdown-menu dropdown-menu-right" role="menu">
                             <h6 class="dropdown-header">Logged in as {{ current_user.username }}</h6>