refactor: class based wif module (#163)

Author: alexbarnsley

crypto/identity/wif.py

@@ -7,22 +7,24 @@
 
 from crypto.configuration.network import Network
 
-def wif_from_passphrase(passphrase: str, network_wif: Optional[str] = None):
-    """Get wif from passphrase
+class WIF:
+    @staticmethod
+    def from_passphrase(passphrase: str, network_wif: Optional[str] = None):
+        """Get wif from passphrase
 
-    Args:
-        passphrase (str):
-        network_wif (str, optional):
+        Args:
+            passphrase (str):
+            network_wif (str, optional):
 
-    Returns:
-        string: wif
-    """
-    if not network_wif:
-        network = Network.get_network()
+        Returns:
+            string: wif
+        """
+        if not network_wif:
+            network = Network.get_network()
 
-        network_wif = network.wif()
+            network_wif = network.wif()
 
-    private_key = hashlib.sha256(passphrase.encode())
-    seed = write_bit8(int(network_wif, 16)) + private_key.digest() + write_bit8(0x01)
+        private_key = hashlib.sha256(passphrase.encode())
+        seed = write_bit8(int(network_wif, 16)) + private_key.digest() + write_bit8(0x01)
 
-    return b58encode_check(seed).decode()
+        return b58encode_check(seed).decode()

tests/identity/test_wif.py

@@ -1,10 +1,10 @@
 from crypto.configuration.network import Network
-from crypto.identity.wif import wif_from_passphrase
+from crypto.identity.wif import WIF
 from crypto.networks.testnet import Testnet
 
 
 def test_wif_from_passphrase(identity):
     Network.set_network(Testnet())
 
-    result = wif_from_passphrase(identity['passphrase'])
+    result = WIF.from_passphrase(identity['passphrase'])
     assert result == identity['data']['wif']