api-mapper / db.py
tanbushi's picture
db
daa9d8a
raw
history blame
1.62 kB
import sqlite3
import json
class Db():
def __init__(self, db_path):
self.db_path = db_path
# print(self.db_path)
def list_query(self, query):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute(query)
conn.commit()
rows = cursor.fetchall()
# print('results')
# print(results)
cursor.close()
conn.close()
json_rows = []
# 为每一行结果创建一个字典,并添加到列表中
for row in rows:
# 使用cursor.description获取列名和相关信息
columns = dict(zip([col[0] for col in cursor.description], row))
json_rows.append(columns)
return json_rows
def insert_query(self, query):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute(query)
conn.commit()
rowcount = cursor.rowcount
return f"插入的行数: {rowcount}"
def update_query(self, query):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute(query)
conn.commit()
rowcount = cursor.rowcount
return f"影响的行数: {rowcount}"
# from fastapi import FastAPI
# # 依赖注入,获取 FastAPI 应用实例
# def get_app() -> FastAPI:
# return FastAPI()
# def execute_query(query):
# CREATE TABLE api_keys (
# api_key TEXT NOT NULL,
# type TEXT NOT NULL,
# status INTEGER NOT NULL,
# idx INTEGER NOT NULL,
# PRIMARY KEY (api_key, type, status, idx)
# );