Rename list_keys()

3 files changed, 14 insertions, 8 deletions

diff --git a/README.md b/README.md
index 0a0aa82..651832c 100644
--- a/README.md
+++ b/README.md
@@ -49,6 +49,12 @@ Load from file:
 user = stash.load("yuval")
 ```
 
+List all keys existing in stash:
+
+```python
+keys = stash.list_keys()
+```
+
 ## License
 
 [MIT](LICENSE)
diff --git a/tests/test_stash.py b/tests/test_stash.py
index 6a11db9..6b9e47f 100644
--- a/tests/test_stash.py
+++ b/tests/test_stash.py
@@ -23,11 +23,11 @@ def test_stash():
     stash = YmlStash(User, "/tmp/")
     yuval = User(name="yuval", age=42)
     stash.save(yuval, "foo")
-    assert stash.list_all_keys() == ["foo"]
+    assert stash.list_keys() == ["foo"]
     obj = stash.load("foo")
     assert obj == yuval
     stash.drop()
-    assert stash.list_all_keys() == []
+    assert stash.list_keys() == []
 
 
 def test_key_field():
@@ -46,10 +46,10 @@ def test_key_field():
     stash = YmlStash(Dog, "/tmp/")
     terra = Dog(name="terra")
     stash.save(terra)
-    assert stash.list_all_keys() == ["terra"]
+    assert stash.list_keys() == ["terra"]
 
     terra = Dog(name="terra")
     stash.save(terra, key="dupe")  # override key
-    assert stash.list_all_keys() == ["terra", "dupe"]
+    assert stash.list_keys() == ["terra", "dupe"]
 
     stash.drop()
diff --git a/ymlstash/stash.py b/ymlstash/stash.py
index 55bf016..a0c585a 100644
--- a/ymlstash/stash.py
+++ b/ymlstash/stash.py
@@ -36,13 +36,13 @@ class YmlStash:
         with open(self._get_path(key), "w") as f:
             f.write(yaml.dump(asdict(obj)))
 
-    def _list_all_files(self):
+    def _list_files(self):
         return [f for f in os.listdir(self.path) if f.endswith(self.file_suffix)]
 
-    def list_all_keys(self):
-        return [f.replace(self.file_suffix, "") for f in self._list_all_files()]
+    def list_keys(self):
+        return [f.replace(self.file_suffix, "") for f in self._list_files()]
 
     def drop(self):
-        for f in self._list_all_files():
+        for f in self._list_files():
             if f.endswith(self.file_suffix):
                 os.remove(self.path / f)