Falsk框架是轻量级的,不集成数据库等操作,但扩展很灵活,比如flask-sqlalchemy,这个扩展封装了 SQLAlchemy 项目,这是一个对象关系映射器(ORM),再搭配flask-migrate绑定app和数据库,跟踪数据库的更新迁移。
先介绍SQLite的使用然后介绍Mysql,后期使用flask-sqlalchemy扩展操作数据库更方便。
使用SQLite
要使用SQLite,先导入import sqlite3,然后连接数据库
flaskr/db.py
import sqlite3
import click
from flask import current_app, g
from flask.cli import with_appcontext
# 获取数据库连接
def get_db():
if 'db' not in g:
g.db = sqlite3.connect(
current_app.config['DATABASE'],
detect_types=sqlite3.PARSE_DECLTYPES
)
g.db.row_factory = sqlite3.Row
return g.db
# 关闭数据库连接
def close_db(e=None):
db = g.pop('db', None)
if db is not None:
db.close()
g跟current_app都是特殊对象,g的作用域是同一个请求,current_app是代表整个应用实例。
sqlite3.connect()连接数据库
sqlite3.Row返回行数据,类似字典操作,可以按name查找列。
每个请求用完数据库记得关闭。
创建表
flaskr/schema.sql 文件是创建数据库表的命令
DROP TABLE IF EXISTS user;
DROP TABLE IF EXISTS post;
CREATE TABLE user (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT UNIQUE NOT NULL,
password TEXT NOT NULL
);
CREATE TABLE post (
id INTEGER PRIMARY KEY AUTOINCREMENT,
author_id INTEGER NOT NULL,
created TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
title TEXT NOT NULL,
body TEXT NOT NULL,
FOREIGN KEY (author_id) REFERENCES user (id)
);
flaskr/db.py 文件是调用schema.sql 文件初始化数据库的python操作
def init_db():
db = get_db()
with current_app.open_resource('schema.sql') as f:
db.executescript(f.read().decode('utf8'))
@click.command('init-db')
@with_appcontext
def init_db_command():
"""Clear the existing data and create new tables."""
init_db()
click.echo('Initialized the database.')
def init_app(app):
app.teardown_appcontext(close_db)
app.cli.add_command(init_db_command)
open_resource() 读取外部的sql命令行文件,然后调用数据库的executescript()方法执行读取的命令,完成数据库操作。
click.command() 是自定义命令,可以在命令行使用flask init-db直接执行这里的init_db_command函数。
详细可以参照官方文档 Command Line Interface的说明:https://flask.palletsprojects.com/en/1.1.x/cli/#cli
app.teardown_appcontext() 告诉Flask在返回响应后清理时调用该函数,即关闭数据库连接。
app.cli.add_command() 添加一条新命令可以使用flask执行。
在应用初始化的时候调用init_app函数
flaskr/__init__.py
def create_app():
app = ...
# existing code omitted
from . import db
db.init_app(app)
return app
然後就可以使用flask命令执行数据库初始化
$ flask init-db
Initialized the database.