orb.fi/get_wallet.py at master · orexcoder/orb.fi · 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
import os
from web3 import Web3
from config import get_default_provider

def get_main_wallet():
    with open(f'{os.path.dirname(__file__)}/wallets/wallet.txt', 'r') as file:
        _main_wallet = [row.strip() for row in file]
    return _main_wallet

def get_all_wallets(list):
    default_provider = get_default_provider()
    web3 = Web3(Web3.HTTPProvider(default_provider['provider']['rpc']))
    web3.eth.account.enable_unaudited_hdwallet_features()
    _wallets = []
    for wallet in list:
        if len(wallet) == 0:
            continue
        if len(wallet) == 64:
            cWallet = web3.eth.account.from_key(wallet)
            balance = web3.eth.get_balance(cWallet.address)
            nonce = web3.eth.get_transaction_count(cWallet.address)
            _wallets.append({'wallet': cWallet, 'balance': balance, 'nonce': nonce})
        else:
            wallet_format = wallet.split(';')
            if len(wallet_format) == 2:
                num_of_wallet = wallet_format[1]
            elif len(wallet_format) == 3:
                _indexes = wallet_format[2].split(',')
                indexes = []
                for _ind in _indexes:
                    if _ind == '':
                        continue
                    if len(_ind) > 1 and len(_ind.split('-')) == 2:
                        els = _ind.split('-')
                        _start = int(els[0])
                        _finish = int(els[1]) + 1
                        for i in range(_start, _finish):
                            indexes.append(i)
                    else:
                        indexes.append(int(_ind))
            else:
                num_of_wallet = 100
            cWallet = wallet_format[0]
            if len(wallet_format) == 3:
                num_of_wallet = indexes[len(indexes) - 1]
                _all_wallets = []
                for i in range(int(num_of_wallet)):
                    wallet_address = web3.eth.account.from_mnemonic(cWallet, account_path=f"m/44'/60'/0'/0/{i}")
                    balance = web3.eth.get_balance(wallet_address.address)
                    nonce = web3.eth.get_transaction_count(wallet_address.address)
                    _all_wallets.append({'wallet': wallet_address, 'balance': balance, 'nonce': nonce})
                _new_wallets = []
                for el in indexes:
                    _new_wallets.append(_all_wallets[el - 1])
                [_wallets.append(_new) for _new in _new_wallets]
            else:
                for i in range(int(num_of_wallet)):
                    wallet_address = web3.eth.account.from_mnemonic(cWallet, account_path=f"m/44'/60'/0'/0/{i}")
                    balance = web3.eth.get_balance(wallet_address.address)
                    nonce = web3.eth.get_transaction_count(wallet_address.address)
                    _wallets.append({'wallet': wallet_address, 'balance': balance, 'nonce': nonce})

    return _wallets