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('数据库已关闭')
Python中自用的SQLITE模块
最新推荐文章于 2024-08-19 13:32:52 发布