ASBT/ASBT.py at master · adilaiman/ASBT · 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
120
121
122
123
124
125
126
127
128
# Adi's Scuffed Bitcoin Tool (ASBT)
# Author: Adi Laiman
# Date: 01 March 2019 (01-03-2019)
# Contact: github.com/adilaiman

# import bitcoin, qrcode modules duh? don't reinvent the wheel
# pip install bitcoin && pip install qrcode[pil]
from bitcoin import *
import qrcode as qr
from colorama import * # module to allow color output to console

# open file and write out data to txt file and qrcode
def writeOut(data, filename, option=0):

    #  open file to write data
    with open(filename + ".txt", 'w') as f:
        f.write(data)

    # output qr code if option is 1 else don't
    if option == 1:
        img = qr.make(data)
        img.save(filename + ".png")

# generates a private key (this is what you keep secret) and writes to txt file
def genPrivateKey():
    private_key = random_key()
    writeOut(private_key, "privateKey", 1)
    return private_key

# generates a public key using the already generated private key and writes to txt file
def genPublicKey(privateKey):
    public_key = privtopub(privateKey)
    writeOut(public_key, "publicKey")
    return public_key

# generates a wallet address, this is the one you share to receive bitcoins, then writes to txt file and generates a qrcode png
def genWalletAddress(publicKey):
    walletAddress = pubtoaddr(publicKey)
    writeOut(walletAddress, "walletAddress", 1)
    return walletAddress

def parseTransaction(data):
    for k,v in data.items():
        # value into string to output
        v_out = str(v)
        #format output
        if k == "value":
            print("Amount: --> " + v_out + " BTC")
        elif k == "address":
            print("Receive Address --> " + v_out)
        elif k == "output":
            print("Block Hash --> " + v_out)
        elif k == "block_height":
            print("Included In Blocks --> " + v_out)

# gets data from public blockchain and prints latest 10 transactions
def viewTransactions(walletAddress):

    # pull data from public block chain, stored as a list of dictionaries
    transactions = history(walletAddress)

    # Title
    print(Fore.GREEN + "\n---Transactional History---")

    # change satoshis into BTC and remove trailing string from block hash
    for transaction in transactions:
        transaction["value"] = transaction["value"]/(10**8)
        transaction["output"] = transaction["output"][:-2]


    if len(transactions) < 10:
        for counter, transaction in enumerate(transactions):
            print("\n---", counter+1, "---") # Title displaying the number of the transaction
            parseTransaction(transaction)

        print(Style.RESET_ALL)
    else:
        for i in range(10):
            print("\n---", i+1, "---") # Title displaying the number of the transaction
            parseTransaction(transactions[i])
        print(Style.RESET_ALL)

# get wallet balance
def getWalletBalance(walletAddress):

    # pull data from public block chain, stored as a list of dictionaries
    transactions = history(walletAddress)

    # change satoshis into BTC and remove trailing string from block hash
    for transaction in transactions:
        transaction["value"] = transaction["value"]/(10**8)

    balance = 0
    # if receive address is not equal to our wallet address, minus from total else add
    for transaction in transactions:
        if transaction["address"] == walletAddress:
            balance += transaction["value"]
        else:
            balance -= transaction["value"]

    return balance

# prompt to ask user what to do.
if __name__ == "__main__":
    init() # filter ANSI escape characters on windows
    loop = True
    while loop == True:
        print("<---Welcome to Adi's Scuffed Bitcoin Tool (ASBT)!--->")
        choice = input("(1) Generate Private Key. \n(2) Generate Public Key \n(3) Generate Wallet Address \n(4) View Transactions \n(5) Get Balance \n(6) Quit \nEnter your option: ")
        if choice == "1":
            print("\n" + Fore.GREEN + genPrivateKey() + Style.RESET_ALL + "\n")
        elif choice == "2":
            usersPrivateKey = input("Please enter your private key: ")
            print("\n" + Fore.GREEN + genPublicKey(usersPrivateKey) + Style.RESET_ALL + "\n")
        elif choice == "3":
            usersPublicKey = input("Please enter your public key: ")
            print("\n" + Fore.GREEN + genWalletAddress(usersPublicKey) + Style.RESET_ALL + "\n")
        elif choice == "4":
            usersWallet = input("Please enter your wallet address: ")
            viewTransactions(usersWallet)
        elif choice == "5":
            usersWallet = input("Please enter your wallet address: ")
            print("\n" + Fore.GREEN, getWalletBalance(usersWallet), Style.RESET_ALL + "\n")
        elif choice == "6":
            loop = False
            deinit()
        else:
            print("\n" + Fore.RED + "Invalid option, please choice an option between [1-6]" + Style.RESET_ALL + "\n")