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

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

flask
├── app
│   ├── forms.py
│   ├── __init__.py
│   ├── routes.py
│   └── templates
│       ├── base.html
│       ├── index.html
│       └── login.html
├── config.py
├── myblog.py

4 数据库

对于一个完整的网站数据库是很重要的，因为你的数据得有位置读取呀，网上很多数据库都用的sqlite，但是，我想使用mysql，所以接下来咱们就以mysql为数据库来讲解喽。

(venv) duke@coding:~/flask_tutorial/flask$ pip install flask-sqlalchemy

这样就可以对数据库进行操作了，但是实际项目中会经常对数据库进行修改，但是一般不会手动的去数据库里进行改动，通常的做法是修改ORM对应的模型，然后再把模型映射到数据库中。在flask里有一个集成的工具是专门做这个事情的，安装它。

(venv) duke@coding:~/flask_tutorial/flask$ pip install flask-migrate

因为我使用的是MySQL数据库，而在python3中不再支持mysqldb，因此我们还需要安装pymysql.

(venv) duke@coding:~/flask_tutorial/flask$ pip install pymysql

需要的组件都安装好了，接下来在配置文件里配置数据库。

app/config.py : 使用配置文件config.py中的内容

import os
BASE_DIR = os.path.abspath(os.path.dirname(__file__))
class Config(object):
    #.......
    #格式为mysql+pymysql://数据库用户名:密码@数据库地址:端口号/数据库的名字?数据库格式
    SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://root:123456@localhost:3306/flaskblog?charset=utf8'
    #如果你不打算使用mysql，使用这个连接sqlite也可以
    #SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(BASE_DIR,'app.db')
    SQLALCHEMY_TRACK_MODIFICATIONS = False

配置文件设置好了以后，就要对初始化文件进行修改。

app/_ _ init_ _.py : 数据库的初始化设置

from flask import Flask
from config import Config
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
app = Flask(__name__)
app.config.from_object(Config)
#建立数据库关系
db = SQLAlchemy(app)
#绑定app和数据库，以便进行操作
migrate = Migrate(app,db)

from app import routes,models

接下来比较重要的就是设计模型了,model是模型的映射，只有设计好model才能进行一系列的操作。新建一个models.py文件。

(venv) duke@coding:~/flask_tutorial/flask$ touch app/models.py

app/models.py : 用户数据库模型

from app import db

class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.Integer,primary_key=True)
    username = db.Column(db.String(64),index=True,unique=True)
    email = db.Column(db.String(120),index=True,unique=True)
    password_hash = db.Column(db.String(128))

    def __repr__(self):
        return '<用户名:{}>'.format(self.username)

对模型进行验证：

(venv) duke@coding:~/flask_tutorial/flask$ python
Python 3.6.4 (default, May  3 2018, 19:35:55) 
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from app.models import User
>>> u = User(username='duke',email='[email protected]')
>>> u
<用户名:duke>

对模型验证后，发现是正确的,进行数据库初始化。

(venv) duke@coding:~/flask_tutorial/flask$ flask db init
  Creating directory /home/duke/flask_tutorial/flask/migrations ... done
  Creating directory /home/duke/flask_tutorial/flask/migrations/versions ... done
  Generating /home/duke/flask_tutorial/flask/migrations/script.py.mako ... done
  Generating /home/duke/flask_tutorial/flask/migrations/alembic.ini ... done
  Generating /home/duke/flask_tutorial/flask/migrations/README ... done
  Generating /home/duke/flask_tutorial/flask/migrations/env.py ... done
  Please edit configuration/connection/logging settings in
  '/home/duke/flask_tutorial/flask/migrations/alembic.ini' before proceeding.

现在看一看项目结构

flask
├── app
│   ├── forms.py
│   ├── __init__.py
│   ├── models.py
│   ├── routes.py
│   └── templates
│       ├── base.html
│       ├── index.html
│       └── login.html
├── config.py
├── migrations
│   ├── alembic.ini
│   ├── env.py
│   ├── README
│   ├── script.py.mako
│   └── versions
├── myblog.py

接下来创建数据库的管理工具。

(venv) duke@coding:~/flask_tutorial/flask$ flask db migrate -m 'users_table'
INFO  [alembic.runtime.migration] Context impl MySQLImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate.compare] Detected added table 'user'
INFO  [alembic.autogenerate.compare] Detected added index 'ix_user_email' on '['email']'
INFO  [alembic.autogenerate.compare] Detected added index 'ix_user_username' on '['username']'
  Generating
  /home/duke/flask_tutorial/flask/migrations/versions/06ea43ff4439_users_table.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  -> 06ea43ff4439, users_table

