python操作sqlite——pysqlite3

python 操作sqlite3

python3 中已内置 sqlite3 可以直接使用

具体使用

1.import

import sqlite3

2.获取数据库连接

conn = sqlite3.connect('test.db')

相关方法
conn.cursor() 创建一个游标对象
conn.commit() 提交事务
conn.rollback() 回滚事务
conn.close() 关闭连接

3.获取游标对象

cur = conn.cursor()

常用方法
cur.execute(‘sql’) 执行sql
cur.fetchone() 获取第一个结果
cur.fetchmany(n) 获取n个结果
cur.fetchall() 获取全部结果

4.使用相关方法。执行sql。获取结果。

sql = 'select * from table_name'
cur.execute(sql)
res = cur.fetchall()

5.关闭连接

cur.close()
conn.close()

实际应用

import sqlite3


# 查询结果处理
def dict_factory(cursor, row):
    d = {}
    for index, col in enumerate(cursor.description):
        d[col[0]] = row[index]
    return d


conn = sqlite3.connect('select * from `table_name`')
# 转换输出结果
conn.row_factory = dict_factory


try:
    # 执行sql语句
    conn.cursor().execute(sql)
    # 提交到数据库执行
    conn.commit()
except Exception as e:
    print(e)
    # 如果发生错误则回滚
    conn.rollback()
else:
    return conn.cursor().fetchall()

返回值处理

一般情况下fetchall()返回的是一个元组列表，元组内每个元素也是元组，pymysql 可以通过设置 cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)获取json数据，sqlite3没有对应的处理，但是官方给出了解决方案：

import sqlite3

def dict_factory(cursor, row):
    d = {}
    for index, col in enumerate(cursor.description):
        d[col[0]] = row[index]
    return d

conn = sqlite3.connect('fp')
conn.row_factory = dict_factory

添加这一步处理之后，从查询中获取的数据就是json格式了，可以方便的用key直接获取value