refactor save and encryption check

This commit is contained in:
Lex Berezhny 2019-10-18 13:38:51 -04:00
parent d50ebcf9eb
commit 583833017b

View file

@ -141,16 +141,15 @@ class Wallet:
}
def save(self):
if not self.is_locked and self.preferences.get(ENCRYPT_ON_DISK, False):
if self.preferences.get(ENCRYPT_ON_DISK, False):
if self.encryption_password:
self.storage.write(self.to_dict(encrypt_password=self.encryption_password))
return
else:
return self.storage.write(self.to_dict(encrypt_password=self.encryption_password))
elif not self.is_locked:
log.warning(
"Disk encryption requested but no password available for encryption. "
"Saving wallet in an unencrypted state."
)
self.storage.write(self.to_dict())
return self.storage.write(self.to_dict())
@property
def hash(self) -> bytes: