python3 PyQt5 SQLite

其他 2020-04-18 11:12:14 阅读次数: 0

Python3操作 SQLite

# coding:utf-8

import os
import sqlite3


class SQLiteTools():
    def __init__(self):
        pass

    # region 创建数据库连接和数据表

    def createConnection(self, DB_name):
        ''' 创建数据库连接 '''
        #创建数据库,如果存在则打开，否则创建该数据库
        self.con = sqlite3.connect(DB_name)
        #创建一个游标对象
        self.cur = self.con.cursor()

    def selectTableExist(self, tbname):
        '''
        判断指定数据表是否存在
        :param tbname: 数据表名称
        :return: 存在范围True,否则返回False
        '''
        sql = "select * from sqlite_master where type='table' and name = '" + tbname + "';"
        self.cur.execute(sql)
        cursor = self.cur.fetchall()
        if cursor:
            return True
        else:
            return False

    def createSQLtable(self, tbname):
        '''
        创建通用数据表，默认第一列为主键，名称:ID，类型:INTEGER, 自增
        :param tbname: 数据表名称
        '''
        # CREATE TABLE if not exists 表名 (ID INTEGER PRIMARY KEY AUTOINCREMENT);
        sql = u"CREATE TABLE if not exists " + tbname + u" (ID INTEGER PRIMARY KEY AUTOINCREMENT);"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

    # endregion


    # region 添加表格数据

    def addSQLtableColumn(self, tbname, columnName, genre):
        '''
        指定数据表添加列
        :param tbname: 表名
        :param columnName: 列名
        :param genre: 添加列类型
        '''
        # ALTER TABLE 表名 ADD 列名 列类型;
        sql = u"ALTER TABLE " + tbname + u" ADD " + columnName + " " + genre +";"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

    def addSQLtableRow(self, tbname, rowNum):
        '''
        指定数据表添加指定行
        :param tbname: 表格名称
        :param rowNum: 行数
        :return: None
        '''
        # INSERT INTO 表名 (ID) VALUES (行);
        rownum = [(i,) for i in range(rowNum)]
        sql = "INSERT INTO " + tbname + " (ID) VALUES (?);"
        self.cur.executemany(sql, rownum)

        # for row in range(1, rowNum+1):
        #     sql = "INSERT INTO " + tbname + " (ID) VALUES (" + str(row) + ");"
        #     self.cur.execute(sql)

        self.con.commit() # 提交更新至数据库文件

    def setSQLtableValue(self, tbname, column, row, value):
        '''
        更新数据表指定位置的值
        :param tbname: 数据表名称
        :param row: 行数
        :param column: 列数
        :param value: 值
        '''
        # UPDATE 表名 SET 列名=值 WHERE ID=行;
        sql = u"UPDATE " + tbname + u" SET " +column+ "='" +value+ "' WHERE ID=" +str(row)+ ";"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

    # endregion


    # region 获取表格数据
    def getSQLtableRowNum(self, tbname):
        '''
        获取指定表格总行数
        :param tbname: 表格名称
        :return: 行数
        '''
        result = 0
        sql = "SELECT COUNT(*) FROM " + tbname + ";"
        self.cur.execute(sql)
        for value in self.cur:
            return value[0]
        return result

    def getSQLtableColumnName(self, tbname):
        '''
        获取指定表所有字段名称
        :param taname:  表格名称
        :return:  返回列名称列表
        '''
        sql = "PRAGMA table_info([" +tbname+ "])"
        self.cur.execute(sql)
        NameList = []
        for value in self.cur:
            result = value[1]
            if result and result != "ID":
                NameList.append(result)
        return NameList

    def getSQLtableValue(self, tbname, column, row):
        '''
        读取指定数据表的指定数据
        :param tbname: 数据表名称
        :param row: 数据表行
        :param column: 数据表列
        :return 返回查询到的值
        '''
        #SELECT 列名 FROM 表名 WHERE ID = 行号;
        sql = "SELECT " +column+ " FROM " +tbname+ " WHERE ID=" +str(row)+ ";"
        self.cur.execute(sql)
        for value in self.cur:
            return value[0]

    def getSQLtableColumn(self, tbname, column):
        '''
        读取数据表指定列的所有数据
        :param tbname: 数据表名称
        :param column: 列名称
        :return 返回查询到的值列表
        '''
        #SELECT 列名 FROM 表名;
        sql = "SELECT " +column+ " FROM " +tbname+ ";"
        valueList = []
        self.cur.execute(sql)
        for value in self.cur:
            valueList.append(value[0])
        return valueList

    def getSQLtableRow(self, tbname, row):
        '''
        获取指定表格指定行数据
        :param tbname: 表格名称
        :param row: 行数
        :return: 值列表
        '''
        # SELECT * FROM 表名 WHERE ID=行数;
        sql = "SELECT * FROM " + tbname + " WHERE ID="+ str(row) +";"
        self.cur.execute(sql)
        for value in self.cur:
            valueList = list(value)
            valueList.pop(0)
            return valueList

    # endregion


    # region 操作数据表

    def delSQLtableRow(self, tbname, startRow, rowNum=1):
        '''
        指定数据表删除指定行数
        :param tbname: 表格名称
        :param rowNum: 开始行数
        :param rowNum: 行数
        '''
        # DELETE FROM 表名 WHERE ID = 行;
        rowList = [(i,) for i in range(startRow, startRow+rowNum)]
        sql = "DELETE FROM " + tbname + " WHERE ID = ?;"
        self.cur.executemany(sql, rowList)

        # for row in range(startRow, startRow + rowNum):
        #     sql = "DELETE FROM " + tbname + " WHERE ID = " + str(row) + ";"
        #     self.cur.execute(sql)

        self.con.commit()  # 提交更新至数据库文件

    def delSQLtableColumn(self, tbname, columnName):
        '''
        指定数据表删除指定列
        SQLite不支持删除列操作，采用复制新表的方式。
        :param tbname: 表名
        :param columnName: 列名
        :param genre: 添加列类型
        '''
        # ALTER TABLE 表名 drop 列名;   #SQLite不支持
        tbName = "tbName0"
        CloumnNameList = self.getSQLtableColumnName(tbname)
        if columnName in CloumnNameList:
            CloumnNameList.remove(columnName)
        CloumnNameList.insert(0, "ID")
        strName = ','.join(CloumnNameList)
        # create table 零时表名 as select 所有列名称 from 表名 where 1 = 1;
        sql = u"create table "+tbName+" as select "+strName+" from "+tbname+" where 1 = 1"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

        self.delSQLtable(tbname)
        self.removeSQLtableName(tbName, tbname)

    def removeSQLtableName(self, tbname, newTbname):
        '''
        重命名表格
        :param oldTbName:  旧表格名称
        :param newTbName:  新表格名称
        :return:  None
        '''
        # alter table 旧表格名称 rename to 新表格名称;
        sql = u"alter table "+tbname+" rename to "+newTbname+";"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

    def ClearSQLtable(self, tbname):
        '''
        清空指定数据表
        :param tbname: 表名
        '''
        # DELETE FROM table_name WHERE[condition];
        sql = "DELETE FROM " +tbname+ ";"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

    def delSQLtable(self, tbname):
        '''
        删除指定表
        :param tbname:
        :return: None
        '''
        # drop table record;
        sql = u"drop table "+tbname+";"
        self.cur.execute(sql)
        self.con.commit()  # 提交更新至数据库文件

    # endregion


