python 操作sqlite資料庫

標籤：

‘‘‘SQLite資料庫是一款非常小巧的嵌入式開來源資料庫軟體，也就是說
沒有獨立的維護進程，所有的維護都來自於程式本身。
在python中，使用sqlite3建立資料庫的串連，當我們指定的資料庫檔案不存在的時候
連線物件會自動建立資料庫檔案；如果資料庫檔案已經存在，則連線物件不會再建立
資料庫檔案，而是直接開啟該資料庫檔案。
連線物件可以是硬碟上面的資料庫檔案，也可以是建立在記憶體中的，在記憶體中的資料庫
執行完任何操作後，都不需要提交事務的(commit)

建立在硬碟上面： conn = sqlite3.connect(‘e:\\test.db‘)
建立在記憶體上面： conn = sqlite3.connect(‘memory:‘)

下面我們一硬碟上面建立資料庫檔案為例來具體說明：
conn = sqlite3.connect(‘e:\\test.db‘)
其中conn對象是資料庫連結化物件，而對於資料庫連結化物件來說，具有以下操作：

commit() --事務提交
rollback() --交易回復
close() --關閉一個資料庫連結
cursor() --建立一個遊標

cu = conn.cursor()
這樣我們就建立了一個遊標對象：cu
在sqlite3中，所有sql語句的執行都要在遊標對象的參與下完成
對於遊標對象cu，具有以下具體操作：

execute() --執行一條sql語句
executemany() --執行多條sql語句
close() --遊標關閉
fetchone() --從結果中取出一條記錄
fetchmany() --從結果中取出多條記錄
fetchall() --從結果中取出所有記錄
scroll() --遊標滾動

‘‘‘

 

# -*- coding: utf-8 -*-
import sqlite3
import os

class SqliteDB:

　　#是否列印sql
　　print_sql = True

　　#資料庫連接
　　sqlite_Conn = None
　　def __init__(self,dbFile_path):
　　　　‘‘‘初始化資料庫檔案路徑‘‘‘
　　　　filepath = unicode(dbFile_path,‘utf8‘)
　　　　self.sqlite_Conn = self.get_Conn(filepath)

　　def get_Conn(self,dbFile_path):
　　　　‘‘‘擷取到資料庫的連線物件，參數為資料庫檔案的絕對路徑
　　　　如果傳遞的參數是存在，並且是檔案，那麼就返回硬碟上面改
　　　　路徑下的資料庫檔案的連線物件；否則，返回記憶體中的資料接
　　　　連線物件‘‘‘
　　　　if os.path.exists(dbFile_path) and os.path.isfile(dbFile_path):
　　　　　　print(‘硬碟上面:[{}]‘.format(dbFile_path))
　　　　　　return sqlite3.connect(dbFile_path)
　　　　else:
　　　　　　print(‘記憶體上面:[:memory:]‘)
　　　　　　return sqlite3.connect(‘:memory:‘)

　　def commit(self):
　　　　‘‘‘提交資料庫事務‘‘‘
　　　　if self.sqlite_Conn is not None:
　　　　　　self.sqlite_Conn.commit()

　　def get_Cursor(self):
　　　　‘‘‘
　　　　　該方法是擷取資料庫的遊標對象，參數為資料庫的連線物件
　　　　　如果資料庫的連線物件不為None，則返回資料庫連接對象所創
　　　　建的遊標對象；否則返回一個遊標對象，該對象是記憶體中資料
　　　　庫連線物件所建立的遊標對象
　　　　‘‘‘
　　　　if self.sqlite_Conn is not None:
　　　　　　return self.sqlite_Conn.cursor()
　　　　else:
　　　　　　return self.sqlite_Conn.cursor()

　　def close_Cursor(self,cursor):
　　　　‘‘‘關閉資料庫遊標對象和資料庫連接對象‘‘‘
　　　　try:
　　　　　　if cursor is not None:
　　　　　　cursor.close()
　　　　finally:
　　　　　　if cursor is not None:
　　　　　　　　cursor.close()

################################################################
#建立表，刪除表操作
################################################################
　　def create_Table(self, strSql):
　　　　‘‘‘建立資料庫表：‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　cursor = self.get_Cursor()
　　　　　　if self.print_sql:
　　　　　　　　print(‘執行sql:[{}]‘.format(strSql))
　　　　　　cursor.execute(strSql)
　　　　　　self.commit()
　　　　　　print(‘建立資料庫表成功!‘)
　　　　　　self.close_Cursor(cursor)
　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))


　　def drop_Table(self,table):
　　　　‘‘‘如果表存在,則刪除表，如果表中存在資料的時候，使用該
　　　　方法的時候要慎用！‘‘‘
　　　　if table is not None and table != ‘‘:
　　　　　　strSql = ‘DROP TABLE IF EXISTS ‘ + table
　　　　　　if self.print_sql:
　　　　　　　　print(‘執行sql:[{}]‘.format(strSql))
　　　　　　cursor = self.get_Cursor()
　　　　　　cursor.execute(strSql)
　　　　　　self.commit()
　　　　　　print(‘刪除資料庫表[{}]成功!‘.format(table))
　　　　　　self.close_Cursor(cursor)
　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))

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

