Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

encrypt/decrypt/lock/unlock commands return True

Browse source
This commit is contained in:
Lex Berezhny 2019-10-14 12:53:02 -04:00
parent 45eef03576
commit b8b24da292
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torba/torba/client/wallet.py
View file

@ -181,12 +181,14 @@ class Wallet:
for account in self.accounts: for account in self.accounts:
if account.encrypted: if account.encrypted:
account.decrypt(password) account.decrypt(password)
return True
def lock(self): def lock(self):
for account in self.accounts: for account in self.accounts:
if not account.encrypted: if not account.encrypted:
assert account.password is not None, "account was never encrypted" assert account.password is not None, "account was never encrypted"
account.encrypt(account.password) account.encrypt(account.password)
return True
@property @property
def is_encrypted(self) -> bool: def is_encrypted(self) -> bool:
@ -199,6 +201,7 @@ class Wallet:
for account in self.accounts: for account in self.accounts:
account.serialize_encrypted = False account.serialize_encrypted = False
self.save() self.save()
return True
def encrypt(self, password): def encrypt(self, password):
for account in self.accounts: for account in self.accounts:
@ -207,6 +210,7 @@ class Wallet:
account.serialize_encrypted = True account.serialize_encrypted = True
self.save() self.save()
self.unlock(password) self.unlock(password)
return True
class WalletStorage: class WalletStorage: