forked from LBRYCommunity/lbry-sdk
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
import sys
|
|
import os
|
|
import asyncio
|
|
import socket
|
|
import ipaddress
|
|
from lbrynet.conf import Config
|
|
from lbrynet.extras.daemon.storage import SQLiteStorage
|
|
from lbrynet.blob.blob_manager import BlobFileManager
|
|
from lbrynet.blob_exchange.client import BlobExchangeClientProtocol, request_blob
|
|
import logging
|
|
|
|
log = logging.getLogger("lbrynet")
|
|
log.addHandler(logging.StreamHandler())
|
|
log.setLevel(logging.DEBUG)
|
|
|
|
|
|
async def main(blob_hash: str, url: str):
|
|
conf = Config()
|
|
loop = asyncio.get_running_loop()
|
|
host_url, port = url.split(":")
|
|
try:
|
|
host = None
|
|
if ipaddress.ip_address(host_url):
|
|
host = host_url
|
|
except ValueError:
|
|
host = None
|
|
if not host:
|
|
host_info = await loop.getaddrinfo(
|
|
host_url, 'https',
|
|
proto=socket.IPPROTO_TCP,
|
|
)
|
|
host = host_info[0][4][0]
|
|
|
|
storage = SQLiteStorage(conf, os.path.join(conf.data_dir, "lbrynet.sqlite"))
|
|
blob_manager = BlobFileManager(loop, os.path.join(conf.data_dir, "blobfiles"), storage)
|
|
await storage.open()
|
|
await blob_manager.setup()
|
|
|
|
blob = blob_manager.get_blob(blob_hash)
|
|
success, keep = await request_blob(loop, blob, host, int(port), conf.peer_connect_timeout,
|
|
conf.blob_download_timeout)
|
|
print(f"{'downloaded' if success else 'failed to download'} {blob_hash} from {host}:{port}\n"
|
|
f"keep connection: {keep}")
|
|
if blob.get_is_verified():
|
|
await blob_manager.delete_blobs([blob.blob_hash])
|
|
print(f"deleted {blob_hash}")
|
|
|
|
|
|
if __name__ == "__main__": # usage: python download_blob_from_peer.py <blob_hash> [host url:port]
|
|
url = 'reflector.lbry.io:5567'
|
|
if len(sys.argv) > 2:
|
|
url = sys.argv[2]
|
|
asyncio.run(main(sys.argv[1], url))
|