|
import time
|
|
|
|
def db_set(key, value, db_conn):
|
|
if not db_conn:
|
|
return {"status": "error", "error": "Database not initialized"}
|
|
|
|
try:
|
|
cursor = db_conn.cursor()
|
|
cursor.execute("""INSERT OR REPLACE INTO kv_store (key, value, timestamp)
|
|
VALUES (?, ?, ?)""", (key, value, time.time()))
|
|
db_conn.commit()
|
|
return {"status": "success", "message": f"Set {key}"}
|
|
except Exception as e:
|
|
return {"status": "error", "error": str(e)}
|
|
|
|
def db_get(key, db_conn):
|
|
if not db_conn:
|
|
return {"status": "error", "error": "Database not initialized"}
|
|
|
|
try:
|
|
cursor = db_conn.cursor()
|
|
cursor.execute("SELECT value FROM kv_store WHERE key = ?", (key,))
|
|
result = cursor.fetchone()
|
|
if result:
|
|
return {"status": "success", "value": result[0]}
|
|
else:
|
|
return {"status": "error", "error": "Key not found"}
|
|
except Exception as e:
|
|
return {"status": "error", "error": str(e)}
|
|
|
|
def db_query(query, db_conn):
|
|
if not db_conn:
|
|
return {"status": "error", "error": "Database not initialized"}
|
|
|
|
try:
|
|
cursor = db_conn.cursor()
|
|
cursor.execute(query)
|
|
|
|
if query.strip().upper().startswith('SELECT'):
|
|
results = cursor.fetchall()
|
|
columns = [desc[0] for desc in cursor.description] if cursor.description else []
|
|
return {"status": "success", "columns": columns, "rows": results}
|
|
else:
|
|
db_conn.commit()
|
|
return {"status": "success", "rows_affected": cursor.rowcount}
|
|
except Exception as e:
|
|
return {"status": "error", "error": str(e)}
|