执行到这里，表已经在数据库中建立完成。或许你会问，这和我以前的接触过的不一样啊，这么会麻烦这么多，因为你对数据库模型修改后，可以很方便的进行数据库表的迁移，这会大大的减少以后可能会发生的大量的工作量。

网站里不可能只有一张表的，多张表之间肯定会有联系，接下来对表之间的关系进行一个详细介绍。

app/models.py : 添加一张新表，并建立关系

from datetime import datetime
from app import db

class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.Integer,primary_key=True)
    username = db.Column(db.String(64),index=True,unique=True)
    email = db.Column(db.String(120),index=True,unique=True)
    password_hash = db.Column(db.String(128))
    # back是反向引用,User和Post是一对多的关系，backref是表示在Post中新建一个属性author，关联的是Post中的user_id外键关联的User对象。
    #lazy属性常用的值的含义，select就是访问到属性的时候，就会全部加载该属性的数据;joined则是在对关联的两个表进行join操作，从而获取到所有相关的对象;dynamic则不一样，在访问属性的时候，并没有在内存中加载数据，而是返回一个query对象, 需要执行相应方法才可以获取对象，比如.all()
    posts = db.relationship('Post',backref='author',lazy='dynamic')

    def __repr__(self):
        return '<用户名:{}>'.format(self.username)

class Post(db.Model):
    __tablename__ = 'post'
    id = db.Column(db.Integer,primary_key=True)
    body = db.Column(db.String(140))
    timestamp = db.Column(db.DateTime,index=True,default=datetime.utcnow)
    user_id = db .Column(db.Integer,db.ForeignKey('user.id'))

    def __repr__(self):
        return '<Post {}>'.format(self.body)

生成新的数据库关系：

(venv) duke@coding:~/flask_tutorial/flask$ flask db migrate -m 'posts_table'
INFO  [alembic.runtime.migration] Context impl MySQLImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate.compare] Detected added table 'post'
INFO  [alembic.autogenerate.compare] Detected added index 'ix_post_timestamp' on '['timestamp']'
  Generating /home/miguel/microblog/migrations/versions/780739b227a7_posts_table.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  -> 780739b227a7, posts_table

接下来使用shell对进行操作:

(venv) duke@coding:~/flask_tutorial/flask$ python
Python 3.6.4 (default, May  3 2018, 19:35:55) 
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from app import db
>>> from app.models import User,Post

在用户表中添加一条用户信息：

#创建一个User对象实例
>>> u = User(username='duke',email='[email protected]')
#将实例添加
>>> db.session.add(u)
#提交
>>> db.session.commit()

查看数据库后发现，插进一条数据。

在用户表中添加另外一条用户信息:

>>> u = User(username='king',email='[email protected]')
>>> db.session.add(u)
>>> db.session.commit()

通过查询数据库发现也插进了一条数据，但是怎么查询的呢？

查询数据库中信息：

>>> users = User.query.all()
>>> users
[<用户名:duke>, <用户名:king>]
>>> for u in users:
...     print(u.id,u.username)
... 
1 duke
2 king

这是一次性全部查询，那能不能查单条数据呢？答案是肯定的喽。

#根据id可以查询到数据
>>> u = User.query.get(2)
>>> u
<用户名:king>

普通的数据库表咱们会了，那么有外键关联的一对多中数据库表怎么插入数据呢？

#查找到一个User对象
>>> u = User.query.get(1)
#将该User对象与Post建立关系
>>> p = Post(body='我第一次提交数据！',author=u)
>>> db.session.add(p)
>>> db.session.commit()
#插入第二条数据，但是拥有者都是同一个人
>>> p = Post(body='我第二次提交数据了！',author=u)
>>> db.session.add(p)
>>> db.session.commit()

现在通过查询来让你更加的熟悉一对多的关系。

# 获得一个用户的所有提交
>>> u = User.query.get(1)
>>> u
<用户名:duke>
#u.posts.all()中的posts是model中的，User和Post关联的作用。
>>> posts = u.posts.all()
>>> posts
[<Post 我第一次提交数据！>, <Post 我第二次提交数据了！>]

#进行相同的操作，但是换成另外一名用户
>>> u = User.query.get(2)
>>> u
<用户名:king>
>>> u.posts.all()
[]

#对所有的posts进行查询
>>> posts = Post.query.all()
>>> for p in posts:
...     print(p.id,p.author.username,p.body)
... 
1 duke 我第一次提交数据！
2 duke 我第二次提交数据了！

#还可以按照一定的规则进行查询
>>> User.query.order_by(User.username.desc()).all()
[<用户名:king>, <用户名:duke>]

熟悉了查询，接下来就把刚才的测试数据都删除吧！

>>> users = User.query.all()
>>> for u in users:
...     db.session.delete(u)
... 
>>> posts = Post.query.all()
>>> for p in posts:
...     db.session.delete(p)
... 
>>> db.session.commit()