Update sqlite3/db.py

2024-08-23 08:07:11 -04:00
parent db0af1a6a3
commit 0db070bcf2
1 changed files with 10 additions and 1 deletions
--- 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())
+    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())