lessons period 2

period_2/02-sockets/903/Wallet_t.py

@@ -0,0 +1,69 @@
+from SocketUtil import *
+from Transaction import *
+from Signature import *
+
+M_SERVER_IP = 'localhost'
+M_SERVER_PORT = 5005
+
+W_SERVER_IP = 'localhost'
+W_SERVER_PORT = 5006
+
+alex_prv, alex_pbc = generate_keys()
+mike_prv, mike_pbc = generate_keys()
+rose_prv, rose_pbc = generate_keys()
+mara_prv, mara_pbc = generate_keys()
+
+Tx1 = Tx()
+Tx1.add_input(alex_pbc, 4.0)
+Tx1.add_input(mike_pbc, 1.0)
+Tx1.add_output(rose_pbc, 4.8)
+Tx1.sign(alex_prv)
+Tx1.sign(mike_prv)
+
+Tx2 = Tx()
+Tx2.add_input(rose_pbc, 4.0)
+Tx2.add_output(mike_pbc, 4.0)
+Tx2.add_reqd(alex_pbc)
+Tx2.sign(rose_prv)
+Tx2.sign(alex_prv)
+
+print(Tx1.is_valid())
+print(Tx2.is_valid())
+
+try:
+    sendObj(M_SERVER_IP, Tx1, M_SERVER_PORT)
+    print("Sent Tx1")
+    sendObj(M_SERVER_IP, Tx2, M_SERVER_PORT)
+    print("Sent Tx2")
+
+except:
+    print("Error! Connection unsuccessful.")
+
+wallet_server_socket = newServerSocket(W_SERVER_IP, W_SERVER_PORT)
+
+for i in range(10):
+    newBlock = recvObj(wallet_server_socket)
+    if newBlock:
+        break
+wallet_server_socket.close()
+
+if newBlock.is_valid():
+    print("Success! Block is valid.")
+
+if newBlock.good_nonce():
+    print("Success! Nonce is valid.")
+
+tx1_found = tx2_found = False
+for tx in newBlock.data:
+    try:
+        if tx.inputs[0][0] == alex_pbc and tx.inputs[0][1] == 4.0:
+            print("Tx1 is present.")
+            tx1_found = True
+    except:
+        pass
+    try:
+        if tx.inputs[0][0] == rose_pbc and tx.inputs[0][1] == 4.0:
+            print("Tx2 is present.")
+            tx2_found = True
+    except:
+        pass