fix #984 + cleanup (#1006)

CityOfZion · Aug 21, 2019 · 357c00c · 357c00c
1 parent a45ae92
commit 357c00c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/neo/Prompt/Commands/Bootstrap.py b/neo/Prompt/Commands/Bootstrap.py
@@ -1,6 +1,6 @@
 import sys
 from neo.Settings import settings
-from neo.Network.common import blocking_prompt as prompt
+from prompt_toolkit import prompt
 import requests
 from tqdm import tqdm
 import tarfile

diff --git a/neo/bin/import_blocks.py b/neo/bin/import_blocks.py
@@ -3,11 +3,9 @@
 from neo.Core.Blockchain import Blockchain
 from neo.Core.Block import Block
 from neo.Storage.Implementation.DBFactory import getBlockchainDB
-from neo.Storage.Common.DBPrefix import DBPrefix
 from neo.Settings import settings
 from neo.Core.IO.BinaryReader import BinaryReader
-from neo.Core.IO.BinaryWriter import BinaryWriter
-from neo.IO.MemoryStream import StreamManager, MemoryStream
+from neo.IO.MemoryStream import MemoryStream
 import argparse
 import os
 import shutil
@@ -108,7 +106,6 @@ async def _main():
             print("Starting import at %s " % start_block)
 
             if args.totalblocks:
-                # total_blocks = start_block + args.totalblocks - 1
                 total_blocks = args.totalblocks
 
             for _ in trange(start_block, desc='Skipping blocks', unit='Block'):
@@ -185,9 +182,6 @@ async def _main():
             block.DeserializeForImport(reader)
             header_hash_list.append(block.Hash.ToBytes())
 
-            # if block.Index > 942371:
-            #     break
-
             # add
             if block.Index >= start_block:
                 chain.AddHeaders([block.Header])