if __name__ == '__main__':
    print("Python SQLite实例")

    class sqliteTest():
        def __init__(self):
            self.DB_File = "./sqlite.db"
            self.tableName = "sqlite"

            # 如果存在mysql.ini先删除，方便下列代码的测试
            if os.path.exists(self.DB_File):
                os.remove(self.DB_File)

            self.sqlite = SQLiteTools()
            self.sqlite.createConnection(self.DB_File)

        # 创建表
        def createTable(self):
            if not self.sqlite.selectTableExist(self.tableName):
                self.sqlite.createSQLtable(self.tableName)

        # 添加数据
        def addData(self):
            self.sqlite.addSQLtableColumn(self.tableName, "month1", "text")
            self.sqlite.addSQLtableColumn(self.tableName, "month2", "text")
            self.sqlite.addSQLtableColumn(self.tableName, "month3", "text")
            self.sqlite.addSQLtableColumn(self.tableName, "month4", "text")

            self.sqlite.addSQLtableRow(self.tableName, 30)

            self.sqlite.setSQLtableValue(self.tableName, "month1", 0, "1-0")
            self.sqlite.setSQLtableValue(self.tableName, "month1", 1, "1-1")
            self.sqlite.setSQLtableValue(self.tableName, "month2", 2, "2-2")
            self.sqlite.setSQLtableValue(self.tableName, "month2", 3, "2-3")

        # 获取数据
        def getData(self):
            value = self.sqlite.getSQLtableRowNum(self.tableName)
            print("数据表总行数：", value)

            value = self.sqlite.getSQLtableColumnName(self.tableName)
            print("所有列的名称(不包括ID)：", value)

            value = self.sqlite.getSQLtableValue(self.tableName, "month1", 0)
            print("month1列,第一行的值： ", value)

            value  = self.sqlite.getSQLtableColumn(self.tableName, "month2")
            print("month2列的所有值： ", value)

            value = self.sqlite.getSQLtableRow(self.tableName, 1)
            print("第二行的所有值(不包括ID)： ", value)

        # 删除数据
        def delData(self):
            value = self.sqlite.getSQLtableRowNum(self.tableName)
            print("删除前数据表总行数：", value)
            self.sqlite.delSQLtableRow(self.tableName, 10, 5)
            value = self.sqlite.getSQLtableRowNum(self.tableName)
            print("删除后数据表总行数：", value)

            value = self.sqlite.getSQLtableColumnName(self.tableName)
            print("删除前所有列名(不包括ID)", value)
            self.sqlite.delSQLtableColumn(self.tableName, "month3")
            value = self.sqlite.getSQLtableColumnName(self.tableName)
            print("删除后所有列名(不包括ID)", value)

            self.sqlite.ClearSQLtable(self.tableName)
            self.sqlite.delSQLtable(self.tableName)

    win = sqliteTest()  #实例工具类，创建连接
    win.createTable()  # 创建表
    win.addData()  # 添加数据
    win.getData()  # 获取数据
    win.delData()  # 删除数据

