70 lines
2.1 KiB
Python
Executable File
70 lines
2.1 KiB
Python
Executable File
#!env/bin/python3
|
|
from electrum.storage import WalletStorage
|
|
from electrum.util import MyEncoder
|
|
import json
|
|
import sys
|
|
import getpass
|
|
import os
|
|
|
|
default_fees= 100
|
|
|
|
def fix_will_settings_tx_fees(json_wallet):
|
|
tx_fees = json_wallet.get('will_settings',{}).get('tx_fees',False)
|
|
have_to_update=False
|
|
if tx_fees:
|
|
json_wallet['will_settings']['baltx_fees']=tx_fees
|
|
del json_wallet['will_settings']['tx_fees']
|
|
have_to_update=True
|
|
for txid,willitem in json_wallet['will'].items():
|
|
tx_fees=willitem.get('tx_fees',False)
|
|
if tx_fees:
|
|
json_wallet['will'][txid]['baltx_fees']=tx_fees
|
|
del json_wallet['will'][txid]['tx_fees']
|
|
have_to_update=True
|
|
return have_to_update
|
|
|
|
def uninstall_bal(json_wallet):
|
|
del json_wallet['will_settings']
|
|
del json_wallet['will']
|
|
del json_wallet['heirs']
|
|
return True
|
|
|
|
def save(json_wallet,storage):
|
|
human_readable=not storage.is_encrypted()
|
|
storage.write(json.dumps(
|
|
json_wallet,
|
|
indent=4 if human_readable else None,
|
|
sort_keys=bool(human_readable),
|
|
cls=MyEncoder,
|
|
))
|
|
def read_wallet(path,password=False):
|
|
storage=WalletStorage(path)
|
|
if storage.is_encrypted():
|
|
if password==False:
|
|
password = getpass.getpass("Enter wallet password: ", stream = None)
|
|
storage.decrypt(password)
|
|
data=storage.read()
|
|
json_wallet=json.loads('['+data+']')[0]
|
|
return json_wallet
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) <3:
|
|
print("usage: ./bal_wallet_utils <command> <wallet path>")
|
|
print("available commands: uninstall, fix")
|
|
exit(1)
|
|
if not os.path.exists(sys.argv[2]):
|
|
print("Error: wallet not found")
|
|
exit(1)
|
|
command = sys.argv[1]
|
|
path = sys.argv[2]
|
|
json_wallet = read_wallet(path)
|
|
have_to_save=False
|
|
if command == 'fix':
|
|
have_to_save = fix_will_settings_tx_fees(json_wallet)
|
|
if command == 'uninstall':
|
|
have_to_save = uninstall_bal(json_wallet)
|
|
if have_to_save:
|
|
save(json_wallet,storage)
|
|
else:
|
|
print("nothing to do")
|