Add __repr__ for OTPState and OTPGenerator

2 files changed, 12 insertions, 0 deletions

diff --git a/otp2289/generator.py b/otp2289/generator.py
index 4623dc0..723ed04 100644
--- a/otp2289/generator.py
+++ b/otp2289/generator.py
@@ -333,6 +333,11 @@ class OTPGenerator:
                 'Password must be longer than 10 bytes')
         self._password = password
 
+    def __repr__(self):
+        """repr implementation"""
+        return (f'{self.__class__} at {id(self)} (seed={self._seed}, '
+                f'hash_algo={self._hash_algo})')
+
     @staticmethod
     def bit_pair_sum(bit_stream):
         """
diff --git a/otp2289/server.py b/otp2289/server.py
index 119fbff..7322727 100644
--- a/otp2289/server.py
+++ b/otp2289/server.py
@@ -85,6 +85,13 @@ class OTPState:
             self._current_digest = self.validate_hex(ot_hex)
         self._new_digest_hex = None  # set upon a successful validation
 
+    def __repr__(self):
+        """repr implementation"""
+        return (f'{self.__class__} at {id(self)} '
+                f'(ot_hex={self._current_digest}, current_step={self._step}, '
+                f'seed={self._seed}, '
+                f'hash_algo={self._hash_algo})')
+
     @property
     def challenge_string(self):
         """challenge_string-property"""