87 lines
7.1 KiB
Python
Executable File
87 lines
7.1 KiB
Python
Executable File
from web3 import Web3
|
|
import time
|
|
from web3.exceptions import TimeExhausted
|
|
|
|
# RPC = 'https://rpc-amoy.polygon.technology/'
|
|
# CONTRACT_ADDRESS = '0xbE91e2294D12fa78Ce64657fD9Ee137cE3e99881'
|
|
# CHAIN_ID = 80002
|
|
# ABI = '[ { "inputs": [], "stateMutability": "nonpayable", "type": "constructor" }, { "anonymous": false, "inputs": [ { "indexed": false, "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "indexed": false, "internalType": "string", "name": "project", "type": "string" }, { "indexed": false, "internalType": "string", "name": "data", "type": "string" } ], "name": "conversion", "type": "event" }, { "inputs": [ { "internalType": "address", "name": "pubKey", "type": "address" }, { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" }, { "internalType": "string", "name": "_data", "type": "string" } ], "name": "UploadScriptAuditData", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "deleteScriptAudit", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "userId", "type": "uint256" } ], "name": "getProjectId", "outputs": [ { "internalType": "string[]", "name": "", "type": "string[]" } ], "stateMutability": "view", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "getScriptAudit", "outputs": [ { "components": [ { "internalType": "address", "name": "owner", "type": "address" }, { "internalType": "string", "name": "mydata", "type": "string" }, { "internalType": "bool", "name": "Write", "type": "bool" } ], "internalType": "struct ScriptAuditDataBase.privateData", "name": "", "type": "tuple" } ], "stateMutability": "view", "type": "function" } ]'
|
|
|
|
# mainnet
|
|
RPC = 'https://polygon-mainnet.infura.io/v3/017957497a0d4e9c80750c18a431ac1e'
|
|
CONTRACT_ADDRESS = '0x3a5d033CdF38aC4a9fe116CE88EBA2E93260044c'
|
|
CHAIN_ID = 137
|
|
ABI = '[ { "inputs": [], "stateMutability": "nonpayable", "type": "constructor" }, { "anonymous": false, "inputs": [ { "indexed": false, "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "indexed": false, "internalType": "string", "name": "project", "type": "string" }, { "indexed": false, "internalType": "string", "name": "data", "type": "string" } ], "name": "conversion", "type": "event" }, { "inputs": [ { "internalType": "address", "name": "pubKey", "type": "address" }, { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" }, { "internalType": "string", "name": "_data", "type": "string" } ], "name": "UploadScriptAuditData", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "deleteScriptAudit", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "userId", "type": "uint256" } ], "name": "getProjectId", "outputs": [ { "internalType": "string[]", "name": "", "type": "string[]" } ], "stateMutability": "view", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "getScriptAudit", "outputs": [ { "components": [ { "internalType": "address", "name": "owner", "type": "address" }, { "internalType": "string", "name": "mydata", "type": "string" }, { "internalType": "bool", "name": "Write", "type": "bool" } ], "internalType": "struct ScriptAuditDataBase.privateData", "name": "", "type": "tuple" } ], "stateMutability": "view", "type": "function" } ]'
|
|
|
|
web3 = Web3(Web3.HTTPProvider(RPC))
|
|
CAddress =CONTRACT_ADDRESS
|
|
abi = ABI
|
|
contractInst = web3.eth.contract(address=CAddress, abi=abi)
|
|
|
|
def UploadScriptAuditData(privatekey,pubkey,user_id,project,data):
|
|
try:
|
|
acc1 = web3.eth.account.from_key(privatekey).address
|
|
def send_transaction():
|
|
nonce = web3.eth.get_transaction_count(acc1)
|
|
uploadData = contractInst.functions.UploadScriptAuditData(pubkey,user_id,project,data).build_transaction({
|
|
'gasPrice': web3.eth.gas_price, 'chainId': CHAIN_ID, 'from': acc1, 'nonce': nonce})
|
|
signed_transaction = web3.eth.account.sign_transaction(uploadData, private_key=privatekey)
|
|
transaction_hash = web3.eth.send_raw_transaction(signed_transaction.rawTransaction)
|
|
print("transaction hash",transaction_hash)
|
|
return transaction_hash
|
|
max_retries = 3
|
|
retries = 0
|
|
transaction_hash = send_transaction()
|
|
|
|
while retries < max_retries:
|
|
try:
|
|
transaction_receipt = web3.eth.wait_for_transaction_receipt(transaction_hash)
|
|
print("Transaction confirmed:")
|
|
break
|
|
except TimeExhausted:
|
|
retries += 1
|
|
print(f"Transaction not confirmed in {120 * retries} seconds, retrying... ({retries}/{max_retries})")
|
|
transaction_hash = send_transaction()
|
|
else:
|
|
print("Failed to confirm the transaction after multiple attempts.")
|
|
return TimeoutError
|
|
trancation_fee = transaction_receipt.effectiveGasPrice*transaction_receipt.gasUsed
|
|
tx_id = transaction_hash.hex()
|
|
return tx_id,str(trancation_fee)
|
|
except Exception as e:
|
|
# print({"error":str(e)})
|
|
return e
|
|
|
|
def getScriptAudit(private_key,user_id,project):
|
|
ACCOUNT = web3.eth.account.from_key(private_key).address
|
|
try:
|
|
getData = contractInst.functions.getScriptAudit(user_id,project).call({'from':ACCOUNT})
|
|
print(getData)
|
|
return True ,getData
|
|
except Exception as e:
|
|
print("Error calling function:", e)
|
|
return False,e
|
|
|
|
def getUserprojectIds(user_id):
|
|
try:
|
|
getProjectId = contractInst.functions.getProjectId(user_id).call()
|
|
return getProjectId
|
|
except Exception as e:
|
|
return e
|
|
|
|
def deleteScriptAudit(privatekey,user_id,project):
|
|
try:
|
|
acc1 = web3.eth.account.from_key(privatekey).address
|
|
nonce = web3.eth.getTransactionCount(acc1)
|
|
deleteData = contractInst.functions.deleteScriptAudit(user_id,project).buildTransaction({
|
|
'gasPrice': web3.eth.gas_price, 'chainId': CHAIN_ID, 'from': acc1, 'nonce': nonce})
|
|
signed_transaction = web3.eth.account.sign_transaction(deleteData, private_key=privatekey)
|
|
transaction_hash = web3.eth.send_raw_transaction(signed_transaction.rawTransaction)
|
|
transaction_receipt = web3.eth.wait_for_transaction_receipt(transaction_hash)
|
|
tx_id = transaction_hash.hex()
|
|
return tx_id
|
|
except Exception as e:
|
|
print({"PPT_conversion_delete_error":str(e)})
|
|
|
|
|