-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
175 lines (141 loc) · 5.15 KB
/
app.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
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
import os
from uuid import uuid4
from flask import Flask, json, jsonify, request
from flask_login import LoginManager, login_user, login_required, logout_user, current_user
from blockchain import Blockchain
from transaction_manager import TransactionManager
from models.user import User
# Instantiate the Node
app = Flask(__name__)
app.secret_key = os.urandom(24)
login_manager = LoginManager()
login_manager.init_app(app)
# Generate a globally unique address for this node
node_identifier = str(uuid4()).replace('-', '')
# Instantiate the Blockchain
blockchain = Blockchain()
# Create dummy users with ids 1 to 5
users = [User(id, node_identifier) for id in range(1, 6)]
# Instantiate the transaction manager
transaction_manager = TransactionManager(users)
@app.route('/mine', methods=['GET'])
def mine():
# We run the proof of work algorithm to get the next proof...
last_block = blockchain.last_block
proof = blockchain.proof_of_work(last_block)
# We must receive a reward for finding the proof.
# The sender is "0" to signify that this node has mined a new coin.
blockchain.new_transaction(
sender="0",
recipient=node_identifier,
amount=1,
)
# Forge the new Block by adding it to the chain
previous_hash = blockchain.hash(last_block)
block = blockchain.new_block(proof, previous_hash)
response = {
'message': "New Block Forged",
'index': block['index'],
'transactions': block['transactions'],
'proof': block['proof'],
'previous_hash': block['previous_hash'],
}
return jsonify(response), 200
@app.route('/transactions/new', methods=['POST'])
def new_transaction():
values = request.get_json(force=True)
# Check that the required fields are in the POST'ed data
required = ['sender', 'recipient', 'amount', 'sender_account', 'destination_account']
if not all(k in values for k in required):
return 'Missing values', 400
# Create a new Transaction
index = blockchain.new_transaction(values['sender'], values['recipient'], values['amount'])
# Apply trasaction on user account
transaction_manager.apply_transaction(index, values['amount'], values['sender_account'], values['destination_account'])
response = {'message': "Transaction will be added to Block {0}".format(index)}
return jsonify(response), 201
@app.route('/transactions', methods=['GET'])
def list_transactions():
response = {
'transactions': transaction_manager.report_transactions()
}
return jsonify(response), 200
@app.route('/accounts', methods=['GET'])
def list_accounts():
response = {
'accounts': transaction_manager.report_accounts()
}
return jsonify(response), 200
@app.route('/chain', methods=['GET'])
@login_required
def full_chain():
response = {
'chain': blockchain.chain,
'length': len(blockchain.chain),
}
return jsonify(response), 200
@app.route('/nodes/register', methods=['POST'])
def register_nodes():
values = request.get_json(force=True)
nodes = values.get('nodes')
if nodes is None:
return "Error: Please supply a valid list of nodes", 400
for node in nodes:
blockchain.register_node(node)
response = {
'message': 'New nodes have been added',
'total_nodes': list(blockchain.nodes),
}
return jsonify(response), 201
@app.route('/nodes/resolve', methods=['GET'])
def consensus():
replaced = blockchain.resolve_conflicts()
if replaced:
response = {
'message': 'Our chain was replaced',
'new_chain': blockchain.chain
}
else:
response = {
'message': 'Our chain is authoritative',
'chain': blockchain.chain
}
return jsonify(response), 200
@app.route('/login' , methods=['POST'])
def login():
username = request.get_json(force=True)['username']
password = request.get_json(force=True)['password']
user = next(iter([user for user in users if user.name == username]), None)
if user != None and user.password == password:
print('Logged in user %s' % (user.name))
login_user(user)
return 'Success', 200
else:
return 'Error: incorrect credentials', 401
@app.route("/logout", methods=['GET'])
@login_required
def logout():
logout_user()
# handle login failed
@app.errorhandler(401)
def page_not_found(e):
return 'Error: incorrect credentials', 401
# callback to reload the user object
@login_manager.request_loader
def load_user(request):
token = request.headers.get('Authorization')
if token is None:
token = request.args.get('token')
if token != None:
username, password = token.split(":") # naive token
user = next(iter([user for user in users if user.name == username]), None)
if (user != None and user.password == password):
return user
return None
if __name__ == '__main__':
from argparse import ArgumentParser
parser = ArgumentParser()
parser.add_argument('-p', '--port', default=5000, type=int, help='port to listen on')
args = parser.parse_args()
port = args.port
app.run(debug=True, host='0.0.0.0', port=port)