From f32861923e0565d11e10989eeebd6122d1ab130b Mon Sep 17 00:00:00 2001 From: Alex Grintsvayg <grin@lbry.io> Date: Wed, 2 May 2018 13:23:57 -0400 Subject: [PATCH] update blob download script --- scripts/download_blob_from_peer.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/scripts/download_blob_from_peer.py b/scripts/download_blob_from_peer.py index dc688956d..80b4c40c3 100644 --- a/scripts/download_blob_from_peer.py +++ b/scripts/download_blob_from_peer.py @@ -14,7 +14,7 @@ from lbrynet.core import log_support, Wallet, Peer from lbrynet.core.SinglePeerDownloader import SinglePeerDownloader from lbrynet.core.StreamDescriptor import BlobStreamDescriptorReader from lbrynet.core.BlobManager import DiskBlobManager -from lbrynet.dht.hashannouncer import DummyHashAnnouncer +from lbrynet.database.storage import SQLiteStorage log = logging.getLogger() @@ -45,13 +45,13 @@ def main(args=None): @defer.inlineCallbacks def download_it(peer, timeout, blob_hash): tmp_dir = yield threads.deferToThread(tempfile.mkdtemp) - announcer = DummyHashAnnouncer() - tmp_blob_manager = DiskBlobManager(announcer, tmp_dir, tmp_dir) + storage = SQLiteStorage(tmp_dir, reactor) + yield storage.setup() + tmp_blob_manager = DiskBlobManager(tmp_dir, storage) config = {'auto_connect': True} if conf.settings['lbryum_wallet_dir']: config['lbryum_path'] = conf.settings['lbryum_wallet_dir'] - storage = Wallet.InMemoryStorage() wallet = Wallet.LBRYumWallet(storage, config) downloader = SinglePeerDownloader() @@ -76,8 +76,9 @@ def download_it(peer, timeout, blob_hash): pass if info: break - time.sleep( - 0.1) # there's some kind of race condition where it sometimes doesnt write the blob to disk in time + + # there's some kind of race condition where it sometimes doesnt write the blob to disk in time + time.sleep(0.1) if info is not None: pprint(info)