Merge pull request #1490 from wojake/patch-4

Create encode_address.py

content/_code-samples/address_encoding/py/encode_address.py

@@ -0,0 +1,67 @@
+from typing import Union, List
+from hashlib import sha256
+import hashlib
+
+R58dict = b'rpshnaf39wBUDNEGHJKLM4PQRST7VWXYZ2bcdeCg65jkm8oFqi1tuvAxyz'
+
+
+def scrub_input(input: Union[str, bytes]) -> bytes:
+    if isinstance(input, str):
+        input = input.encode('ascii')
+    return input
+
+
+def b58encode_int(
+        integer: int, default_one: bool = True, alphabet: bytes = R58dict
+) -> bytes:
+    """
+    Encode an integer using Base58
+    """
+    if not integer and default_one:
+        return alphabet[0:1]
+    string = b""
+    base = len(alphabet)
+    while integer:
+        integer, idx = divmod(integer, base)
+        string = alphabet[idx:idx + 1] + string
+    return string
+
+
+def b58encode(
+        v: Union[str, bytes], alphabet: bytes = R58dict
+) -> bytes:
+    """
+    Encode a string using Base58
+    """
+    v = scrub_input(v)
+
+    origlen = len(v)
+    v = v.lstrip(b'\0')
+    newlen = len(v)
+
+    acc = int.from_bytes(v, byteorder='big')
+
+    result = b58encode_int(acc, default_one=False, alphabet=alphabet)
+    return alphabet[0:1] * (origlen - newlen) + result
+
+
+_CLASSIC_ADDRESS_PREFIX: List[int] = [0x0]
+
+# Public Key -> AccountID
+# Ed25519 key:
+public_key = "ED9434799226374926EDA3B54B1B461B4ABF7237962EAE18528FEA67595397FA32"
+
+
+# Calculate the RIPEMD160 hash of the SHA-256 hash of the public key
+# This is the "Account ID"
+sha_hash = hashlib.sha256(bytes.fromhex(public_key)).digest()
+account_id = hashlib.new("ripemd160", sha_hash).digest()
+
+encoded_prefix = bytes(_CLASSIC_ADDRESS_PREFIX)
+payload = encoded_prefix + account_id
+v = scrub_input(payload)
+digest = sha256(sha256(v).digest()).digest()
+check = b58encode(v + digest[:4], alphabet=R58dict)
+print(check.decode("utf-8"))
+# rDTXLQ7ZKZVKz33zJbHjgVShjsBnqMBhmN (Ed25519)
+