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

Extract restaking confirmation actions with more verbose confirmation messages.

pull/1339/head
Kieran Prasch 2019-09-19 11:21:04 -07:00
parent 6761fde377
commit 3d51918125
No known key found for this signature in database
GPG Key ID: 199AB839D4125A62
2 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
nucypher/cli/actions.py
View File

@ -369,3 +369,21 @@ def confirm_deployment(emitter, deployer_interface) -> bool:
raise click.Abort()
return True
def confirm_enable_restaking_lock(emitter, staking_address: str, release_period: int) -> bool:
restaking_lock_agreement = f"""
By enabling the re-staking lock for {staking_address}, you are committing to automatically
re-stake all rewards until period a future period. You will not be able to disable re-staking until {release_period}.
"""
emitter.message(restaking_lock_agreement)
click.confirm(f"Confirm enable re-staking lock for staker {staking_address} until {release_period}?", abort=True)
return True
def confirm_enable_restaking(emitter, staking_address: str) -> bool:
restaking_lock_agreement = f"By enabling the re-staking for {staking_address}, " \
f"All staking rewards will be automatically added to your existing stake."
emitter.message(restaking_lock_agreement)
click.confirm(f"Confirm enable automatic re-staking for staker {staking_address}?", abort=True)
return True

15
nucypher/cli/characters/stake.py
View File

@ -19,12 +19,19 @@ along with nucypher. If not, see <https://www.gnu.org/licenses/>.
import click
from web3 import Web3
from nucypher.characters.lawful import StakeHolder
from nucypher.blockchain.eth.interfaces import BlockchainInterface, BlockchainInterfaceFactory
from nucypher.blockchain.eth.token import NU
from nucypher.blockchain.eth.utils import datetime_at_period
from nucypher.characters.lawful import StakeHolder
from nucypher.cli import painting, actions
from nucypher.cli.actions import confirm_staged_stake, get_client_password, select_stake, select_client_account
from nucypher.cli.actions import (
confirm_staged_stake,
get_client_password,
select_stake,
select_client_account,
confirm_enable_restaking_lock,
confirm_enable_restaking
)
from nucypher.cli.config import nucypher_click_config
from nucypher.cli.painting import paint_receipt_summary
from nucypher.cli.types import (
@ -308,13 +315,13 @@ def stake(click_config,
# Inner Exclusive Switch
if lock_until:
if not force:
click.confirm(f"Confirm enable re-staking lock for staker {staking_address} until {lock_until}?", abort=True)
confirm_enable_restaking_lock(emitter, staking_address=staking_address, release_period=lock_until)
receipt = STAKEHOLDER.enable_restaking_lock(release_period=lock_until)
emitter.echo(f'Successfully enabled re-staking lock for {staking_address} until {lock_until}',
color='green', verbosity=1)
elif enable:
if not force:
click.confirm(f"Confirm enable re-staking for staker {staking_address}?", abort=True)
confirm_enable_restaking(emitter, staking_address=staking_address)
receipt = STAKEHOLDER.enable_restaking()
emitter.echo(f'Successfully enabled re-staking for {staking_address}', color='green', verbosity=1)
else: