在python中使用SQLite

时间:2022-06-01 23:38:59

在python中使用SQLite--连接数据库  

导入库:
from pysqlite2 import dbapi2 as sqlite

建立数据库文件或打开数据库:
connection = sqlite.connect('test.db')
如数据库文件test.db文件不存在,则它将会创建;如存在,则会打开

在内存中创建一个临时数据库:
memoryConnect = sqlite.connect(':memory:')

一旦创建一个connect对象后,得创建一个cursor对象。
cursor = connection.cursor()


在python中使用SQLite--数据库中存入数据  

用cursor.execute来创建表格:
>>> cursor.execute(''CREATE TABLE names (id INTEGER PRIMARY KEY,
name VARCHAR(50), email VARCHAR(50))')

插入数据:
>>> cursor.execute('INSERT INTO names VALUES (null, "John Doe",
"jdoe@jdoe.zz")')
>>> cursor.execute('INSERT INTO names VALUES (null, "Mary Sue",
"msue@msue.yy")')

获取最后一行的id:
>>> cursor.lastrowid
2

因为用户的输入是不安全的。可以使用下面的方法阻止sql注入攻击:
>>> name = "Luke Skywalker"
>>> email ="use@the.force"
>>> cursor.execute('INSERT INTO names VALUES (null, ?, ?)',
(name, email))

数据库中的提交事务,回滚事务:
>>> connection.commit()
>>> connection.rollback()


在python中使用SQLite--数据库中取数据  

从表中取数据用SELECT SQL语句
>>> cursor.execute('SELECT * FROM names')

取回数据的方法有种:
一、使用fetchall
>>> cursor.execute('SELECT * FROM names')
>>> print cursor.fetchall()
[(1, u'John Doe', u'jdoe@jdoe.zz'), (2, u'Mary Sue',
u'msue@msue.yy'), (3, u'Luke Skywalker', u'use@the.force')]

>> cursor.execute('SELECT * FROM names')
>>> for row in cursor:
                print '-'*10
                print 'ID:', row[0]
                print 'Name:', row[1]
                print 'E-Mail:', row[2]
                print '-'*10        
结果省略
但这个方法搜索出来的数据只能使用一次

二、使用fetchmany:
使用fetchmany可以取回指定条数的记录,比较适合用于分页情况
>>> cursor.execute('SELECT * FROM names')
>>> print cursor.fetchmany(2)
[(1, u'John Doe', u'jdoe@jdoe.zz'), (2, u'Mary Sue',
u'msue@msue.yy')]

可以继续取后续的数据
>>> print cursor.fetchmany()
[(3, u'Luke Skywalker', u'use@the.force')]

三、使用fetchone
一次取一条记录
>>> cursor.execute('SELECT * FROM names')
>>> cursor.fetchone()
(1, u'John Doe', u'jdoe@jdoe.zz')

四、使用next
类似fetchone,但到最后一条之后,再使用时抛出StopIteration错误
>>> cursor.execute('SELECT * FROM names')
>>> cursor.next()
(1, u'John Doe', u'jdoe@jdoe.zz')
>>> cursor.next()
(2, u'Mary Sue', u'msue@msue.yy')
>>> cursor.next()
(3, u'Luke Skywalker', u'use@the.force')
>>> cursor.next()
Traceback (most recent call last):
  File "<pyshell#201>", line 1, in -toplevel-
    cursor.next()
StopIteration