96 lines
2.2 KiB
Python
96 lines
2.2 KiB
Python
|
# coding: utf-8
|
|||
|
|
# Author：tajochen
|
|||
|
|
|
|||
|
|
import sqlite3
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SimpleSQLite3Tool:
|
|||
|
|
"""
|
|||
|
|
simpleToolSql for sqlite3
|
|||
|
|
简单数据库工具类
|
|||
|
|
编写这个类主要是为了封装sqlite，继承此类复用方法
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
def __init__(self, filename="stsql"):
|
|||
|
|
"""
|
|||
|
|
初始化数据库，默认文件名 stsql.db
|
|||
|
|
filename：文件名
|
|||
|
|
"""
|
|||
|
|
self.filename = filename
|
|||
|
|
self.db = sqlite3.connect(self.filename)
|
|||
|
|
self.c = self.db.cursor()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def close(self):
|
|||
|
|
"""
|
|||
|
|
关闭数据库
|
|||
|
|
"""
|
|||
|
|
self.c.close()
|
|||
|
|
self.db.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def execute(self, sql, param=None):
|
|||
|
|
"""
|
|||
|
|
执行数据库的增、删、改
|
|||
|
|
sql：sql语句
|
|||
|
|
param：数据，可以是list或tuple，亦可是None
|
|||
|
|
return：成功返回True
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
if param is None:
|
|||
|
|
self.c.execute(sql)
|
|||
|
|
else:
|
|||
|
|
if type(param) is list:
|
|||
|
|
self.c.executemany(sql, param)
|
|||
|
|
else:
|
|||
|
|
self.c.execute(sql, param)
|
|||
|
|
count = self.db.total_changes
|
|||
|
|
self.db.commit()
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e)
|
|||
|
|
return False, e
|
|||
|
|
if count > 0:
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def query(self, sql, param=None):
|
|||
|
|
"""
|
|||
|
|
查询语句
|
|||
|
|
sql：sql语句
|
|||
|
|
param：参数,可为None
|
|||
|
|
return：成功返回True
|
|||
|
|
"""
|
|||
|
|
if param is None:
|
|||
|
|
self.c.execute(sql)
|
|||
|
|
else:
|
|||
|
|
self.c.execute(sql, param)
|
|||
|
|
return self.c.fetchall()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# def set(self,table,field=" * ",where="",isWhere=False):
|
|||
|
|
# self.table = table
|
|||
|
|
# self.filed = field
|
|||
|
|
# if where != "" :
|
|||
|
|
# self.where = where
|
|||
|
|
# self.isWhere = True
|
|||
|
|
# return True
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 数据库文件位置
|
|||
|
|
sql = SimpleSQLite3Tool("../test.db")
|
|||
|
|
f = sql.execute("create table test (id int not null,name text not null,age int);")
|
|||
|
|
print("ok")
|
|||
|
|
sql.execute("insert into test (id,name,age) values (?,?,?);", [(1, 'abc', 15), (2, 'bca', 16)])
|
|||
|
|
res = sql.query("select * from test;")
|
|||
|
|
print(res)
|
|||
|
|
sql.execute("insert into test (id,name) values (?,?);", (3, 'bac'))
|
|||
|
|
res = sql.query("select * from test where id=?;", (3,))
|
|||
|
|
res = sql.query("select * from data_collection_info;")
|
|||
|
|
print(res)
|
|||
|
|
sql.close()
|