dendisuhubdy · December 13, 2021 07:51 · Dec 13, 2021 · Jan 27, 2018 · Jan 27, 2018
diff --git a/eth_wallets.py b/eth_wallets.py
@@ -12,10 +12,22 @@ def main(count):
 
 
 def generate_pair():
+    priv = generate_private_key()
+    pubc = generate_public_key(priv)
+    address = generate_public_address(pub)
+    return priv.to_string().hex(), address
+
+def generate_private_key():
     priv = SigningKey.generate(curve=SECP256k1)
+    return priv
+
+def generate_public_key(priv):
     pub = priv.get_verifying_key().to_string()
+    return pub
+
+def generate_public_address(pub):
     address = keccak_256(pub).hexdigest()[24:]
-    return priv.to_string().hex(), address
+    return address
 
 
 if __name__ == '__main__':

diff --git a/eth_wallets.py b/eth_wallets.py
@@ -8,7 +8,7 @@
 def main(count):
     for i in range(count):
         priv, addr = generate_pair()
-        print(priv, addr)
+        print(priv, '0x' + addr)
 
 
 def generate_pair():

diff --git a/eth_wallets.py b/eth_wallets.py
@@ -0,0 +1,22 @@
+from ecdsa import SigningKey, SECP256k1
+from sha3 import keccak_256
+import click
+
+
+@click.command()
+@click.argument('count', type=click.types.IntRange(1, 1000), default=1)
+def main(count):
+    for i in range(count):
+        priv, addr = generate_pair()
+        print(priv, addr)
+
+
+def generate_pair():
+    priv = SigningKey.generate(curve=SECP256k1)
+    pub = priv.get_verifying_key().to_string()
+    address = keccak_256(pub).hexdigest()[24:]
+    return priv.to_string().hex(), address
+
+
+if __name__ == '__main__':
+    main()