SQLite数据库操作

详细见:https://blog.csdn.net/yingshukun/article/details/94005900

操作实例:

#python sqlite

#Author : Hongten

#MailTo : hongtenzone@foxmail.com

#QQ? ? : 648719819

#Blog? : http://www.cnblogs.com/hongten

#Create : 2013-08-09

#Version: 1.0

#DB-API 2.0 interface for SQLite databases

import sqlite3

import os

'''SQLite数据库是一款非常小巧的嵌入式开源数据库软件,也就是说

没有独立的维护进程,所有的维护都来自于程序本身。

在python中,使用sqlite3创建数据库的连接,当我们指定的数据库文件不存在的时候

连接对象会自动创建数据库文件;如果数据库文件已经存在,则连接对象不会再创建

数据库文件,而是直接打开该数据库文件。

? ? 连接对象可以是硬盘上面的数据库文件,也可以是建立在内存中的,在内存中的数据库

? ? 执行完任何操作后,都不需要提交事务的(commit)

? ? 创建在硬盘上面: conn = sqlite3.connect('c:\\test\\test.db')

? ? 创建在内存上面: conn = sqlite3.connect('"memory:')

? ? 下面我们一硬盘上面创建数据库文件为例来具体说明:

? ? conn = sqlite3.connect('c:\\test\\hongten.db')

? ? 其中conn对象是数据库链接对象,而对于数据库链接对象来说,具有以下操作:

? ? ? ? commit()? ? ? ? ? ? --事务提交

? ? ? ? rollback()? ? ? ? ? --事务回滚

? ? ? ? close()? ? ? ? ? ? --关闭一个数据库链接

? ? ? ? cursor()? ? ? ? ? ? --创建一个游标

? ? cu = conn.cursor()

? ? 这样我们就创建了一个游标对象:cu

? ? 在sqlite3中,所有sql语句的执行都要在游标对象的参与下完成

? ? 对于游标对象cu,具有以下具体操作:

? ? ? ? execute()? ? ? ? ? --执行一条sql语句

? ? ? ? executemany()? ? ? --执行多条sql语句

? ? ? ? close()? ? ? ? ? ? --游标关闭

? ? ? ? fetchone()? ? ? ? ? --从结果中取出一条记录

? ? ? ? fetchmany()? ? ? ? --从结果中取出多条记录

? ? ? ? fetchall()? ? ? ? ? --从结果中取出所有记录

? ? ? ? scroll()? ? ? ? ? ? --游标滚动

'''

#global var

#数据库文件绝句路径

DB_FILE_PATH = ''

#表名称

TABLE_NAME = ''

#是否打印sql

SHOW_SQL = True

def get_conn(path):

? ? '''获取到数据库的连接对象,参数为数据库文件的绝对路径

? ? 如果传递的参数是存在,并且是文件,那么就返回硬盘上面改

? ? 路径下的数据库文件的连接对象;否则,返回内存中的数据接

? ? 连接对象'''

? ? conn = sqlite3.connect(path)

? ? if os.path.exists(path) and os.path.isfile(path):

? ? ? ? print('硬盘上面:[{}]'.format(path))

? ? ? ? return conn

? ? else:

? ? ? ? conn = None

? ? ? ? print('内存上面:[:memory:]')

? ? ? ? return sqlite3.connect(':memory:')

def get_cursor(conn):

? ? '''该方法是获取数据库的游标对象,参数为数据库的连接对象

? ? 如果数据库的连接对象不为None,则返回数据库连接对象所创

? ? 建的游标对象;否则返回一个游标对象,该对象是内存中数据

? ? 库连接对象所创建的游标对象'''

? ? if conn is not None:

? ? ? ? return conn.cursor()

? ? else:

? ? ? ? return get_conn('').cursor()

###############################################################

####? ? ? ? ? ? 创建|删除表操作? ? START

###############################################################

def drop_table(conn, table):

? ? '''如果表存在,则删除表,如果表中存在数据的时候,使用该

? ? 方法的时候要慎用!'''

? ? if table is not None and table != '':

? ? ? ? sql = 'DROP TABLE IF EXISTS ' + table

? ? ? ? if SHOW_SQL:

? ? ? ? ? ? print('执行sql:[{}]'.format(sql))

? ? ? ? cu = get_cursor(conn)

? ? ? ? cu.execute(sql)

? ? ? ? conn.commit()

? ? ? ? print('删除数据库表[{}]成功!'.format(table))

? ? ? ? close_all(conn, cu)

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

def create_table(conn, sql):

? ? '''创建数据库表:student'''

