Flask-自定义头像

狗书上面的头像是gavatar提供的，可能需要翻墙。因此我需要做一个能够自定义头像的功能。主要参考了https://ncitycode.com/python/flask/2016/6/18/，很快就做出来了。

P1 模型

狗书上的User模型是avatar_hash，这里换成avatar.并且把下面的gavatar()函数删掉，因为我们已经不用它提供的头像了

P2 表单

编辑个人资料的表单添加一个上传文件字段FileField

效果图

P3 View

request.files[‘avatar’]:截取表单的avatar字段文件
UPLOAD_FOLDER：上传的文件最终放置的位置，这里我们在static文件夹里新建一个avatar文件夹，把上传的文件都放在这
ALLOWED_EXTENSIONS:后缀名白名单，这里我们只需要图片的后缀名
flag这一段比较有意思，首先判断文件名中要有’.’，并且将文件名以’.’切割一次，也就是分成两个子串，再选取第2个片段([1])，说这么多就是想截取个后缀名而已。然后看在不在后缀名白名单内。只有flag符合条件，就开始保存文件到UPLOAD_FOLDER里，并且还重新命名，格式的话自己随便取。这里是采取”用户名_文件名”的格式。
之后就是给用户的avatar列赋值了，把上传的图片地址传过来，然后db,session.add/commit

P4修改下模板

在user.html和_posts.html中，里面都含有头像部分，所以要做必要的修改。使用if语句，如果已上传了头像，就用这个头像，如果未上传，就用一个默认头像。默认头像自己放在avatar文件夹里.

user.html

_posts.html