#資料庫操作
#####################################################################
　　def insert_MultiData(self,strSql, data):
　　　　‘‘‘插入資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　　if data is not None:
　　　　　　　　cursor = self.get_Cursor()
　　　　　　　　for d in data:
　　　　　　　　　　if self.print_sql:
　　　　　　　　　　　　print(‘執行sql:[{}],參數:[{}]‘.format(strSql, d))
　　　　　　　　　　cursor.execute(strSql, d)
　　　　　　　　　　self.commit()
　　　　　　　　self.close_Cursor(cursor)
　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))

　　def insert_Data(self,strSql):
　　　　‘‘‘插入資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　cursor = self.get_Cursor()
　　　　　　print(‘執行sql:[{}]‘.format(strSql))
　　　　　　cursor.execute(strSql)
　　　　　　self.commit()
　　　　　　self.close_Cursor(cursor)

　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))

　　def get_All_Item(self,strSql):
　　　　‘‘‘查詢所有資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　cursor = self.get_Cursor()
　　　　　　if self.print_sql:
　　　　　　　　print(‘執行sql:[{}]‘.format(strSql))
　　　　　　cursor.execute(strSql)
　　　　　　listR = cursor.fetchall()

　　　　　　self.close_Cursor(cursor)
　　　　　　return listR

　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))
　　　　　　return None

　　def get_One_Item(self,strSql, data):
　　　　‘‘‘查詢一條資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　if data is not None:
　　　　　　#Do this instead
　　　　　　d = (data,)
　　　　　　cursor = self.get_Cursor()
　　　　　　if self.print_sql:
　　　　　　　　print(‘執行sql:[{}],參數:[{}]‘.format(strSql, data))
　　　　　　cursor.execute(strSql, d)
　　　　　　r = cursor.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(strSql))

　　def update_Data(self,strSql, data):
　　　　‘‘‘更新資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　if data is not None:
　　　　　　　　cursor = self.get_Cursor()
　　　　　　　　for d in data:
　　　　　　　　　　if self.print_sql:
　　　　　　　　　　　　print(‘執行sql:[{}],參數:[{}]‘.format(strSql, d))
　　　　　　　　　　cursor.execute(strSql, d)
　　　　　　　　　　self.commit()
　　　　　　　　self.close_Cursor(cursor)
　　　　　　else:
　　　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))

　　def delete_multiData(self,strSql, data):
　　　　‘‘‘刪除多條sql資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　if data is not None:
　　　　　　　　cursor = self.get_Cursor()
　　　　　　　　for d in data:
　　　　　　　　　　if self.print_sql:
　　　　　　　　　　　　print(‘執行sql:[{}],參數:[{}]‘.format(strSql, d))
　　　　　　　　　　cursor.execute(strSql, d)
　　　　　　　　　　self.commit()
　　　　　　　　self.close_Cursor(cursor)
　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))

　　def delete_Data(self,strSql):
　　　　‘‘‘刪除一條sql資料‘‘‘
　　　　if strSql is not None and strSql != ‘‘:
　　　　　　if self.print_sql:
　　　　　　　　print(‘執行sql:[{}]‘.format(strSql))
　　　　　　cursor = self.get_Cursor()
　　　　　　cursor.execute(strSql)
　　　　　　self.commit()
　　　　　　self.close_Cursor(cursor)
　　　　else:
　　　　　　print(‘the [{}] is empty or equal None!‘.format(strSql))

#########################################################################
#測試代碼
#########################################################################

 

db = SqliteDB(‘e:\\test.db‘)

#刪除資料表
db.drop_Table(‘person‘)

#建立資料庫表
create_table_sql = ‘‘‘CREATE TABLE `person` (
`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,
PRIMARY KEY (`id`)
)‘‘‘

db.create_Table(create_table_sql)

#刪除資料、
delSql=‘delete from person ‘
db.delete_Data(delSql)

#插入資料測試,插入一條語句
insert_sql =‘‘‘INSERT INTO person VALUES (3, ‘xiaoli‘, ‘女‘, 18, ‘山東‘)‘‘‘
db.insert_Data(insert_sql)

#插入資料測試,插入多條語句

insert_sql = ‘‘‘INSERT INTO person values (?, ?, ?, ?, ?)‘‘‘
‘‘‘
data = [[1, ‘xiaowang‘, u‘男‘, 20, u‘廣東‘],
[2, ‘xiaozhang‘, u‘男‘, 22, u‘河南‘],
[3, ‘xiaoli‘, u‘男‘, 18, u‘山東‘],
[4, ‘xiaoliu‘, u‘女‘, 21, u‘山西‘]]
‘‘‘

data = [[1, ‘xiaowang‘, ‘男‘, 20, ‘廣東‘],
[2, ‘xiaozhang‘, ‘男‘, 22, ‘河南‘],
[3, ‘xiaoli‘, ‘男‘, 18, ‘山東‘],
[4, ‘xiaoliu‘, ‘女‘, 21, ‘山西‘]]

for item in data:
item[2] = unicode(item[2],‘utf8‘)
item[4] = unicode(item[4],‘utf8‘)

db.insert_MultiData(insert_sql,data)

#查詢資料
print db.get_All_Item(‘select * from person‘)

python 操作sqlite資料庫