? ? if sql is not None and sql != '':

? ? ? ? cu = get_cursor(conn)

? ? ? ? if SHOW_SQL:

? ? ? ? ? ? print('执行sql:[{}]'.format(sql))

? ? ? ? cu.execute(sql)

? ? ? ? conn.commit()

? ? ? ? print('创建数据库表[student]成功!')

? ? ? ? close_all(conn, cu)

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

###############################################################

####? ? ? ? ? ? 创建|删除表操作? ? END

###############################################################

def close_all(conn, cu):

? ? '''关闭数据库游标对象和数据库连接对象'''

? ? try:

? ? ? ? if cu is not None:

? ? ? ? ? ? cu.close()

? ? finally:

? ? ? ? if cu is not None:

? ? ? ? ? ? cu.close()

###############################################################

####? ? ? ? ? ? 数据库操作CRUD? ? START

###############################################################

def save(conn, sql, data):

? ? '''插入数据'''

? ? if sql is not None and sql != '':

? ? ? ? if data is not None:

? ? ? ? ? ? cu = get_cursor(conn)

? ? ? ? ? ? for d in data:

? ? ? ? ? ? ? ? if SHOW_SQL:

? ? ? ? ? ? ? ? ? ? print('执行sql:[{}],参数:[{}]'.format(sql, d))

? ? ? ? ? ? ? ? cu.execute(sql, d)

? ? ? ? ? ? ? ? conn.commit()

? ? ? ? ? ? close_all(conn, cu)

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

def fetchall(conn, sql):

? ? '''查询所有数据'''

? ? if sql is not None and sql != '':

? ? ? ? cu = get_cursor(conn)

? ? ? ? if SHOW_SQL:

? ? ? ? ? ? print('执行sql:[{}]'.format(sql))

? ? ? ? cu.execute(sql)

? ? ? ? r = cu.fetchall()

? ? ? ? if len(r) > 0:

? ? ? ? ? ? for e in range(len(r)):

? ? ? ? ? ? ? ? print(r[e])

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

def fetchone(conn, sql, data):

? ? '''查询一条数据'''

? ? if sql is not None and sql != '':

? ? ? ? if data is not None:

? ? ? ? ? ? #Do this instead

? ? ? ? ? ? d = (data,)

? ? ? ? ? ? cu = get_cursor(conn)

? ? ? ? ? ? if SHOW_SQL:

? ? ? ? ? ? ? ? print('执行sql:[{}],参数:[{}]'.format(sql, data))

? ? ? ? ? ? cu.execute(sql, d)

? ? ? ? ? ? r = cu.fetchall()

? ? ? ? ? ? if len(r) > 0:

? ? ? ? ? ? ? ? for e in range(len(r)):

? ? ? ? ? ? ? ? ? ? print(r[e])

? ? ? ? else:

? ? ? ? ? ? print('the [{}] equal None!'.format(data))

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

def update(conn, sql, data):

? ? '''更新数据'''

? ? if sql is not None and sql != '':

? ? ? ? if data is not None:

? ? ? ? ? ? cu = get_cursor(conn)

? ? ? ? ? ? for d in data:

? ? ? ? ? ? ? ? if SHOW_SQL:

? ? ? ? ? ? ? ? ? ? print('执行sql:[{}],参数:[{}]'.format(sql, d))

? ? ? ? ? ? ? ? cu.execute(sql, d)

? ? ? ? ? ? ? ? conn.commit()

? ? ? ? ? ? close_all(conn, cu)

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

def delete(conn, sql, data):

? ? '''删除数据'''

? ? if sql is not None and sql != '':

? ? ? ? if data is not None:

? ? ? ? ? ? cu = get_cursor(conn)

? ? ? ? ? ? for d in data:

? ? ? ? ? ? ? ? if SHOW_SQL:

? ? ? ? ? ? ? ? ? ? print('执行sql:[{}],参数:[{}]'.format(sql, d))

? ? ? ? ? ? ? ? cu.execute(sql, d)

? ? ? ? ? ? ? ? conn.commit()

? ? ? ? ? ? close_all(conn, cu)

? ? else:

? ? ? ? print('the [{}] is empty or equal None!'.format(sql))

###############################################################

####? ? ? ? ? ? 数据库操作CRUD? ? END

###############################################################

###############################################################

####? ? ? ? ? ? 测试操作? ? START

###############################################################

def drop_table_test():

? ? '''删除数据库表测试'''

? ? print('删除数据库表测试...')

? ? conn = get_conn(DB_FILE_PATH)

? ? drop_table(conn, TABLE_NAME)

def create_table_test():

? ? '''创建数据库表测试'''

? ? print('创建数据库表测试...')

? ? create_table_sql = '''CREATE TABLE `student` (

? ? ? ? ? ? ? ? ? ? ? ? ? `id` int(11) NOT NULL,

? ? ? ? ? ? ? ? ? ? ? ? ? `name` varchar(20) NOT NULL,

? ? ? ? ? ? ? ? ? ? ? ? ? `gender` varchar(4) DEFAULT NULL,

? ? ? ? ? ? ? ? ? ? ? ? ? `age` int(11) DEFAULT NULL,

? ? ? ? ? ? ? ? ? ? ? ? ? `address` varchar(200) DEFAULT NULL,

? ? ? ? ? ? ? ? ? ? ? ? ? `phone` varchar(20) DEFAULT NULL,

? ? ? ? ? ? ? ? ? ? ? ? ? PRIMARY KEY (`id`)

? ? ? ? ? ? ? ? ? ? ? ? )'''

? ? conn = get_conn(DB_FILE_PATH)

? ? create_table(conn, create_table_sql)

def save_test():

? ? '''保存数据测试...'''

? ? print('保存数据测试...')

? ? save_sql = '''INSERT INTO student values (?, ?, ?, ?, ?, ?)'''

? ? data = [(1, 'Hongten', '男', 20, '广东省广州市', '13423****62'),

? ? ? ? ? ? (2, 'Tom', '男', 22, '美国旧金山', '15423****63'),

? ? ? ? ? ? (3, 'Jake', '女', 18, '广东省广州市', '18823****87'),

? ? ? ? ? ? (4, 'Cate', '女', 21, '广东省广州市', '14323****32')]

? ? conn = get_conn(DB_FILE_PATH)

? ? save(conn, save_sql, data)

def fetchall_test():

? ? '''查询所有数据...'''

? ? print('查询所有数据...')

? ? fetchall_sql = '''SELECT * FROM student'''

? ? conn = get_conn(DB_FILE_PATH)

? ? fetchall(conn, fetchall_sql)

def fetchone_test():

? ? '''查询一条数据...'''

? ? print('查询一条数据...')

? ? fetchone_sql = 'SELECT * FROM student WHERE ID = ? '

? ? data = 1

? ? conn = get_conn(DB_FILE_PATH)

? ? fetchone(conn, fetchone_sql, data)

def update_test():

? ? '''更新数据...'''

? ? print('更新数据...')

? ? update_sql = 'UPDATE student SET name = ? WHERE ID = ? '

? ? data = [('HongtenAA', 1),

? ? ? ? ? ? ('HongtenBB', 2),

? ? ? ? ? ? ('HongtenCC', 3),

? ? ? ? ? ? ('HongtenDD', 4)]

? ? conn = get_conn(DB_FILE_PATH)

? ? update(conn, update_sql, data)

def delete_test():

? ? '''删除数据...'''

? ? print('删除数据...')

? ? delete_sql = 'DELETE FROM student WHERE NAME = ? AND ID = ? '

? ? data = [('HongtenAA', 1),

? ? ? ? ? ? ('HongtenCC', 3)]

? ? conn = get_conn(DB_FILE_PATH)

? ? delete(conn, delete_sql, data)

###############################################################

####? ? ? ? ? ? 测试操作? ? END

###############################################################

def init():

? ? '''初始化方法'''

? ? #数据库文件绝句路径

? ? global DB_FILE_PATH

? ? DB_FILE_PATH = 'c:\\test\\hongten.db'

? ? #数据库表名称

? ? global TABLE_NAME

? ? TABLE_NAME = 'student'

? ? #是否打印sql

? ? global SHOW_SQL

? ? SHOW_SQL = True

? ? print('show_sql : {}'.format(SHOW_SQL))

? ? #如果存在数据库表,则删除表

? ? drop_table_test()

? ? #创建数据库表student

? ? create_table_test()

? ? #向数据库表中插入数据

? ? save_test()


def main():

? ? init()

? ? fetchall_test()

? ? print('#' * 50)

? ? fetchone_test()

? ? print('#' * 50)

? ? update_test()

? ? fetchall_test()

? ? print('#' * 50)

? ? delete_test()

? ? fetchall_test()

if __name__ == '__main__':

? ? main()

?著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,100评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,308评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事?!?“怎么了?”我有些...
    开封第一讲书人阅读 159,718评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,275评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,376评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,454评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,464评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,248评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,686评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,974评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,150评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,817评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,484评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,140评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,374评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,012评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,041评论 2 351

推荐阅读更多精彩内容