Flask从入门到做出一个博客的大型教程(五)

在开始之前，先来看下项目的整体结构。

flask/
├── app
│   ├── forms.py
│   ├── __init__.py
│   ├── models.py
│   ├── routes.py
│   └── templates
│       ├── base.html
│       ├── index.html
│       ├── login.html
│       ├── register.html
│       └── test.html
├── config.py
├── migrations
│   ├── alembic.ini
│   ├── env.py
│   ├── README
│   ├── script.py.mako
│   └── versions
│       ├── 063e1979f08c_questions.py
├── question.py

6 用户个人资料页

登录模块完成了，接下来做一做用户个人资料页的显示。

app/routes.py : 添加用户方法

@app.route('/user/<username>')
@login_required
def user(username):user = User.query.filter_by(username=username).first_or_404()posts = [{'author':user,'body':'测试Post #1号'},{'author':user,'body':'测试Post #2号'}]return render_template('user.html',user=user,posts=posts)

既然添加了方法，肯定要添加对应的模板了。

app/templates/user.html : 用户个人中心模板

{% extends "base.html" %}{% block content %}<h1>用户: {{ user.username }}</h1><hr>{% for post in posts %}<p>{{ post.author.username }} 说: <b>{{ post.body }}</b></p>{% endfor %}
{% endblock %}

对基类模板进行小小的改动，当用户是未登录状态看不到用户中心的超链接，登录后看到。这会很有感意思。

app/templates/base.html : 完善基类模板

  <div>博客 :<a href="{{ url_for('index') }}">首页</a>{% if current_user.is_anonymous %}<a href="{{ url_for('login') }}">登录</a>{% else %}<a href="{{ url_for('user', username=current_user.username) }}">用户中心</a><a href="{{ url_for('logout') }}">退出</a>{% endif %}</div>

好了，现在登录看看效果吧！

用户头像，我使用的是Gravatar来提供，使用这个网站生成头像是https:// s.gravatar.com/avatar/(hash) 这种格式，会把邮箱地址用md5加密拼接在后面。如果感兴趣可以去上传一个玩一玩，如果不想去的话可以直接使用我提供的。https://www.gravatar.com/avatar/6b541a0a667f5558208aad7309c22936，默认像素是80x80，但是你可以在这个地址后面添加参数?s=128，会变成128x128像素，当然你可以尝试改成更大的数字也可以，很是方便。可以根据你的需求改变大小。接下来就来驶入它吧！

app/models.py : 添加avatar地址

from hashlib import md5
# ...class User(UserMixin, db.Model):# ...def avatar(self, size):digest = md5(self.email.lower().encode('utf-8')).hexdigest()return 'https://www.gravatar.com/avatar/{}?d=identicon&s={}'.format(digest, size)

方法写好了，接下来就修改模板，在网页中显示吧！

app/templates/user.html : 显示用户头像

{% extends "base.html" %}{% block content %}<table><tr valign="top">#可以更改用户头像的大小<td><img src="{{ user.avatar(128) }}"></td><td><h1>用户: {{ user.username }}</h1></td></tr></table><hr>{% for post in posts %}<table><tr valign="top"><td><img src="{{ post.author.avatar(36) }}"></td><td>{{ post.author.username }} 说:<br>{{ post.body }}</td></tr></table>{% endfor %}
{% endblock %}

现在用户个人资料页面设计好了，这样用户写的帖子以及他们的头像都可以显示出来。现在我想让首页也有类似的布局，但是复制/粘贴处理这种方法显然不太合适，因此新建模板用来被调用。

app/templates/_post.html : 用来放posts里面的内容

 <table><tr valign="top"><td><img src="{{ post.author.avatar(36) }}"></td><td>{{ post.author.username }} 说:<br>{{ post.body }}</td></tr></table>

app/templates/user.html : 在用户模板中被调用。

{% extends "base.html" %}{% block content %}<table><tr valign="top"><td><img src="{{ user.avatar(128) }}"></td><td><h1>User: {{ user.username }}</h1></td></tr></table><hr>{% for post in posts %}{% include '_post.html' %}{% endfor %}
{% endblock %}

把格式完善的更丰满些。

