Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

In Allocator, using increaseAllowance() is more flexible than approve()

pull/1946/head
David Núñez 2020-05-04 20:07:04 +02:00
parent b7bca790e6
commit 45de39e394
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
nucypher/blockchain/eth/actors.py
View File

@ -562,17 +562,19 @@ class Allocator:
self._add_substake(staker, amount, lock_periods)
total_to_allocate += amount
token_agent = ContractAgency.get_agent(NucypherTokenAgent,
registry=registry) # type: NucypherTokenAgent
token_agent = ContractAgency.get_agent(NucypherTokenAgent, registry=registry) # type: NucypherTokenAgent
balance = token_agent.get_balance(deployer_address)
if balance < total_to_allocate:
raise ValueError(f"Not enough tokens to allocate. We need at least {NU.from_nunits(total_to_allocate)}.")
self.log.debug(f"New allocator for batch deposits. Allocating a total of {NU.from_nunits(total_to_allocate)}")
approve_function = token_agent.contract.functions.approve(self.staking_agent.contract_address, total_to_allocate)
_approve_receipt = self.staking_agent.blockchain.send_transaction(contract_function=approve_function,
sender_address=deployer_address)
allowance = token_agent.get_allowance(owner=deployer_address, spender=self.staking_agent.contract_address)
if allowance < total_to_allocate:
self.log.debug(f"Allocating a total of {NU.from_nunits(total_to_allocate)}")
allowance_function = token_agent.contract.functions.increaseAllowance(self.staking_agent.contract_address,
total_to_allocate - allowance)
_allowance_receipt = self.staking_agent.blockchain.send_transaction(contract_function=allowance_function,
sender_address=deployer_address)
def _add_substake(self, staker, amount, lock_periods):
try: