Add back __srt__ and __repr__ for Wallet and Keypair

pyproject.toml

@@ -77,4 +77,4 @@ dev = [
     "bittensor>=9.0.1",
     "substrate-interface==1.7.11",
     "scalecodec~=1.2.11",
-]
+]

src/keypair.rs

@@ -7,6 +7,7 @@ use sodiumoxide::crypto::secretbox;
 use sodiumoxide::crypto::secretbox::{Key, Nonce};
 use sp_core::crypto::Ss58Codec;
 use sp_core::{sr25519, ByteArray, Pair};
+use std::fmt;
 
 const PKCS8_HEADER: &[u8] = &[48, 83, 2, 1, 1, 48, 5, 6, 3, 43, 101, 112, 4, 34, 4, 32];
 const PKCS8_DIVIDER: &[u8] = &[161, 35, 3, 33, 0];
@@ -117,6 +118,17 @@ impl Keypair {
         Ok(kp)
     }
 
+    fn __str__(&self) -> Result<String, String> {
+        match self.ss58_address() {
+            Some(address) => Ok(format!("<Keypair (address={})>", address)),
+            None => Ok("<Keypair (address=None)>".to_string()),
+        }
+    }
+
+    fn __repr__(&self) -> Result<String, String> {
+        self.__str__()
+    }
+
     pub fn generate_mnemonic(n_words: usize) -> Result<String, String> {
         let mnemonic = Mnemonic::generate(n_words).map_err(|e| e.to_string())?;
         Ok(mnemonic.to_string())
@@ -421,3 +433,23 @@ impl Default for Keypair {
         }
     }
 }
+
+impl fmt::Display for Keypair {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let address = self.ss58_address();
+        match address {
+            Some(addr) => write!(f, "<Keypair (address={})>", addr),
+            None => write!(f, "<Keypair (address=None)>"),
+        }
+    }
+}
+
+impl fmt::Debug for Keypair {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let address = self.ss58_address();
+        match address {
+            Some(addr) => write!(f, "<Keypair (address={})>", addr),
+            None => write!(f, "<Keypair (address=None)>"),
+        }
+    }
+}

src/python_bindings.rs

@@ -213,6 +213,14 @@ impl PyKeypair {
         Ok(PyKeypair { inner: keypair })
     }
 
+    fn __str__(&self) -> PyResult<String> {
+        Ok(self.inner.to_string())
+    }
+
+    fn __repr__(&self) -> PyResult<String> {
+        self.__str__()
+    }
+
     #[staticmethod]
     #[pyo3(signature = (n_words=12))]
     fn generate_mnemonic(n_words: usize) -> PyResult<String> {
@@ -766,6 +774,10 @@ impl Wallet {
         Ok(self.inner.to_string())
     }
 
+    fn __repr__(&self) -> PyResult<String> {
+        self.__str__()
+    }
+
     /// Accept specific arguments from parser.
     #[classmethod]
     #[pyo3(signature = (parser, prefix = None))]