时间戳

首页 > 解决方案 > 如何将项目列表添加到 sqlite3 烧瓶中？

python - 如何将项目列表添加到 sqlite3 烧瓶中？

问题描述

我最近开始使用蓝图烧瓶。有一个 dicts => list[{'title':'ajsdhkd', 'author': 'askjdh qweqwqw'}, {'title':'ajsdhkd', 'author': 'askjdh qweqwqw'}] 的列表需要添加到sql。我想用创建的包含这个字典的表来初始化我的数据库。这样我以后可以从蓝图烧瓶路线执行它。但我不明白怎么做。

在 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 query_db(query, args=(), one=False):
    cur = get_db().execute(query, args)
    rv = cur.fetchall()
    cur.close()
    return (rv[0] if rv else None) if one else rv
    
def close_db(e=None): 
    db = g.pop('db', None)

    if db is not None:
        db.close()

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():
    init_db()
    click.echo('Initialized the database.')

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

有一个 schema.sql 文件，其中： DROP TABLE IF EXISTS books；

CREATE TABLE books (
    title varchar(50),
    author varchar(100),

);

......

标签： pythonsqliteflaskblueprint

解决方案

使用普通的 sqlite3，下面的函数应该创建一个名为“books”的 SQLite 表，然后将您描述的 dicts 列表添加到该表中。您可能想要current_app.config['DATABASE']替换'test.db'.

import sqlite3

def init_db():
    data = [{'title':'ajsdhkd', 'author': 'askjdh qweqwqw'}, {'title':'ajsdhkd', 'author': 'askjdh qweqwqw'}]

    conn = sqlite3.connect('test.db', detect_types=sqlite3.PARSE_DECLTYPES)
    cursor = conn.cursor()

    cursor.execute("CREATE TABLE 'books' (title TEXT, author TEXT);")    

    sql = "INSERT INTO 'books' (title, author) VALUES (?, ?);"
    for book in data:
        cursor.execute(sql, (book['title'], book['author']))

    conn.commit()
    conn.close()

您也可以考虑使用我的“easy_db”库来简化构建“books”表并向其添加数据的替代方案。此函数与前一个函数执行相同的操作。

import easy_db

def init_db2():    
    data = [{'title':'ajsdhkd', 'author': 'askjdh qweqwqw'}, {'title':'ajsdhkd', 'author': 'askjdh qweqwqw'}]
    
    db = easy_db.DataBase('test.db')    
    db.append('books', data)

单行db.append('books', data)既创建了“书”表data，又将字典列表添加到该表中。“title”和“author”列是根据输入数据自动创建的 TEXT 类型。

推荐阅读