From 36ac6856df74df6b07b2d09bf024e900cfe7d5f1 Mon Sep 17 00:00:00 2001
From: ooojpeg <ooojpeg@orangehalo.net>
Date: Tue, 23 Mar 2021 21:47:37 +0000
Subject: [PATCH] Adding ability to retrieve entries that match string.

---
 pickledb.py | 2 ++
 tests.py    | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/pickledb.py b/pickledb.py
index 160b0d5..c67400a 100644
--- a/pickledb.py
+++ b/pickledb.py
@@ -290,3 +290,5 @@ def deldb(self):
         self._autodumpdb()
         return True
 
+    def getkmatch(self, match):
+        return dict(filter(lambda item: match in item[0], self.db.items()))
diff --git a/tests.py b/tests.py
index 7db5a6d..400d3c8 100644
--- a/tests.py
+++ b/tests.py
@@ -108,6 +108,14 @@ def test_not_dexists(self):
         assert x is False
         self.db.drem('dict')
 
+    def test_key_match(self):
+        self.db.set('key1', 'value')
+        self.db.set('key2', 'value2')
+        self.db.set('something', 'value')
+        x = self.db.getkmatch('key')
+        y = {'key1': 'value', 'key2': 'value2'}
+        assert x == y
+
 
 if __name__ == "__main__":
     tests = TestClass()