PyQt5 操作 SQLite

# coding:utf-8

import os
from PyQt5.QtSql import QSqlDatabase, QSqlQuery


class SQLiteTools():
    def __init__(self):
        pass

    # region 创建数据库连接和数据表

    def createConnection(self, DB_name):
        '''
        创建数据库连接
        :param DB_name: 数据库名称
        :return: null
        '''
        #选择数据库类型，这里为sqlite3数据库
        db=QSqlDatabase.addDatabase("QSQLITE")
        #创建数据库test0.db,如果存在则打开，否则创建该数据库
        db.setDatabaseName(DB_name)
        #打开数据库
        db.open()
        #此语句必须放在打开数据库之后
        self.que = QSqlQuery()

    def selectTableExist(self, tbname):
        '''
        判断指定数据表是否存在
        :param tbname: 表格名称
        :return: 在返回True,否则返回False
        '''
        sql = u"select * from sqlite_master where type='table' and name = '" + tbname + u"';"
        self.que.exec_(sql)
        return self.que.next()

    def createSQLtable(self, tbname):
        '''
        创建通用数据表，默认第一列为主键，名称:ID，类型:INTEGER, 自增
        :param tbname: 数据表名称
        :return:  null
        '''
        # CREATE TABLE if not exists 表名 (ID INTEGER PRIMARY KEY AUTOINCREMENT);
        sql = u"CREATE TABLE if not exists " + tbname + u" (ID INTEGER PRIMARY KEY AUTOINCREMENT);"
        self.que.exec_(sql)

    # endregion


    # region 添加表格数据

    def addSQLtableColumn(self, tbname, columnName, genre):
        '''
        指定数据表添加列
        :param tbname: 表名
        :param columnName: 列名
        :param genre: 添加列类型
        :return: null
        '''
        # ALTER TABLE 表名 ADD 列名 列类型;
        sql = u"ALTER TABLE " + tbname + u" ADD " + columnName + " " + genre +";"
        self.que.exec_(sql)

    def addSQLtableRow(self, tbname, rowNum):
        '''
        指定数据表添加指定行
        :param tbname: 表格名称
        :param rowNum: 行数
        :return: null
        '''
        # INSERT INTO 表名 (ID) VALUES (行);
        for row in range(rowNum):
            sql = "INSERT INTO " + tbname + " (ID) VALUES (" + str(row) + ");"
            self.que.exec_(sql)

    def setSQLtableValue(self, tbname, column, row, value):
        '''
        更新数据表指定位置的值
        :param tbname: 数据表名称
        :param column: 行数
        :param row: 列数
        :param value: 值
        :return: null
        '''
        # UPDATE 表名 SET 列名=值 WHERE ID=行;
        sql = u"UPDATE " + tbname + u" SET " +column+ "='" +value+ "' WHERE ID=" +str(row)+ ";"
        self.que.exec_(sql)

    # endregion


    # region 获取表格数据
    def getSQLtableRowNum(self, tbname):
        '''
        获取指定表格总行数
        :param tbname: 表格名称
        :return: 行数
        '''
        result = 0
        sql = "SELECT COUNT(*) FROM " + tbname + ";"
        self.que.exec_(sql)
        if self.que.next():
            result = self.que.value(0)
        return result

    def getSQLtableColumnName(self, tbname):
        '''
        获取指定表所有字段名称
        :param taname:  表格名称
        :return:  列名称列表
        '''
        sql = "PRAGMA table_info([" +tbname+ "])"
        self.que.exec_(sql)
        NameList = []
        while self.que.next():
            result = self.que.value(1)
            if result and result != "ID":
                NameList.append(result)
        return NameList

    def getSQLtableValue(self, tbname, column, row):
        '''
        读取指定数据表的指定数据
        :param tbname: 数据表名称
        :param row: 数据表行
        :param column: 数据表列
        :return: 值
        '''
        #SELECT 列名 FROM 表名 WHERE ID = 行号;
        value = 0
        sql = "SELECT " +column+ " FROM " +tbname+ " WHERE ID=" +str(row)+ ";"
        self.que.exec_(sql)
        if self.que.next():
            value = self.que.value(0)
        return value

    def getSQLtableColumn(self, tbname, column):
        '''
        读取数据表指定列的所有数据
        :param tbname: 数据表名称
        :param column: 列名称
        :return: 值列表
        '''
        #SELECT 列名 FROM 表名;
        sql = "SELECT " +column+ " FROM " +tbname+ ";"
        value_list = []
        if self.que.exec_(sql):
            column_index = self.que.record().indexOf(column) #获取列索引值
            while self.que.next():
                value = self.que.value(column_index)
                value_list.append(value)
        return value_list

    def getSQLtableRow(self, tbname, row):
        '''
        获取指定表格指定行数据
        :param tbname: 表格名称
        :param row: 行数
        :return: 值列表
        '''
        # SELECT * FROM 表名 WHERE ID=行数;
        columnNum = len(self.getSQLtableColumnName(tbname))
        sql = "SELECT * FROM " + tbname + " WHERE ID="+ str(row) +";"
        self.que.exec_(sql)
        valueList = []
        while self.que.next():
            for i in range(1, columnNum+1):
                result = self.que.value(i)
                valueList.append(result)
        return valueList

    # endregion


    # region 操作数据表

    def delSQLtableColumn(self, tbname, columnName):
        '''
        指定数据表删除指定列
        SQLite不支持删除列操作，采用复制新表的方式。
        :param tbname: 表名
        :param columnName: 列名
        :return: null
        '''
        # ALTER TABLE 表名 drop 列名;   #SQLite不支持
        tbName = "tbName0"
        CloumnNameList = self.getSQLtableColumnName(tbname)
        if columnName in CloumnNameList:
            CloumnNameList.remove(columnName)
        CloumnNameList.insert(0, "ID")
        strName = ','.join(CloumnNameList)
        # create table 零时表名 as select 所有列名称 from 表名 where 1 = 1;
        sql = u"create table "+tbName+" as select "+strName+" from "+tbname+" where 1 = 1"
        self.que.exec_(sql)
        self.delSQLtable(tbname)
        self.removeSQLtableName(tbName, tbname)

    def delSQLtableRow(self, tbname, startRow, rowNum=1):
        '''
        指定数据表删除指定行数
        :param tbname: 表格名称
        :param startRow: 开始行数
        :param rowNum: 要删除总行数
        :return: null
        '''
        # DELETE FROM 表名 WHERE ID = 行;
        for row in range(startRow, startRow + rowNum):
            sql = "DELETE FROM " + tbname + " WHERE ID = " + str(row) + ";"
            self.que.exec_(sql)

    def removeSQLtableName(self, tbname, newTbname):
        '''
        重命名表格
        :param tbname:  旧表格名称
        :param newTbname:  新表格名称
        :return:  null
        '''
        # alter table 旧表格名称 rename to 新表格名称;
        sql = u"alter table "+tbname+" rename to "+newTbname+";"
        self.que.exec_(sql)

    def ClearSQLtableRowValue(self, tbname):
        '''
        清空指定数据表
        :param tbname: 表名
        :return:  null
        '''
        # DELETE FROM table_name WHERE[condition];
        sql = "DELETE FROM " +tbname+ ";"
        self.que.exec_(sql)

    def delSQLtable(self, tbname):
        '''
        删除指定表
        :param tbname:
        :return: null
        '''
        # drop table record;
        sql = u"drop table "+tbname+";"
        self.que.exec_(sql)

    # endregion