app/models.py : 增加新的字段

class User(UserMixin, db.Model):# ...about_me = db.Column(db.String(140))last_seen = db.Column(db.DateTime, default=datetime.utcnow)

修改了模型，还记得大明湖畔的migrate吗？

(venv) duke@coding:~/flask_tutorial/flask$ flask db migrate -m 'User增加新的字段'
INFO  [alembic.runtime.migration] Context impl MySQLImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate.compare] Detected added column 'users.about_me'
INFO  [alembic.autogenerate.compare] Detected added column 'users.last_seen'Generating /home/duke/prequestion/migrations/versions/cb586afdff6a_user增加新的字段.py ... done

继续下一步，写入数据库中！

(venv) duke@coding:~/flask_tutorial/flask$ flask db upgrade
INFO  [alembic.runtime.migration] Context impl MySQLImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.runtime.migration] Running upgrade 063e1979f08c -> cb586afdff6a, User增加新的字段

好的，现在user模型又新增了字段，那么模板也要相应的改变啦。

app/templates/user.html : 添加字段

{% extends "base.html" %}{% block content %}<table><tr valign="top"><td><img src="{{ user.avatar(128) }}"></td><td><h1>User: {{ user.username }}</h1>{% if user.about_me %}<p>{{ user.about_me }}</p>{% endif %}{% if user.last_seen %}<p>最近登录: {{ user.last_seen }}</p>{% endif %}</td></tr></table>...
{% endblock %}

模型、模板都改了，那中间最重要的视图部分当然也要修改啦。

app/routes.py : 添加对应的方法

#...
from datetime import datetime@app.before_request
def before_request():if current_user.is_authenticated:current_user.last_seen = datetime.utcnow()db.session.commit()#...

刚才在User模型添加了2个字段，上面完成了一个最近登录，还有一个关于我about_me，首先要在forms.py进行设计。

from wtforms import StringField, TextAreaField, SubmitField
from wtforms.validators import DataRequired, Length# ...class EditProfileForm(FlaskForm):username = StringField('用户名', validators=[DataRequired(message='请输入用户名!')])about_me = TextAreaField('关于我', validators=[Length(min=0, max=140)])submit = SubmitField('提交')

既然forms里进行修改，那么就要新增加一个专门的编辑个人资料的页面更合适些。

app/templates/edit_profile.html : 个人资料编辑

{% extends "base.html" %}{% block content %}<h1>个人资料编辑</h1><form action="" method="post">{{ form.hidden_tag() }}<p>{{ form.username.label }}<br>{{ form.username(size=32) }}<br>{% for error in form.username.errors %}<span style="color: red;">[{{ error }}]</span>{% endfor %}</p><p>{{ form.about_me.label }}<br>{{ form.about_me(cols=50, rows=4) }}<br>{% for error in form.about_me.errors %}<span style="color: red;">[{{ error }}]</span>{% endfor %}</p><p>{{ form.submit() }}</p></form>
{% endblock %}

这些都完成差不多了，接下来就是增加视图函数了。

app/routes.py : 编辑个人资料的视图函数

from app.forms import EditProfileForm
# ....@app.route('/edit_profile', methods=['GET', 'POST'])
@login_required
def edit_profile():form = EditProfileForm()if form.validate_on_submit():current_user.username = form.username.datacurrent_user.about_me = form.about_me.datadb.session.commit()flash('你的提交已变更.')return redirect(url_for('edit_profile'))elif request.method == 'GET':form.username.data = current_user.usernameform.about_me.data = current_user.about_mereturn render_template('edit_profile.html', title='个人资料编辑',form=form)

在用户界面添加一个可以点击的超链接，就完成个人资料编辑的页面啦。

app/templates/user.html : 编辑页面的链接

<td><h1>用户名: {{ user.username }}</h1>{% if user.about_me %}<p>{{ user.about_me }}</p>{% endif %}{% if user.last_seen %}<p>最近登录: {{ user.last_seen }}</p>{% endif %}{% if user == current_user %}<p><a href="{{ url_for('edit_profile') }}">个人资料编辑</a></p>{% endif %}
</td>

当当当～，最后对整个的显示如下。