mmgen-addrimport 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. #!/usr/bin/env python
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C) 2013 by philemon <mmgen-py@yandex.com>
  5. #
  6. # This program is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation, either version 3 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. """
  19. mmgen-addrimport: Import addresses into a bitcoind watching wallet.
  20. """
  21. import sys
  22. from mmgen.Opts import *
  23. from mmgen.license import *
  24. from mmgen.util import check_infile,confirm_or_exit,msg,msg_r,secs_to_hms,get_lines_from_file
  25. from mmgen.tx import connect_to_bitcoind,parse_addrs_file
  26. from mmgen.bitcoin import verify_addr
  27. help_data = {
  28. 'prog_name': sys.argv[0].split("/")[-1],
  29. 'desc': """Import addresses (both mmgen and non-mmgen) into a bitcoind
  30. watching wallet""",
  31. 'usage':"[opts] [mmgen address file]",
  32. 'options': """
  33. -h, --help Print this help message
  34. -l, --addrlist f Import the non-mmgen Bitcoin addresses listed in file 'f'
  35. -q, --quiet Suppress warnings
  36. """
  37. }
  38. short_opts = "hl:q"
  39. long_opts = "help", "addrlist=", "quiet"
  40. opts,cmd_args = process_opts(sys.argv,help_data,"".join(short_opts),long_opts)
  41. if len(cmd_args) != 1 and not 'addrlist' in opts:
  42. msg("You must specify an mmgen address list (and/or non-mmgen addresses with the '--addrlist' option)")
  43. sys.exit(1)
  44. check_opts(opts,long_opts)
  45. if cmd_args:
  46. check_infile(cmd_args[0])
  47. seed_id,addr_data = parse_addrs_file(cmd_args[0])
  48. else:
  49. seed_id,addr_data = "",[]
  50. if 'addrlist' in opts:
  51. l = get_lines_from_file(
  52. opts['addrlist'],"non-mmgen addresses",remove_comments=True)
  53. addr_data += [(None,i) for i in l]
  54. msg_r("Validating addresses...")
  55. for i in addr_data:
  56. if not verify_addr(i[1]):
  57. msg("%s: invalid address" % i)
  58. sys.exit(2)
  59. msg("OK")
  60. import mmgen.config as g
  61. g.http_timeout = 3600
  62. c = connect_to_bitcoind()
  63. if not 'quiet' in opts:
  64. message = """
  65. Importing addresses can take a long time (>30 min.) on a low-powered computer.
  66. """
  67. confirm_or_exit(message, "continue", expect="YES")
  68. import threading
  69. import time
  70. err_flag = False
  71. def import_address(addr,label):
  72. try:
  73. c.importaddress(addr,label)
  74. except:
  75. global err_flag
  76. err_flag = True
  77. w1 = len(str(len(addr_data))) * 2 + 2
  78. w2 = len(str(max([i[0] for i in addr_data if i[0]]))) + 12
  79. msg_fmt = "\r%s %-" + str(w1) + "s %-34s %-" + str(w2) + "s"
  80. msg("Importing addresses")
  81. for n,i in enumerate(addr_data):
  82. if i[0]:
  83. comment = " " + i[2] if len(i) == 3 else ""
  84. label = "%s:%s%s" % (seed_id,i[0],comment)
  85. else: label = "non-mmgen"
  86. t = threading.Thread(target=import_address, args = (i[1],label))
  87. t.daemon = True
  88. t.start()
  89. start = int(time.time())
  90. while True:
  91. if t.is_alive():
  92. elapsed = int(time.time() - start)
  93. msg_r(msg_fmt % (
  94. secs_to_hms(elapsed),
  95. ("%s/%s:" % (n+1,len(addr_data))),
  96. i[1], "(" + label + ")"
  97. )
  98. )
  99. time.sleep(1)
  100. else:
  101. if err_flag:
  102. msg("\nImport failed")
  103. sys.exit(2)
  104. msg("\nOK")
  105. break