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
Python
61 lines
2.2 KiB
Python
2 years 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
|
||
|
|
|
||
|
|
|
||
|
|
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')
|
||
|
|
exit()
|
||
|
|
|
||
|
|
# 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:
|
||
|
|
f.write(f'{phrase}\n{hashed}')
|
||
|
|
|
||
|
|
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')
|