if __name__ == '__main__':
    print("PyQt5 SQLite实例")
    class sqliteTest():
        def __init__(self):
            self.PathFile = "./sqlite.db"
            self.tableName = "sqlite"

            # 如果存在mysql.ini先删除，方便下列代码的测试
            if os.path.exists(self.PathFile):
                os.remove(self.PathFile)

            self.sqlite = SQLiteTools()
            self.sqlite.createConnection(self.PathFile)

        #创建表
        def createTable(self):
            if not self.sqlite.selectTableExist(self.tableName):
                self.sqlite.createSQLtable(self.tableName)

        # 添加数据
        def addData(self):
            self.sqlite.addSQLtableColumn(self.tableName, "month1", "text")
            self.sqlite.addSQLtableColumn(self.tableName, "month2", "text")
            self.sqlite.addSQLtableColumn(self.tableName, "month3", "text")
            self.sqlite.addSQLtableColumn(self.tableName, "month4", "text")

            self.sqlite.addSQLtableRow(self.tableName, 30)

            self.sqlite.setSQLtableValue(self.tableName, "month1", 0, "1-0")
            self.sqlite.setSQLtableValue(self.tableName, "month1", 1, "1-1")
            self.sqlite.setSQLtableValue(self.tableName, "month2", 2, "2-2")
            self.sqlite.setSQLtableValue(self.tableName, "month2", 3, "2-3")

        # 获取数据
        def getData(self):
            value = self.sqlite.getSQLtableRowNum(self.tableName)
            print("数据表总行数：", value)

            value = self.sqlite.getSQLtableColumnName(self.tableName)
            print("所有列的名称(不包括ID列)：", value)

            value = self.sqlite.getSQLtableValue(self.tableName, "month1", 0)
            print("month1列，第一行的值：", value)

            value  = self.sqlite.getSQLtableColumn(self.tableName, "month2")
            print("month2列的所有值：", value)

            value = self.sqlite.getSQLtableRow(self.tableName, 2)
            print("第三行的所有值(不包括ID)： ", value)

        # 删除数据
        def delData(self):
            value = self.sqlite.getSQLtableRowNum(self.tableName)
            print("删除前数据表总行数：", value)
            self.sqlite.delSQLtableRow(self.tableName, 10, 5)
            value = self.sqlite.getSQLtableRowNum(self.tableName)
            print("删除后数据表总行数：", value)

            value = self.sqlite.getSQLtableColumnName(self.tableName)
            print("删除前所有列名(不包括ID)：", value)
            self.sqlite.delSQLtableColumn(self.tableName, "month3")
            value = self.sqlite.getSQLtableColumnName(self.tableName)
            print("删除后所有列名(不包括ID)", value)
            self.sqlite.ClearSQLtableRowValue(self.tableName)
            self.sqlite.delSQLtable(self.tableName)

    win = sqliteTest()
    win.createTable()
    win.addData()
    win.getData()
    win.delData()
