Add __repr__ to store classes

obstore/python/obstore/store/__init__.pyi

@@ -1,4 +1,6 @@
 # TODO: move to reusable types package
+from pathlib import Path
+
 from ._aws import S3ConfigKey as S3ConfigKey
 from ._aws import S3Store as S3Store
 from ._azure import AzureConfigKey as AzureConfigKey
@@ -16,11 +18,14 @@ class LocalStore:
     Can optionally be created with a directory prefix.
 
     ```py
+    from pathlib import Path
+
     store = LocalStore()
     store = LocalStore(prefix="/path/to/directory")
+    store = LocalStore(prefix=Path("."))
     ```
     """
-    def __init__(self, prefix: str | None = None) -> None: ...
+    def __init__(self, prefix: str | Path | None = None) -> None: ...
 
 class MemoryStore:
     """A fully in-memory implementation of ObjectStore.

pyo3-object_store/src/aws.rs

@@ -140,6 +140,11 @@ impl PyS3Store {
         }
         Ok(Self(Arc::new(builder.build()?)))
     }
+
+    fn __repr__(&self) -> String {
+        let repr = self.0.to_string();
+        repr.replacen("AmazonS3", "S3Store", 1)
+    }
 }
 
 #[derive(Debug, PartialEq, Eq, Hash)]

pyo3-object_store/src/azure.rs

@@ -78,6 +78,11 @@ impl PyAzureStore {
         }
         Ok(Self(Arc::new(builder.build()?)))
     }
+
+    fn __repr__(&self) -> String {
+        let repr = self.0.to_string();
+        repr.replacen("MicrosoftAzure", "AzureStore", 1)
+    }
 }
 
 #[derive(Debug, PartialEq, Eq, Hash)]

pyo3-object_store/src/gcp.rs

@@ -78,6 +78,11 @@ impl PyGCSStore {
         }
         Ok(Self(Arc::new(builder.build()?)))
     }
+
+    fn __repr__(&self) -> String {
+        let repr = self.0.to_string();
+        repr.replacen("GoogleCloudStorage", "GCSStore", 1)
+    }
 }
 
 #[derive(Debug, PartialEq, Eq, Hash)]

pyo3-object_store/src/http.rs

@@ -23,6 +23,10 @@ impl PyHttpStore {
     pub fn into_inner(self) -> Arc<HttpStore> {
         self.0
     }
+
+    fn __repr__(&self) -> String {
+        self.0.to_string()
+    }
 }
 
 #[pymethods]

pyo3-object_store/src/local.rs

@@ -34,4 +34,9 @@ impl PyLocalStore {
         };
         Ok(Self(Arc::new(fs)))
     }
+
+    fn __repr__(&self) -> String {
+        let repr = self.0.to_string();
+        repr.replacen("LocalFileSystem", "LocalStore", 1)
+    }
 }

pyo3-object_store/src/memory.rs

@@ -1,7 +1,9 @@
 use std::sync::Arc;
 
 use object_store::memory::InMemory;
+use pyo3::intern;
 use pyo3::prelude::*;
+use pyo3::types::PyString;
 
 /// A Python-facing wrapper around an [`InMemory`].
 #[pyclass(name = "MemoryStore")]
@@ -13,11 +15,15 @@ impl AsRef<Arc<InMemory>> for PyMemoryStore {
     }
 }
 
-impl PyMemoryStore {
+impl<'py> PyMemoryStore {
     /// Consume self and return the underlying [`InMemory`].
     pub fn into_inner(self) -> Arc<InMemory> {
         self.0
     }
+
+    fn __repr__(&'py self, py: Python<'py>) -> &Bound<'py, PyString> {
+        intern!(py, "MemoryStore")
+    }
 }
 
 #[pymethods]

tests/store/test_local.py

@@ -0,0 +1,17 @@
+from pathlib import Path
+
+import obstore as obs
+from obstore.store import LocalStore
+
+
+def test_local_store():
+    here = Path(".")
+    store = LocalStore(here)
+    list_result = obs.list(store).collect()
+    assert any(x["path"] == "test_local.py" for x in list_result)
+
+
+def test_repr():
+    here = Path(".")
+    store = LocalStore(here)
+    assert repr(store).startswith("LocalStore")