From 0db070bcf2d44fa85e15a5f9d0194b1b7a80e598 Mon Sep 17 00:00:00 2001 From: jay817 Date: Fri, 23 Aug 2024 08:07:11 -0400 Subject: [PATCH] Update sqlite3/db.py --- sqlite3/db.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/sqlite3/db.py b/sqlite3/db.py index ecf1d89..ab7e0b8 100644 --- a/sqlite3/db.py +++ b/sqlite3/db.py @@ -49,4 +49,13 @@ class DB(sqlite3.Connection): ''', (pattern,)).fetchall() return [row[0] for row in result] - def __repr__(self): return repr(self.keys()) \ No newline at end of file + def __repr__(self): return repr(self.keys()) + + def exists(self, key): + with self: + cur = self.execute(''' + SELECT 1 FROM kv_store + WHERE key = ? + LIMIT 1 + ''', (key,)) + return bool(cur.fetchone()) \ No newline at end of file