python操作sqlite数据库

Question

Answer 1

# coding:utf-8
# Project   :CK10S
# Name      :test
# Author    :Nini
# Date      :2021-09-23 09:21:34
# Remarks   :


import sqlite3


class Test:
    def __init__(self):
        self.conn = sqlite3.connect('test.db')
        self.cursor = self.conn.cursor()

    def close_db(self):
        """关闭数据库连接"""
        self.cursor.close()
        self.conn.close()

    def create_tab(self):
        """创建表"""
        sql = """CREATE TABLE "stu" (
              "id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
              "stu_name" TEXT,
              "stu_grade" TEXT
            );
        """
        try:
            self.cursor.execute(sql)
        except Exception as e:
            print('create tab error:', e)

    def insert_data(self, data):
        """插入单条数据"""
        sql = 'insert into stu (stu_name,stu_grade) values (?,?);'
        try:
            self.cursor.execute(sql, data)
            self.conn.commit()
        except Exception as e:
            self.conn.rollback()
            print('insert data error:', e)

    def insert_all_data(self, data_list):
        """插入所有数据"""
        for i in data_list:
            self.insert_data(i)


if __name__ == '__main__':
    data_list = [('a', 13), ('b', 50), ('c', 99)]
    obj = Test()
    obj.create_tab()
    obj.insert_all_data(data_list)
    obj.close_db()

希望采纳！！！