123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148 |
- #!/usr/bin/env python
- #
- # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
- # Copyright (C) 2013-2014 by philemon <mmgen-py@yandex.com>
- #
- # This program is free software: you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation, either version 3 of the License, or
- # (at your option) any later version.
- #
- # This program is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with this program. If not, see <http://www.gnu.org/licenses/>.
- """
- mmgen-addrimport: Import addresses into a bitcoind watching wallet.
- """
- import sys
- from mmgen.Opts import *
- from mmgen.license import *
- from mmgen.util import *
- from mmgen.tx import connect_to_bitcoind,parse_addrs_file
- help_data = {
- 'prog_name': sys.argv[0].split("/")[-1],
- 'desc': """Import addresses (both mmgen and non-mmgen) into a bitcoind
- watching wallet""",
- 'usage':"[opts] [mmgen address file]",
- 'options': """
- -h, --help Print this help message
- -l, --addrlist f Import the non-mmgen Bitcoin addresses listed in file 'f'
- -r, --rescan Rescan the blockchain. Required if address to import is
- on the blockchain and has a balance. Rescanning is slow.
- -q, --quiet Suppress warnings
- """
- }
- short_opts = "hl:qr"
- long_opts = "help", "addrlist=", "quiet", "rescan"
- opts,cmd_args = process_opts(sys.argv,help_data,"".join(short_opts),long_opts)
- if 'quiet' in opts: g.quiet = True
- if len(cmd_args) != 1 and not 'addrlist' in opts:
- msg("You must specify an mmgen address list (and/or non-mmgen addresses with the '--addrlist' option)")
- sys.exit(1)
- check_opts(opts,long_opts)
- if cmd_args:
- check_infile(cmd_args[0])
- seed_id,addr_data = parse_addrs_file(cmd_args[0])
- else:
- seed_id,addr_data = "",[]
- if 'addrlist' in opts:
- l = get_lines_from_file(opts['addrlist'],"non-mmgen addresses",
- remove_comments=True)
- addr_data += [(None,i) for i in l]
- from mmgen.bitcoin import verify_addr
- qmsg_r("Validating addresses...")
- for i in addr_data:
- if not verify_addr(i[1],verbose=True):
- msg("%s: invalid address" % i)
- sys.exit(2)
- qmsg("OK")
- import mmgen.config as g
- g.http_timeout = 3600
- c = connect_to_bitcoind()
- m = """
- WARNING: You've chosen the '--rescan' option. Rescanning the block chain is
- necessary only if an address you're importing is already on the block chain
- and has a balance. Note that the rescanning process is very slow (>30 min.
- for each imported address on a low-powered computer).
- """.strip() if "rescan" in opts else """
- WARNING: If any of the addresses you're importing is already on the block chain
- and has a balance, you must exit the program now and rerun it using the
- '--rescan' option. Otherwise you may ignore this message and continue.
- """.strip()
- confirm_or_exit(m, "continue", expect="YES")
- err_flag = False
- def import_address(addr,label,rescan):
- try:
- c.importaddress(addr,label,rescan)
- except:
- global err_flag
- err_flag = True
- w1 = len(str(len(addr_data))) * 2 + 2
- w2 = len(str(max([i[0] for i in addr_data if i[0]]))) + 12
- if "rescan" in opts:
- import threading
- import time
- msg_fmt = "\r%s %-" + str(w1) + "s %-34s %-" + str(w2) + "s"
- else:
- msg_fmt = "\r%-" + str(w1) + "s %-34s %-" + str(w2) + "s"
- msg("Importing addresses")
- for n,i in enumerate(addr_data):
- if i[0]:
- comment = " " + i[2] if len(i) == 3 else ""
- label = "%s:%s%s" % (seed_id,i[0],comment)
- else: label = "non-mmgen"
- if "rescan" in opts:
- t = threading.Thread(target=import_address, args=(i[1],label,True))
- t.daemon = True
- t.start()
- start = int(time.time())
- while True:
- if t.is_alive():
- elapsed = int(time.time() - start)
- msg_r(msg_fmt % (
- secs_to_hms(elapsed),
- ("%s/%s:" % (n+1,len(addr_data))),
- i[1], "(" + label + ")"
- )
- )
- time.sleep(1)
- else:
- if err_flag: msg("\nImport failed"); sys.exit(2)
- msg("\nOK")
- break
- else:
- import_address(i[1],label,rescan=False)
- msg_r(msg_fmt % (
- ("%s/%s:" % (n+1,len(addr_data))),
- i[1], "(" + label + ")"
- )
- )
- if err_flag: msg("\nImport failed"); sys.exit(2)
- msg(" - OK")
|