string repr for Track and Storage

1 files changed, 11 insertions, 0 deletions

diff --git a/hittekaart-py/src/lib.rs b/hittekaart-py/src/lib.rs
index 2f99793..f14515a 100644
--- a/hittekaart-py/src/lib.rs
+++ b/hittekaart-py/src/lib.rs
@@ -78,6 +78,10 @@ impl Track {
                 .collect(),
         }
     }
+
+    fn __repr__(&self) -> String {
+        format!("<Track [{} coordinates]>", self.inner.len())
+    }
 }
 
 #[derive(Debug, Clone, PartialEq, Eq)]
@@ -120,6 +124,13 @@ impl Storage {
         let path = OsStr::from_bytes(path);
         Storage(StorageType::Sqlite(path.into()))
     }
+
+    fn __repr__(&self) -> String {
+        match self.0 {
+            StorageType::Folder(ref path) => format!("<Storage.Folder path='{}'>", path.display()),
+            StorageType::Sqlite(ref path) => format!("<Storage.Sqlite path='{}'>", path.display()),
+        }
+    }
 }
 
 impl Storage {