[12] [烧瓶里的部落格] 02. 定义和使用资料库 - 使用 SQLite

什麽是 SQLite

SQLite 是遵守ACID的关联式资料库管理系统,基於单一文件所组成且格式定义明确
因为不需要单独设定一个资料库服务,所以很适合开发阶段
但因为写入是按顺序执行的,当并发请求同时要写入时会比较慢一点
小型应用程序没有问题,大型应用程序就不太适合

Python 内建了对 SQLite 的支援,对应的模组是 sqlite3

连接资料库

当要使用 SQLite 资料库,或是其他多数的资料库时,第一件事情就是建立一个资料库连线
所有查询和操作都要透过该连线来执行,并且执行完之後要关闭连线

建立档案 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

g 是一个特殊物件,独立於每一个请求
在处理请求过程中用於储存可能多个函数都会用到的资料
把连线储存於其中,就可以重复使用
而不用在同一个请求中每次呼叫 get_db 时都建立一个新的连线,实现单例模式

sqlite3.connect()

建立一个资料库连线,连线会指向设定中的DATABASE指定的档案
这个档案现在还没有建立,後面会在初始化资料库的时候建立

current_app

这是另一个特殊物件,该物件指向正在处理请求的 Flask 应用
使用工厂模式之後,在专案的程序中就不会直接呼叫应用程序物件
建立应用程序後,处理请求时会呼叫get_db,这时候就可以使用 current_app

sqlite3.Row

告诉连线回传类似於字典(dict)的 row,这样可以通过 columns 名称来存取资料

close_db

通过检查 g.db 来确定连线是否已经建立,如果连接已建立就关闭连线
之後会在应用工厂中告诉应用程序close_db函数
这样每次请求结束後就会呼叫它来关闭连线

建立资料表

在 SQLite 中,资料储存在 tablescolumns 中,在储存和读取资料之前需要先建立它们
Flaskr 会把使用者资料储存在user表中,把部落格内容储存post表中
底下我们建立一个文件储存用於创建空 table 的 SQL 指令

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)
);

接着在db.py中添加 Python 函数,用来执行这个 SQL 指令

flaskr/db.py

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.')

open_resource()

打开一个档案,档名是相对於flaskr的,这样就不需要考虑以後应用具体部署在哪个路径
get_db 回传一个资料库连线,用於执行档案中的指令

click.command()

定义一个名为init-db指令,它呼叫init_db函数,并为使用者显示一个成功的讯息
更多关於指令的写法可以参考:Command Line Interface

在应用程序中注册

close_dbinit_db_command函数需要在应用的实体中注册,否则无法使用
因为我们使用了工厂函数,所以光是写好函数是无法直接使用的
我们需要写一个函数,把应用作为参数,在函数中进行注册

flaskr/db.py

def init_app(app):
    app.teardown_appcontext(close_db)
    app.cli.add_command(init_db_command)

app.teardown_appcontext()

告诉 Flask 在回传 response 後呼叫这个函数来断开连线

app.cli.add_command()

添加一个新的、可以被flask指令呼叫的指令方法

下一步是在工厂中导入并呼叫这个函数
在工厂函数中把新的程序放到create_app函数的最後,return app之前

flaskr/init.py

def create_app():
    app = ...
    # existing code omitted

    from . import db
    db.init_app(app)

    return app

初始化资料库档案

现在init-db已经在应用中注册好了,可以和flask指令一起使用
使用方式和之前的flask run类似

停止之前执行中的 flask 服务,或是另外开一个终端机视窗
如果是开新的终端机执行,那必须重新设定FLASK_APPFLASK_ENV

执行完之後会有一个flaskr.sqlite档案出现在专案资料夹的instance资料夹中!
一般来说会把 instance 资料夹加入.gitignore


<<:  10 - BAT - 使用高亮预览文件

>>:  Day12_HTML语法9

故事二十八:人口密度高的地区,每一户的所得总额是不是也比较高呢?

     最近,有事没事就会逛逛open data的网站,我今天很幸运的看到了两个资料集:分别是 综...

Day26 Let's ODOO: ODOO Studio

今天来介绍Odoo studio,对於程序苦手或非程序人员,运用Odoo studio可以直接透过见...

[Day 2] 差异性安装

如需在地端环境操作 那需要去理解 什麽是node JS 什麽是NPM 需要参照 本地安装 使用 np...

OpenStack 介绍 2

本系列文章同步发布於笔者网站 前一篇文章以比较非技术角度介绍了 OpenStack 这个专案。今天开...

[day14]Vue.js 网站基本建置

其实今年才刚学Vue.js ,目前的程度大概就是写几个简单的功能而已,要写一个比较完整的网站还是十分...