【程序员技术练级】学习一门脚本语言 python（三）跟数据库打交道

接着上一篇，该篇讲述使用python对数据库进行基本的CRUD操作，这边以sqlite3为例子，进行说明。sqlite3 是一个非常轻型的数据库，安装和使用它是非常简单的，这边就不进行讲述了。

在python下，已经自带了sqlite3模块，供我们使用，这个模块的用途其实和JAVA当中的jdbc驱动是类似的，通过它，我们就可以非常方便的和sqlite3进行各种操作了。

好了，开始吧。

python的数据库模块有统一的接口标准，所有的操作都有统一的模式，大概是分为以下几步（假设使用的是sqlite3模块）：

1. 用sqlite3.connect()获取数据库的连接，假设连接为conn
2. 如果我们的操作不需要返回结果，那么就可以直接执行conn.execute(sql)。
3. 由于数据库事务隔离级别的不同，如果我们执行的是对数据库的修改或者插入，那么我们需要执行conn.commit()
4. 如果需要返回结果，那么conn.cursor()创建游标对象cursor，通过cursor.execute查询数据库，用cursor.fetchall/cur.fetchone/cur.fetchmany返回查询结果。
5. 由于数据库事务隔离级别的不同，如果我们执行的是对数据库的修改或者插入，那么我们需要执行cur.commit()
6. 最后，最重要的是 对conn和cursor进行关闭，cursor.close()、conn.close() (先关闭cursor，在关闭conn)

具体的相关CRUD实现代码如下：

import os
import sqlite3

#根据传入的path获取一个数据库连接，如果path非法，那么创建一个临时的内存数据库
def get_conn(path):

    conn = sqlite3.connect(path)

    if os.path.exists(path) and os.path.isfile(path):
        print('database path is %s' % format(path))
        return conn
    else:
        conn = None
        return sqlite3.connect(':memory')

#获取游标
def get_cursor(conn):
    if conn is not None:
        return conn.cursor()
    else:
        return get_conn('').cursor()

#创建表数据
def create_table(conn, sql):
    if sql is not None and sql != '':
        cursor = get_cursor(conn)
        cursor.execute(sql)
        conn.commit()
        close_all(conn, cursor)

#关闭数据库连接
def close_all(conn, cursor):
    try:
        if cursor is not None:
            cursor.close()
    finally:
        if conn is not None:
            conn.close()

#CURD操作，需要返回值，查询是肯定有返回值
def crud(conn, sql, data):
    cursor = get_cursor(conn)
    n = cursor.execute(sql, data)
    print("n", n.rowcount)
    fetchall = cursor.fetchall()
    conn.commit()
    close_all(conn, cursor)
    return fetchall

#CRUD操作，不需要返回值，查询是肯定有返回值
def crud(conn, sql, data):
    cursor = get_cursor(conn)
    n = cursor.execute(sql, data)
    conn.commit()
    close_all(conn, cursor)

一个简单的例子：

conn = get_conn('D:/temp/sqlite3/test.db')
fetchall = crud(conn, 'select * from student', '')

#假设返回的数据有三个字段
for row in fetchall:
    print('name: %s, age: %d, address: %s' % (row[0], row[1], row[2]))

好了，上面的代码就是最简单的对python的数据库操作了。

大家如果有什么问题或者文件有什么错误的话，可以留言大家一起探讨！