我们经常在一些网站上看到，在用户没有自定义头像的情况下，会给每个用户都生成一个头像，这让网站显得更美观，那这个是怎么实现的呢？在Flask中有一个插件，叫做Flask-avatars，专门提供头像解决方案。里面集成了各种头像解决方案。下面就来讲解一下。
一、安装
pip install flask-avatars
二、初始化
扩展需要以通常的方式初始化，然后才能使用：
from flask_avatars import Avatars
app = Flask(__name__)
avatars = Avatars(app)
三、配置
下面列出了可用的配置选项：
四、头像
Flask-Avatarsavatars在模板上下文中提供了一个对象，您可以使用它来获取头像 URL。
1. Gravatar：
关于什么是Gravatar，这里引用了维基百科的一段介绍供读者参考：
在Gravatar上，用户可以用他们的电子邮件注册一个帐号，并且上传一个与之绑定的头像。许多流行的博客程序都支持Gravatar,包括WordpressTypecho等著名博客程序，当用户发布一个评论并填写了他的电子邮件地址时，博客程序会自动查找在Gravatar上是否有与之绑定的头像。如果有，则这个头像将会与评论一起显示出来。WordPress v2.5 开始原生地提供对Gravatar的支持。此外还有许多程序通过插件来支持Gravatar，例如论坛程序Discuz!。
一个Gravatar头像可以使用高达512像素的图片，并且默认地以80*80的尺寸显示出来。如果上传的头像不是这个尺寸，Gravatar会对头像进行缩放。用户可以按照MPAA分级制度确定自己的头像级别，这样网站管理员可以在他们的站点上显示合适的头像。
为了防止用户的电子邮件地址遭到泄漏而收到大量垃圾邮件，Gravatar在传递用户的邮箱地址时采用的是通过MD5散列运算的邮件地址。
更多关于Gravatar的介绍请参考：https://zh.wikipedia.org/wiki/Gravatar
您可以使用avatars.gravatar()获取Gravatar提供的头像 URL ，传递电子邮件哈希：
<img src="{{ avatars.gravatar(email_hash) }}"/>
您可以像这样获得电子邮件哈希：
import hashlib
avatar_hash = hashlib.md5(my_email.lower().encode('utf-8')).hexdigest()
2. 标识生成
Flask-Avatars 提供了一个Identicon生成identicon avatar的类，大部分代码基于randomavatar。首先，您需要设置配置变量AVATARS_SAVE_PATH来告诉 Flask-Avatars 保存生成的头像的路径。一般来说，我们会在创建用户记录的时候生成头像，所以生成头像最好的地方是在用户数据库模型类中：
首先在exts中新建avatars对象
from flask_avatars import Avatars
avatars = Avatars()
在app.py中
注册的时候就保存到数据库
在配置文件config中保存头像的路径
# 头像配置
AVATARS_SAVE_PATH = os.path.join(BASE_DIR, "media", "avatars")
在项目的根路径下新建media/avatars 用以存放所有头像
然后在apps中新建media包/views文件
注册保存头像
用户上传自定义图像
@bp.post("/avatar/upload")
@login_required
def upload_avatar():
form = UploadImageForm(request.files)
if form.validate():
image = form.image.data
# 不要使用用户上传上来的文件名，否则容易被黑客攻击
filename = image.filename
# xxx.png,xx.jpeg 分割拿到后缀名
_, ext = os.path.splitext(filename)
filename = md5((g.user.email + str(time.time())).encode("utf-8")).hexdigest() + ext
image_path = os.path.join(current_app.config['AVATARS_SAVE_PATH'], filename)
image.save(image_path)
# 看个人需求，是否图片上传完成后要立马修改用户的头像字段
g.user.avatar = filename
db.session.commit()
return restful.ok(data={"avatar": filename})
else:
message = form.messages[0]
return restful.params_error(message=message)
class UploadImageForm(BaseForm):
image = FileField(validators=[FileAllowed(['jpg', 'jpeg', 'png'], message="图片格式不符合要求！"), FileSize(max_size=1024*1024*5, message="图片最大不能超过5M！")])
个性签名
@bp.post("/profile/edit")
@login_required
def edit_profile():
form = EditProfileForm(request.form)
if form.validate():
signature = form.signature.data
g.user.signature = signature
db.session.commit()
return restful.ok()
else:
return restful.params_error(message=form.messages[0])
class EditProfileForm(BaseForm):
signature = StringField(validators=[Length(min=1, max=50, message="个性签名长度在1-50字之间！")])
命令行自定义命令的初始化
增加板块的模型类
from exts import db
from datetime import datetime
# from sqlalchemy_serializer import SerializerMixin
# 帖子板块模型类
class BoardModel(db.Model):
serialize_only = ("id", "name", "priority", "create_time")
__tablename__ = "board"
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
name = db.Column(db.String(20), unique=True)
# 板块权重
priority = db.Column(db.Integer, default=1)
create_time = db.Column(db.DateTime, default=datetime.now)
# 帖子
class PostModel(db.Model):
serialize_only = ("id", "title", "content", "create_time", "board", "author")
__tablename__ = "post"
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
title = db.Column(db.String(200), nullable=False)
content = db.Column(db.Text, nullable=False)
create_time = db.Column(db.DateTime, default=datetime.now)
# 帖子板块外键
board_id = db.Column(db.Integer, db.ForeignKey("board.id"))
# 作者
author_id = db.Column(db.String(100), db.ForeignKey("user.id"))
#通过帖子属性可以直接访问所属板块 反向引用 有板块对象可以访问板块下所有模型
board = db.relationship("BoardModel", backref=db.backref("posts"))
author = db.relationship("UserModel", backref=db.backref("posts"))
# 轮播图
class BannerModel(db.Model):
__tablename__ = 'banner'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
# 图片名字
name = db.Column(db.String(255), nullable=False)
# 图片链接
image_url = db.Column(db.String(255), nullable=False)
# 跳转链接
link_url = db.Column(db.String(255), nullable=False)
# 优先级
priority = db.Column(db.Integer, default=0)
create_time = db.Column(db.DateTime, default=datetime.now)
# 评论
class CommentModel(db.Model):
__tablename__ = 'comment'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
content = db.Column(db.Text, nullable=False)
create_time = db.Column(db.DateTime, default=datetime.now)
# 所属帖子的id
post_id = db.Column(db.Integer, db.ForeignKey("post.id"))
# 作者的id
author_id = db.Column(db.String(100), db.ForeignKey("user.id"), nullable=False)
# 评论排序 帖子删除评论也删除
post = db.relationship("PostModel", backref=db.backref('comments', order_by="CommentModel.create_time.desc()", cascade="delete, delete-orphan"))
author = db.relationship("UserModel", backref='comments')
获取板块数据渲染模板
boards = BoardModel.query.order_by(BoardModel.priority.desc()).all()