z178443085
发布了34 篇原创文章 · 获赞 54 · 访问量 2万+
私信 关注

猜你喜欢

转载自blog.csdn.net/weixin_42686768/article/details/103411990
今日推荐
面壁智能发布 Eurux-8x22B 开源大模型 —— 堪称「理科状元」
开源日报 | 谷歌扶持鸿蒙上位；开源Rabbit R1；Docker加持的安卓手机；微软的焦虑和野心；海尔电器把开放平台关了
中国码农的“35岁魔咒”
蘭雅 CorelDRAW 插件 2024.5.1 国际劳动节版，免费下载
Arc Browser for Windows 1.0 正式 GA
90后程序员开发视频搬运软件、不到一年获利超 700 万，结局很刑！
周排行
【转】spring中对控制反转和依赖注入的理解
tms webcore 安装和使用
java程序员进阶相关书籍
SpringMVC接受请求参数、
如何保存训练好的机器学习模型
MyEclipse、Eclipse设置项目JDK的三个地方
商超行业微信小程序开发定制一般多少钱 （行业技术人员解读）
Markdown编辑器语言——30分钟入门到到精通
Linux系统下MongoDB的简单安装与基本操作
Power Strings
每日归档
更多
2024-05-07(14)
2024-05-06(40)
2024-05-05(0)
2024-05-04(7)
2024-05-03(19)
2024-05-02(0)
2024-05-01(4)
2024-04-30(1)
2024-04-29(40)
2024-04-28(0)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022