-
Notifications
You must be signed in to change notification settings - Fork 51
/
key.py
71 lines (52 loc) · 2.07 KB
/
key.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import hashlib
import random
import time
import base58
from .util import doublehash256
class Key:
def __init__(self, seed=None):
self.hash = ''
self.hashdigest = ''
self.hex = ''
self.wif = ''
self.wif_c = ''
self.wif_testnet = ''
self.wif_c_testnet = ''
def generate(self, seed=None):
self._generate_raw(seed)
self._generate_hex()
self._generate_wif()
self._generate_wif_testnet()
return {'hex': self.hex, 'wif': self.wif, 'wifc': self.wif_c, 'testnet': {'hex': self.hex, 'wif': self.wif_testnet, 'wifc': self.wif_c_testnet}}
def _seed(self):
t = int(time.time())
return str(random.getrandbits(3000) - t)
def _generate_raw(self, seed=None):
if (seed == None): seed = self._seed()
self.hash = hashlib.sha256(seed.encode())
self.hashdigest = self.hash.digest()
def _generate_hex(self):
self.hex = self.hash.hexdigest()
def _generate_wif(self):
prefix = b'\x80'
suffix = b'\x01'
d = prefix + self.hashdigest
checksum = doublehash256(d).digest()[:4]
checksum_c = doublehash256(d + suffix).digest()[:4]
self.wif = base58.b58encode(d + checksum).decode('utf-8')
self.wif_c = base58.b58encode(d + suffix + checksum_c).decode('utf-8')
def _generate_wif_testnet(self):
prefix = b'\xEF'
suffix = b'\x01'
d = prefix + self.hashdigest
checksum = doublehash256(d).digest()[:4]
checksum_c = doublehash256(d + suffix).digest()[:4]
self.wif_testnet = base58.b58encode(d + checksum).decode('utf-8')
self.wif_c_testnet = base58.b58encode(d + suffix + checksum_c).decode('utf-8')
def __str__(self):
return """Private Key HEX: %s\n
\rPrivate Key WIF: %s
\rPrivate Key WIF compressed: %s
\rPrivate Key WIF (TESTNET): %s
\rPrivate Key WIF compressed (TESTNET): %s
""" % (self.hex, self.wif, self.wif_c, self.wif_testnet, self.wif_c_testnet)