update blob download script

This commit is contained in:
Alex Grintsvayg 2018-05-02 13:23:57 -04:00
parent 1f1b348638
commit f32861923e

View file

@ -14,7 +14,7 @@ from lbrynet.core import log_support, Wallet, Peer
from lbrynet.core.SinglePeerDownloader import SinglePeerDownloader from lbrynet.core.SinglePeerDownloader import SinglePeerDownloader
from lbrynet.core.StreamDescriptor import BlobStreamDescriptorReader from lbrynet.core.StreamDescriptor import BlobStreamDescriptorReader
from lbrynet.core.BlobManager import DiskBlobManager from lbrynet.core.BlobManager import DiskBlobManager
from lbrynet.dht.hashannouncer import DummyHashAnnouncer from lbrynet.database.storage import SQLiteStorage
log = logging.getLogger() log = logging.getLogger()
@ -45,13 +45,13 @@ def main(args=None):
@defer.inlineCallbacks @defer.inlineCallbacks
def download_it(peer, timeout, blob_hash): def download_it(peer, timeout, blob_hash):
tmp_dir = yield threads.deferToThread(tempfile.mkdtemp) tmp_dir = yield threads.deferToThread(tempfile.mkdtemp)
announcer = DummyHashAnnouncer() storage = SQLiteStorage(tmp_dir, reactor)
tmp_blob_manager = DiskBlobManager(announcer, tmp_dir, tmp_dir) yield storage.setup()
tmp_blob_manager = DiskBlobManager(tmp_dir, storage)
config = {'auto_connect': True} config = {'auto_connect': True}
if conf.settings['lbryum_wallet_dir']: if conf.settings['lbryum_wallet_dir']:
config['lbryum_path'] = conf.settings['lbryum_wallet_dir'] config['lbryum_path'] = conf.settings['lbryum_wallet_dir']
storage = Wallet.InMemoryStorage()
wallet = Wallet.LBRYumWallet(storage, config) wallet = Wallet.LBRYumWallet(storage, config)
downloader = SinglePeerDownloader() downloader = SinglePeerDownloader()
@ -76,8 +76,9 @@ def download_it(peer, timeout, blob_hash):
pass pass
if info: if info:
break 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: if info is not None:
pprint(info) pprint(info)