Python中自用的SQLITE模块

import sqlite3

class DB:
    def __init__(self, db_name):
        self.conn = sqlite3.connect('数据库地址')  # 连接数据库
        self.c = self.conn.cursor()  # 创建游标
        # 验证 db_name是否正确
        name= [ 'photo','image']
        if db_name in name:
            self.db_name = db_name 
        else:
            raise ValueError("不存在的表")


    def SET_DB(self):  # 读取数据库文件
        self.c.execute(f"SELECT * FROM {self.db_name}")  # 执行SQL查询  (查询 *  FROM 表)
        infos = self.c.fetchall()
        return infos  # 返回查询的信息

    def INSERT_DB(self, title, link):  # 插入数据
        self.c.execute(f'INSERT INTO {self.cos_tag} (title, link) VALUES (?, ?)', (title, link))
        self.conn.commit()  # 提交保存数据
        print(f'采集成功>>{title}>>的数据')

    def UPDATE_DB(self, title):  # 更新数据 打标记
        self.c.execute(f'UPDATE {self.db_name } SET TAG = ?  WHERE NAME = ?', ('A', title,))
        self.conn.commit()  # 提交保存数据
        print(f'数据更新>>{title}的数据完成标识>>已写入数据库')
    def __del__(self):
        self.conn.commit()  # 提交数据
        self.c.close()  # 关闭游标
        self.conn.close()  # 关闭数据库
        print('数据库已关闭')