pyncoin/main.py at master · mrtj/pyncoin · 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
# pyncoin/main.py

''' Implements the CLI of the blockchain node '''

import argparse
import sys

from twisted.internet import reactor
from twisted.web.server import Site
from twisted.web.wsgi import WSGIResource

from webserver import app as web_app
from p2p import Application as P2PApplication
from blockchain import Blockchain
from wallet import Wallet
from transaction_pool import TransactionPool
from utils import bytes_to_hex

from twisted.internet.defer import setDebugging
from twisted.logger import globalLogPublisher, textFileLogObserver

log_output = textFileLogObserver(sys.stdout)
globalLogPublisher.addObserver(log_output)

if __name__ == '__main__':

    setDebugging(True)

    parser = argparse.ArgumentParser()
    parser.add_argument('web_port', help='web server port', default=5000, type=int)
    parser.add_argument('p2p_port', help='p2p server port', default=6000, type=int)
    parser.add_argument('-k', '--key_location',
                        help='location of wallet private key (defaults to "wallet/pk.pem")',
                        default='wallet/pk.pem', type=str)
    args = parser.parse_args()

    tx_pool = TransactionPool()
    blockchain = Blockchain(tx_pool)
    wallet = Wallet(args.key_location)
    p2p_application = P2PApplication(blockchain)
    blockchain.p2p_application = p2p_application
    web_app.blockchain = blockchain
    web_app.p2p_application = p2p_application
    web_app.wallet = wallet

    print('My pubblic address is: {}'.format(bytes_to_hex(wallet.get_public_key())))

    server_url = 'ws://127.0.0.1:{}'.format(args.p2p_port)
    print('Starting p2p server at {}'.format(server_url))
    p2p_application.start_server(server_url)

    # pylint: disable=maybe-no-member
    resource = WSGIResource(reactor, reactor.getThreadPool(), web_app)
    site = Site(resource)
    print('Starting web server at http://127.0.0.1:{}'.format(args.web_port))
    reactor.listenTCP(args.web_port, site)
    reactor.run()