PyBlockchain/Wallet.py at master · abdelrahman146/PyBlockchain · GitHub

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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
"""
project: PyBlockchain
author: Abdel Rahman Hussein
email: abdelrahman146@outlook.com
github: http://github.com/abdelrahman146

This class defines the wallet
Wallet is the interface of the blockchain user. where the user can create new smart contracts
"""
import json

from Blockchain.Blockchain import get_blockchain
from Mempool import get_mempool
from SmartContracts.IssueContract import IssueContract
from SmartContracts.RedeemContract import RedeemContract
from SmartContracts.SendContract import SendContract
from helpers import crypto, hashing
import os

class Wallet:
    def __init__(self, name):
        self.name = name
        self.private_key_file_path = 'storage/private_key_%s.json' % name
        if os.path.isfile(self.private_key_file_path):
            self.private_key, self.password = self.load_from_file()
        else:
            self.password = crypto.generate_password()
            self.private_key = crypto.generate_private_pem_string(password=self.password)
            self.save_to_file()
        self.public_key = crypto.generate_public_pem_string(private_pem_string=self.private_key, password=self.password)
        self.blockchain = get_blockchain()
        self.mempool = get_mempool()

    def get_balance(self):
        return self.blockchain.get_pk_balance(self.public_key)

    def send(self, amount, contract_counterparty_pk, msg):
        dict = {
            'contract_name': 'SEND',
            'contract_issuer_pk': hashing.hash(self.public_key),
            'contract_counterparty_pk': hashing.hash(contract_counterparty_pk),
            'amount': amount,
            'msg': msg
        }
        dict_json = json.dumps(dict)
        dict.update({'contract_hash': hashing.hash(dict_json)})
        contract_json = json.dumps(dict)
        signature = crypto.sign(
            private_pem_string=self.private_key,
            password=self.password,
            message=hashing.hash(contract_json)
        )
        contract = SendContract(
            contract_issuer_pk=self.public_key,
            contract_counterparty_pk=contract_counterparty_pk,
            amount=amount,
            msg=msg,
            signature=signature
        )
        self.mempool.insert_contract(contract=contract)

    def issue(self, amount, msg=''):
        dict = {
            'contract_name': 'ISSUE',
            'contract_issuer_pk': hashing.hash(self.public_key),
            'amount': amount,
            'msg': msg
        }
        dict_json = json.dumps(dict)
        dict.update({'contract_hash': hashing.hash(dict_json)})
        contract_json = json.dumps(dict)
        signature = crypto.sign(
            private_pem_string=self.private_key,
            password=self.password,
            message=hashing.hash(contract_json)
        )
        contract = IssueContract(
            contract_issuer_pk=self.public_key,
            amount=amount,
            msg=msg,
            signature= signature
        )
        self.mempool.insert_contract(contract=contract)

    def redeem(self, amount, msg=''):
        dict = {
            'contract_name': 'REDEEM',
            'contract_issuer_pk': hashing.hash(self.public_key),
            'amount': amount,
            'msg': msg
        }
        dict_json = json.dumps(dict)
        dict.update({'contract_hash': hashing.hash(dict_json)})
        contract_json = json.dumps(dict)
        signature = crypto.sign(
            private_pem_string=self.private_key,
            password=self.password,
            message=hashing.hash(contract_json)
        )
        contract = RedeemContract(
            contract_issuer_pk=self.public_key,
            amount=amount,
            msg=msg,
            signature=signature
        )
        self.mempool.insert_contract(contract=contract)

    def save_to_file(self):
        data = {
            "private_key": self.private_key,
            "password": self.password
        }
        with open(self.private_key_file_path, "w") as output:
            output.write(json.dumps(data))

    def load_from_file(self):
        with open(self.private_key_file_path, "r") as input_file:
            data = json.loads(input_file.read())
            return data["private_key"], data["password"]