APIGenAI / database.py
vishwas3086's picture
Create database.py
29fee85 verified
raw
history blame contribute delete
715 Bytes
import sqlite3
import os
DB_PATH = "results.db"
# Agar database file exist nahi karti toh create karo
if not os.path.exists(DB_PATH):
conn = sqlite3.connect(DB_PATH)
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS results (
id INTEGER PRIMARY KEY AUTOINCREMENT,
api_name TEXT,
test_status TEXT,
run_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
)
''')
conn.commit()
conn.close()
# Function to get past results
def show_past_results():
conn = sqlite3.connect(DB_PATH)
cursor = conn.cursor()
cursor.execute("SELECT * FROM results")
data = cursor.fetchall()
conn.close()
return data