You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

61 lines
2.2 KiB

1 year ago
#!/usr/bin/env python3
import json
from os import getenv, path
from secrets import token_urlsafe
from datetime import datetime
from hashlib import sha256
from random import choices
from web3 import Web3
from dotenv import load_dotenv
def get_eth_contract(_ca, _rp):
compiled_contract_path = path.abspath(_rp)
with open(compiled_contract_path) as file:
contract_json = json.load(file)
contract_abi = contract_json['abi']
return w3.eth.contract(address=w3.toChecksumAddress(_ca), abi=contract_abi)
def sendit(w3, t, nonce):
t['from'] = w3.eth.defaultAccount
t['nonce'] = nonce
s = w3.eth.account.sign_transaction(t, private_key=getenv(f'{net}_KEY'))
return w3.eth.send_raw_transaction(s.rawTransaction)
if __name__ == '__main__':
net = getenv('NET').upper()
if not getenv(f'{net}_CONTRACT') or not getenv(f'{net}_KEY') or not getenv(f'{net}_RPC'):
print('invalid env vars set')
# web3 setup
WEB3_PROVIDER_URI = getenv(f'{net}_RPC')
w3 = Web3(Web3.HTTPProvider(WEB3_PROVIDER_URI))
w3.eth.defaultAccount = w3.eth.account.from_key(getenv(f'{net}_KEY')).address
contract = get_eth_contract(getenv(f'{net}_CONTRACT'), './out/Lottery.sol/Lottery.json')
nonce = w3.eth.get_transaction_count(w3.eth.defaultAccount)
# Calculate gas
if getenv('HASH'):
tokens = choices(range(0, 5000), k=10)
phrase = f'{datetime.utcnow()},{token_urlsafe(12).replace(",", "")},{" ".join([str(i) for i in tokens])}'
hashed = sha256(phrase.encode('utf-8')).hexdigest()
r = sendit(w3, contract.functions.setWinningPhrase(hashed).build_transaction(), nonce)
print(f'sent tx {r.hex()} with nonce {nonce} to update the winning phrase')
with open('output.txt', 'w') as f:
if getenv('TOKENS'):
msg = ''
with open('output.txt', 'r') as f:
msg = f.readlines()[0]
_tokens = msg.split(',')[2]
tokens = [int(i) for i in _tokens.split()]
r = sendit(w3, contract.functions.setWinningTokens(tokens).build_transaction(), nonce)
print(f'sent tx {r.hex()} with nonce {nonce} to update the winning tokens')