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()