Update playwright/README.md

playwright/README.md

@@ -49,35 +49,27 @@ class DB(sqlite3.Connection):
     def __init__(self, db_name=".db.sqlite"):
         super().__init__(db_name)
         with self:
-            self.execute(
+            cur = self.execute("""
-                """
                 CREATE TABLE IF NOT EXISTS kv_store
                 (key TEXT PRIMARY KEY, value BLOB)
-            """
+            """)
-            )
 
     def __setitem__(self, key, value):
-        if not isinstance(value, bytes):
+        value = value if isinstance(value, bytes) else json.dumps(value)
-            value = json.dumps(value)
         with self:
-            cur = self.execute(
+            cur = self.execute("""
-                """
                 INSERT OR REPLACE INTO kv_store
                 (key, value) VALUES (?, ?)
-            """,
+            """, (key, value))
-                (key, value),
-            )
             return {"modified_count": cur.rowcount}
 
     def __getitem__(self, key):
         with self:
-            result = self.execute(
+            cur = self.execute("""
-                """
                 SELECT value FROM kv_store
                 WHERE key = ?
-            """,
+            """, (key,))
-                (key,),
+            result = cur.fetchone()
-            ).fetchone()
         if result:
             if isinstance(result[0], str):
                 try:
@@ -88,42 +80,32 @@ class DB(sqlite3.Connection):
 
     def delete(self, key):
         with self:
-            cur = self.execute(
+            cur = self.execute("""
-                """
                 DELETE FROM kv_store 
                 WHERE key = ?
-            """,
+            """, (key,))
-                (key,),
-            )
             return {"deleted_count": cur.rowcount}
 
     def keys(self, pattern="*"):
-        pattern = pattern.translate(
+        pattern = pattern.translate(str.maketrans(
-            str.maketrans({"\\": "\\\\", "%": "\\%", "_": "\\_", "*": "%", "?": "_"})
+            {"\\": "\\\\", "%": "\\%", "_": "\\_", "*": "%", "?": "_"}))
-        )
         with self:
-            result = self.execute(
+            cur = self.execute("""
-                """
                 SELECT key FROM kv_store
                 WHERE key LIKE ? ESCAPE '\\'
-            """,
+            """, (pattern,))
-                (pattern,),
+            result = cur.fetchall()
-            ).fetchall()
         return [row[0] for row in result]
 
-    def __repr__(self):
+    def __repr__(self): return repr(self.keys())
-        return repr(self.keys())
 
     def exists(self, key):
         with self:
-            cur = self.execute(
+            cur = self.execute("""
-                """
                 SELECT 1 FROM kv_store
                 WHERE key = ?
                 LIMIT 1
-            """,
+            """, (key,))
-                (key,),
-            )
             return bool(cur.fetchone())