forked from LBRYCommunity/lbry-sdk
Merge pull request #317 from lbryio/remove-lbrynet-console
Remove lbrynet-console
This commit is contained in:
Jack Robison
GitHub
commit
f84df50d7f
22 changed files with 1 additions and 4858 deletions
|
|
@ -1,574 +0,0 @@
|
|||
# TODO: THERE IS A LOT OF CODE IN THIS MODULE THAT SHOULD BE REMOVED
|
||||
# AS IT IS REPEATED IN THE LBRYDaemon MODULE
|
||||
import logging
|
||||
import os.path
|
||||
import argparse
|
||||
import requests
|
||||
import locale
|
||||
import sys
|
||||
import webbrowser
|
||||
|
||||
if sys.platform == "darwin":
|
||||
from appdirs import user_data_dir
|
||||
from yapsy.PluginManager import PluginManager
|
||||
from twisted.internet import defer, threads, stdio, task, error
|
||||
from lbrynet.lbrynet_daemon.auth.client import LBRYAPIClient
|
||||
|
||||
from lbrynet import analytics
|
||||
from lbrynet.core.Session import Session
|
||||
from lbrynet.lbrynet_console.ConsoleControl import ConsoleControl
|
||||
from lbrynet.lbrynet_console.Settings import Settings
|
||||
from lbrynet.lbryfilemanager.EncryptedFileManager import EncryptedFileManager
|
||||
from lbrynet.conf import settings
|
||||
from lbrynet.core.utils import generate_id
|
||||
from lbrynet.core.StreamDescriptor import StreamDescriptorIdentifier
|
||||
from lbrynet.core.PaymentRateManager import PaymentRateManager
|
||||
from lbrynet.core.server.BlobRequestHandler import BlobRequestHandlerFactory
|
||||
from lbrynet.core.server.ServerProtocol import ServerProtocolFactory
|
||||
from lbrynet.core.PTCWallet import PTCWallet
|
||||
from lbrynet.lbryfile.client.EncryptedFileOptions import add_lbry_file_to_sd_identifier
|
||||
from lbrynet.lbryfile.client.EncryptedFileDownloader import EncryptedFileOpenerFactory
|
||||
from lbrynet.lbryfile.StreamDescriptor import EncryptedFileStreamType
|
||||
from lbrynet.lbryfile.EncryptedFileMetadataManager import DBEncryptedFileMetadataManager, TempEncryptedFileMetadataManager
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ApplicationStatusFactory, GetWalletBalancesFactory, ShutDownFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ImmediateAnnounceAllBlobsFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import EncryptedFileStatusFactory, DeleteEncryptedFileChooserFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ToggleEncryptedFileRunningChooserFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ModifyApplicationDefaultsFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import CreateEncryptedFileFactory, PublishStreamDescriptorChooserFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ShowPublishedSDHashesChooserFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import CreatePlainStreamDescriptorChooserFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ShowEncryptedFileStreamHashChooserFactory, AddStreamFromHashFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import AddStreamFromSDFactory, AddStreamFromLBRYcrdNameFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ClaimNameFactory, GetNewWalletAddressFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ShowServerStatusFactory, ModifyServerSettingsFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import ModifyEncryptedFileOptionsChooserFactory, StatusFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import PeerStatsAndSettingsChooserFactory, PublishFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import BlockchainStatusFactory
|
||||
from lbrynet.core.Wallet import LBRYumWallet
|
||||
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
alert = logging.getLogger("lbryalert." + __name__)
|
||||
|
||||
|
||||
class Console():
|
||||
"""A class which can upload and download file streams to and from the network"""
|
||||
def __init__(self, peer_port, dht_node_port, known_dht_nodes, fake_wallet,
|
||||
lbrycrd_conf, lbrycrd_dir, use_upnp, data_dir, created_data_dir,
|
||||
lbrycrdd_path):
|
||||
"""
|
||||
@param peer_port: the network port on which to listen for peers
|
||||
|
||||
@param dht_node_port: the network port on which to listen for dht node requests
|
||||
|
||||
@param known_dht_nodes: a list of (ip_address, dht_port) which will be used to join the DHT network
|
||||
"""
|
||||
self.peer_port = peer_port
|
||||
self.dht_node_port = dht_node_port
|
||||
self.known_dht_nodes = known_dht_nodes
|
||||
self.fake_wallet = fake_wallet
|
||||
self.lbrycrd_conf = lbrycrd_conf
|
||||
self.lbrycrd_dir = lbrycrd_dir
|
||||
if not self.lbrycrd_dir:
|
||||
if sys.platform == "darwin":
|
||||
self.lbrycrd_dir = os.path.join(os.path.expanduser("~"), "Library/Application Support/lbrycrd")
|
||||
else:
|
||||
self.lbrycrd_dir = os.path.join(os.path.expanduser("~"), ".lbrycrd")
|
||||
if not self.lbrycrd_conf:
|
||||
self.lbrycrd_conf = os.path.join(self.lbrycrd_dir, "lbrycrd.conf")
|
||||
self.lbrycrdd_path = lbrycrdd_path
|
||||
self.use_upnp = use_upnp
|
||||
self.lbry_server_port = None
|
||||
self.session = None
|
||||
self.lbry_file_metadata_manager = None
|
||||
self.lbry_file_manager = None
|
||||
self.db_dir = data_dir
|
||||
self.current_db_revision = 1
|
||||
self.blobfile_dir = os.path.join(self.db_dir, "blobfiles")
|
||||
self.created_data_dir = created_data_dir
|
||||
self.plugin_manager = PluginManager()
|
||||
self.plugin_manager.setPluginPlaces([
|
||||
os.path.join(self.db_dir, "plugins"),
|
||||
os.path.join(os.path.dirname(__file__), "plugins"),
|
||||
])
|
||||
self.command_handlers = []
|
||||
self.query_handlers = {}
|
||||
|
||||
self.settings = Settings(self.db_dir)
|
||||
self.blob_request_payment_rate_manager = None
|
||||
self.lbryid = None
|
||||
self.sd_identifier = StreamDescriptorIdentifier()
|
||||
self.plugin_objects = []
|
||||
self.db_migration_revisions = None
|
||||
|
||||
def start(self):
|
||||
"""Initialize the session and restore everything to its saved state"""
|
||||
d = self._setup_controller()
|
||||
d.addCallback(lambda _: threads.deferToThread(self._setup_data_directory))
|
||||
d.addCallback(lambda _: self._check_db_migration())
|
||||
d.addCallback(lambda _: self._get_settings())
|
||||
d.addCallback(lambda _: self._get_session())
|
||||
d.addCallback(lambda _: add_lbry_file_to_sd_identifier(self.sd_identifier))
|
||||
d.addCallback(lambda _: self._setup_lbry_file_manager())
|
||||
d.addCallback(lambda _: self._setup_lbry_file_opener())
|
||||
d.addCallback(lambda _: self._setup_control_handlers())
|
||||
d.addCallback(lambda _: self._setup_query_handlers())
|
||||
d.addCallback(lambda _: self._load_plugins())
|
||||
d.addCallback(lambda _: self._setup_server())
|
||||
d.addCallback(lambda _: self._start_controller())
|
||||
d.addErrback(self._show_start_error)
|
||||
return d
|
||||
|
||||
def _show_start_error(self, error):
|
||||
print error.getTraceback()
|
||||
log.error("An error occurred during start up: %s", error.getTraceback())
|
||||
return error
|
||||
|
||||
def _show_shutdown_error(self, error):
|
||||
print error.getErrorMessage()
|
||||
log.error("An error occurred during shutdown: %s", error.getTraceback())
|
||||
return error
|
||||
|
||||
def shut_down(self):
|
||||
"""Stop the session, all currently running streams, and stop the server"""
|
||||
d = self._shut_down()
|
||||
if self.session is not None:
|
||||
d.addCallback(lambda _: self.session.shut_down())
|
||||
d.addErrback(self._show_shutdown_error)
|
||||
return d
|
||||
|
||||
def add_control_handlers(self, control_handlers):
|
||||
for control_handler in control_handlers:
|
||||
self.command_handlers.append(control_handler)
|
||||
|
||||
def add_query_handlers(self, query_handlers):
|
||||
|
||||
def _set_query_handlers(statuses):
|
||||
from future_builtins import zip
|
||||
for handler, (success, status) in zip(query_handlers, statuses):
|
||||
if success is True:
|
||||
self.query_handlers[handler] = status
|
||||
|
||||
ds = []
|
||||
for handler in query_handlers:
|
||||
ds.append(self.settings.get_query_handler_status(handler.get_primary_query_identifier()))
|
||||
dl = defer.DeferredList(ds)
|
||||
dl.addCallback(_set_query_handlers)
|
||||
return dl
|
||||
|
||||
def _setup_data_directory(self):
|
||||
alert.info("Loading databases...")
|
||||
if self.created_data_dir:
|
||||
db_revision = open(os.path.join(self.db_dir, "db_revision"), mode='w')
|
||||
db_revision.write(str(self.current_db_revision))
|
||||
db_revision.close()
|
||||
log.debug("Created the db revision file: %s", str(os.path.join(self.db_dir, "db_revision")))
|
||||
if not os.path.exists(self.blobfile_dir):
|
||||
os.mkdir(self.blobfile_dir)
|
||||
log.debug("Created the blobfile directory: %s", str(self.blobfile_dir))
|
||||
|
||||
def _check_db_migration(self):
|
||||
old_revision = 1
|
||||
db_revision_file = os.path.join(self.db_dir, "db_revision")
|
||||
if os.path.exists(db_revision_file):
|
||||
old_revision = int(open(db_revision_file).read().strip())
|
||||
if old_revision < self.current_db_revision:
|
||||
from lbrynet.db_migrator import dbmigrator
|
||||
print "Upgrading your databases..."
|
||||
d = threads.deferToThread(dbmigrator.migrate_db, self.db_dir, old_revision, self.current_db_revision)
|
||||
|
||||
def print_success(old_dirs):
|
||||
success_string = "Finished upgrading the databases. It is now safe to delete the"
|
||||
success_string += " following directories, if you feel like it. It won't make any"
|
||||
success_string += " difference.\nAnyway here they are: "
|
||||
for i, old_dir in enumerate(old_dirs):
|
||||
success_string += old_dir
|
||||
if i + 1 < len(old_dir):
|
||||
success_string += ", "
|
||||
print success_string
|
||||
|
||||
d.addCallback(print_success)
|
||||
return d
|
||||
return defer.succeed(True)
|
||||
|
||||
def _get_settings(self):
|
||||
d = self.settings.start()
|
||||
d.addCallback(lambda _: self.settings.get_lbryid())
|
||||
d.addCallback(self.set_lbryid)
|
||||
return d
|
||||
|
||||
def set_lbryid(self, lbryid):
|
||||
if lbryid is None:
|
||||
return self._make_lbryid()
|
||||
else:
|
||||
self.lbryid = lbryid
|
||||
|
||||
def _make_lbryid(self):
|
||||
self.lbryid = generate_id()
|
||||
d = self.settings.save_lbryid(self.lbryid)
|
||||
return d
|
||||
|
||||
def _get_session(self):
|
||||
def get_default_data_rate():
|
||||
d = self.settings.get_default_data_payment_rate()
|
||||
d.addCallback(lambda rate: {"default_data_payment_rate": rate if rate is not None else settings.data_rate})
|
||||
return d
|
||||
|
||||
def get_wallet():
|
||||
if self.fake_wallet:
|
||||
d = defer.succeed(PTCWallet(self.db_dir))
|
||||
else:
|
||||
d = defer.succeed(LBRYumWallet(self.db_dir))
|
||||
d.addCallback(lambda wallet: {"wallet": wallet})
|
||||
return d
|
||||
|
||||
d1 = get_default_data_rate()
|
||||
d2 = get_wallet()
|
||||
|
||||
def combine_results(results):
|
||||
r = {}
|
||||
for success, result in results:
|
||||
if success is True:
|
||||
r.update(result)
|
||||
return r
|
||||
|
||||
def create_session(results):
|
||||
|
||||
alert.info("Databases loaded.")
|
||||
|
||||
self.session = Session(results['default_data_payment_rate'], db_dir=self.db_dir, lbryid=self.lbryid,
|
||||
blob_dir=self.blobfile_dir, dht_node_port=self.dht_node_port,
|
||||
known_dht_nodes=self.known_dht_nodes, peer_port=self.peer_port,
|
||||
use_upnp=self.use_upnp, wallet=results['wallet'])
|
||||
|
||||
dl = defer.DeferredList([d1, d2], fireOnOneErrback=True)
|
||||
|
||||
dl.addCallback(combine_results)
|
||||
|
||||
dl.addCallback(create_session)
|
||||
|
||||
dl.addCallback(lambda _: self.session.setup())
|
||||
|
||||
dl.addCallback(lambda _: self.check_first_run())
|
||||
|
||||
dl.addCallback(self._show_first_run_result)
|
||||
|
||||
return dl
|
||||
|
||||
def check_first_run(self):
|
||||
d = self.session.wallet.is_first_run()
|
||||
d.addCallback(lambda is_first_run: self._do_first_run() if is_first_run else 0.0)
|
||||
return d
|
||||
|
||||
def _do_first_run(self):
|
||||
d = self.session.wallet.get_new_address()
|
||||
|
||||
def send_request(url, data):
|
||||
r = requests.post(url, json=data)
|
||||
if r.status_code == 200:
|
||||
return r.json()['credits_sent']
|
||||
return 0.0
|
||||
|
||||
def log_error(err):
|
||||
log.warning("unable to request free credits. %s", err.getErrorMessage())
|
||||
return 0.0
|
||||
|
||||
def request_credits(address):
|
||||
url = "http://credreq.lbry.io/requestcredits"
|
||||
data = {"address": address}
|
||||
d = threads.deferToThread(send_request, url, data)
|
||||
d.addErrback(log_error)
|
||||
return d
|
||||
|
||||
d.addCallback(request_credits)
|
||||
return d
|
||||
|
||||
@staticmethod
|
||||
def _show_first_run_result(credits_received):
|
||||
if credits_received != 0.0:
|
||||
points_string = locale.format_string("%.2f LBC", (round(credits_received, 2),),
|
||||
grouping=True)
|
||||
alert.info("\n\nThank you for testing the alpha version of LBRY!\n\n"
|
||||
"You have been given %s for free because we love you.\n"
|
||||
"Please give them a few minutes to show up while you\n"
|
||||
"catch up with our blockchain.\n", points_string)
|
||||
|
||||
def _setup_lbry_file_manager(self):
|
||||
self.lbry_file_metadata_manager = DBEncryptedFileMetadataManager(self.db_dir)
|
||||
d = self.lbry_file_metadata_manager.setup()
|
||||
|
||||
def set_lbry_file_manager():
|
||||
self.lbry_file_manager = EncryptedFileManager(self.session, self.lbry_file_metadata_manager, self.sd_identifier)
|
||||
return self.lbry_file_manager.setup()
|
||||
|
||||
d.addCallback(lambda _: set_lbry_file_manager())
|
||||
|
||||
return d
|
||||
|
||||
def _setup_lbry_file_opener(self):
|
||||
stream_info_manager = TempEncryptedFileMetadataManager()
|
||||
downloader_factory = EncryptedFileOpenerFactory(self.session.peer_finder, self.session.rate_limiter,
|
||||
self.session.blob_manager, stream_info_manager,
|
||||
self.session.wallet)
|
||||
self.sd_identifier.add_stream_downloader_factory(EncryptedFileStreamType, downloader_factory)
|
||||
return defer.succeed(True)
|
||||
|
||||
def _setup_control_handlers(self):
|
||||
handlers = [
|
||||
ApplicationStatusFactory(self.session.rate_limiter, self.session.dht_node),
|
||||
GetWalletBalancesFactory(self.session.wallet),
|
||||
ModifyApplicationDefaultsFactory(self),
|
||||
ShutDownFactory(self),
|
||||
PeerStatsAndSettingsChooserFactory(self.session.peer_manager),
|
||||
EncryptedFileStatusFactory(self.lbry_file_manager),
|
||||
AddStreamFromSDFactory(self.sd_identifier, self.session.base_payment_rate_manager,
|
||||
self.session.wallet),
|
||||
DeleteEncryptedFileChooserFactory(self.lbry_file_metadata_manager, self.session.blob_manager,
|
||||
self.lbry_file_manager),
|
||||
ToggleEncryptedFileRunningChooserFactory(self.lbry_file_manager),
|
||||
CreateEncryptedFileFactory(self.session, self.lbry_file_manager),
|
||||
PublishStreamDescriptorChooserFactory(self.lbry_file_metadata_manager,
|
||||
self.session.blob_manager),
|
||||
ShowPublishedSDHashesChooserFactory(self.lbry_file_metadata_manager,
|
||||
self.lbry_file_manager),
|
||||
CreatePlainStreamDescriptorChooserFactory(self.lbry_file_manager),
|
||||
ShowEncryptedFileStreamHashChooserFactory(self.lbry_file_manager),
|
||||
ModifyEncryptedFileOptionsChooserFactory(self.lbry_file_manager),
|
||||
AddStreamFromHashFactory(self.sd_identifier, self.session, self.session.wallet),
|
||||
StatusFactory(self, self.session.rate_limiter, self.lbry_file_manager,
|
||||
self.session.blob_manager, self.session.wallet if not self.fake_wallet else None),
|
||||
ImmediateAnnounceAllBlobsFactory(self.session.blob_manager)
|
||||
]
|
||||
self.add_control_handlers(handlers)
|
||||
if not self.fake_wallet:
|
||||
lbrycrd_handlers = [
|
||||
AddStreamFromLBRYcrdNameFactory(self.sd_identifier, self.session,
|
||||
self.session.wallet),
|
||||
ClaimNameFactory(self.session.wallet, self.lbry_file_manager,
|
||||
self.session.blob_manager),
|
||||
GetNewWalletAddressFactory(self.session.wallet),
|
||||
PublishFactory(self.session, self.lbry_file_manager, self.session.wallet),
|
||||
BlockchainStatusFactory(self.session.wallet)
|
||||
]
|
||||
self.add_control_handlers(lbrycrd_handlers)
|
||||
if self.peer_port is not None:
|
||||
server_handlers = [
|
||||
ShowServerStatusFactory(self),
|
||||
ModifyServerSettingsFactory(self),
|
||||
]
|
||||
self.add_control_handlers(server_handlers)
|
||||
|
||||
def _setup_query_handlers(self):
|
||||
handlers = [
|
||||
self.session.wallet.get_wallet_info_query_handler_factory(),
|
||||
]
|
||||
|
||||
def get_blob_request_handler_factory(rate):
|
||||
self.blob_request_payment_rate_manager = PaymentRateManager(
|
||||
self.session.base_payment_rate_manager, rate
|
||||
)
|
||||
handlers.append(BlobRequestHandlerFactory(
|
||||
self.session.blob_manager, self.session.wallet,
|
||||
self.blob_request_payment_rate_manager, analytics.Track()
|
||||
))
|
||||
|
||||
d1 = self.settings.get_server_data_payment_rate()
|
||||
d1.addCallback(get_blob_request_handler_factory)
|
||||
|
||||
dl = defer.DeferredList([d1])
|
||||
dl.addCallback(lambda _: self.add_query_handlers(handlers))
|
||||
return dl
|
||||
|
||||
def _load_plugins(self):
|
||||
d = threads.deferToThread(self.plugin_manager.collectPlugins)
|
||||
|
||||
def setup_plugins():
|
||||
ds = []
|
||||
for plugin in self.plugin_manager.getAllPlugins():
|
||||
self.plugin_objects.append(plugin.plugin_object)
|
||||
ds.append(plugin.plugin_object.setup(self))
|
||||
return defer.DeferredList(ds)
|
||||
|
||||
d.addCallback(lambda _: setup_plugins())
|
||||
return d
|
||||
|
||||
def _stop_plugins(self):
|
||||
ds = []
|
||||
for plugin_object in self.plugin_objects:
|
||||
ds.append(defer.maybeDeferred(plugin_object.stop))
|
||||
return defer.DeferredList(ds)
|
||||
|
||||
def _setup_server(self):
|
||||
|
||||
def restore_running_status(running):
|
||||
if running is True:
|
||||
return self.start_server()
|
||||
return defer.succeed(True)
|
||||
|
||||
dl = self.settings.get_server_running_status()
|
||||
dl.addCallback(restore_running_status)
|
||||
return dl
|
||||
|
||||
def start_server(self):
|
||||
|
||||
if self.peer_port is not None:
|
||||
|
||||
server_factory = ServerProtocolFactory(self.session.rate_limiter,
|
||||
self.query_handlers,
|
||||
self.session.peer_manager)
|
||||
from twisted.internet import reactor
|
||||
try:
|
||||
self.lbry_server_port = reactor.listenTCP(self.peer_port, server_factory)
|
||||
except error.CannotListenError as e:
|
||||
import traceback
|
||||
log.error("Couldn't bind to port %d. %s", self.peer_port, traceback.format_exc())
|
||||
raise ValueError("%s lbrynet may already be running on your computer.", str(e))
|
||||
return defer.succeed(True)
|
||||
|
||||
def stop_server(self):
|
||||
if self.lbry_server_port is not None:
|
||||
self.lbry_server_port, p = None, self.lbry_server_port
|
||||
return defer.maybeDeferred(p.stopListening)
|
||||
else:
|
||||
return defer.succeed(True)
|
||||
|
||||
def _setup_controller(self):
|
||||
self.controller = ConsoleControl()
|
||||
stdio.StandardIO(self.controller)
|
||||
logger = logging.getLogger()
|
||||
formatter = logging.Formatter("%(message)s")
|
||||
alert_handler = logging.StreamHandler(self.controller)
|
||||
alert_handler.setFormatter(formatter)
|
||||
alert_handler.addFilter(logging.Filter("lbryalert"))
|
||||
alert_handler.setLevel(logging.DEBUG)
|
||||
logger.addHandler(alert_handler)
|
||||
return defer.succeed(True)
|
||||
|
||||
def _start_controller(self):
|
||||
return self.controller.start(self.command_handlers)
|
||||
|
||||
def _shut_down(self):
|
||||
self.plugin_manager = None
|
||||
ds = []
|
||||
if self.lbry_file_metadata_manager is not None:
|
||||
d = self.lbry_file_metadata_manager.stop()
|
||||
d.addCallback(lambda _: self.lbry_file_manager.stop())
|
||||
ds.append(d)
|
||||
ds.append(self.stop_server())
|
||||
ds.append(self._stop_plugins())
|
||||
dl = defer.DeferredList(ds)
|
||||
return dl
|
||||
|
||||
|
||||
def launch_lbry_console():
|
||||
from twisted.internet import reactor
|
||||
|
||||
parser = argparse.ArgumentParser(description="Launch a lbrynet console")
|
||||
parser.add_argument("--no_listen_peer",
|
||||
help="Don't listen for incoming data connections.",
|
||||
action="store_true")
|
||||
parser.add_argument("--peer_port",
|
||||
help="The port on which the console will listen for incoming data connections.",
|
||||
type=int, default=3333)
|
||||
parser.add_argument("--no_listen_dht",
|
||||
help="Don't listen for incoming DHT connections.",
|
||||
action="store_true")
|
||||
parser.add_argument("--dht_node_port",
|
||||
help="The port on which the console will listen for DHT connections.",
|
||||
type=int, default=4444)
|
||||
parser.add_argument("--fake_wallet",
|
||||
help="Testing purposes only. Use a non-blockchain wallet.",
|
||||
action="store_true")
|
||||
parser.add_argument("--no_dht_bootstrap",
|
||||
help="Don't try to connect to the DHT",
|
||||
action="store_true")
|
||||
parser.add_argument("--dht_bootstrap_host",
|
||||
help="The hostname of a known DHT node, to be used to bootstrap into the DHT. "
|
||||
"Must be used with --dht_bootstrap_port",
|
||||
type=str, default='104.236.42.182')
|
||||
parser.add_argument("--dht_bootstrap_port",
|
||||
help="The port of a known DHT node, to be used to bootstrap into the DHT. Must "
|
||||
"be used with --dht_bootstrap_host",
|
||||
type=int, default=4000)
|
||||
parser.add_argument("--disable_upnp",
|
||||
help="Don't try to use UPnP to enable incoming connections through the firewall",
|
||||
action="store_true")
|
||||
parser.add_argument("--data_dir",
|
||||
help=("The full path to the directory in which lbrynet data and metadata will be stored. "
|
||||
"Default: ~/.lbrynet on linux, ~/Library/Application Support/lbrynet on OS X"),
|
||||
type=str)
|
||||
parser.add_argument("--lbrycrdd_path",
|
||||
help="The path to lbrycrdd, which will be launched if it isn't running. If"
|
||||
"this option is chosen, lbrycrdd will be used as the interface to the"
|
||||
"blockchain. By default, a lightweight interface is used.")
|
||||
parser.add_argument("--lbrycrd_wallet_dir",
|
||||
help="The directory in which lbrycrd data will stored. Used if lbrycrdd is "
|
||||
"launched by this application.")
|
||||
parser.add_argument("--lbrycrd_wallet_conf",
|
||||
help="The configuration file for the LBRYcrd wallet. Default: ~/.lbrycrd/lbrycrd.conf",
|
||||
type=str)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.no_dht_bootstrap:
|
||||
bootstrap_nodes = []
|
||||
else:
|
||||
bootstrap_nodes = [(args.dht_bootstrap_host, args.dht_bootstrap_port)]
|
||||
|
||||
if args.no_listen_peer:
|
||||
peer_port = None
|
||||
else:
|
||||
peer_port = args.peer_port
|
||||
|
||||
if args.no_listen_dht:
|
||||
dht_node_port = None
|
||||
else:
|
||||
dht_node_port = args.dht_node_port
|
||||
|
||||
created_data_dir = False
|
||||
if not args.data_dir:
|
||||
if sys.platform == "darwin":
|
||||
data_dir = user_data_dir("LBRY")
|
||||
else:
|
||||
data_dir = os.path.join(os.path.expanduser("~"), ".lbrynet")
|
||||
else:
|
||||
data_dir = args.data_dir
|
||||
if not os.path.exists(data_dir):
|
||||
os.mkdir(data_dir)
|
||||
created_data_dir = True
|
||||
|
||||
daemon = LBRYAPIClient.config()
|
||||
try:
|
||||
daemon.is_running()
|
||||
log.info("Attempt to start lbrynet-console while lbrynet-daemon is running")
|
||||
print "lbrynet-daemon is running, you must turn it off before using lbrynet-console"
|
||||
print "If you're running the app, quit before starting lbrynet-console"
|
||||
print "If you're running lbrynet-daemon in a terminal, run 'stop-lbrynet-daemon' to turn it off"
|
||||
|
||||
webbrowser.open("http://localhost:5279")
|
||||
|
||||
except:
|
||||
log_format = "(%(asctime)s)[%(filename)s:%(lineno)s] %(funcName)s(): %(message)s"
|
||||
formatter = logging.Formatter(log_format)
|
||||
|
||||
logger = logging.getLogger()
|
||||
logger.setLevel(logging.DEBUG)
|
||||
file_handler = logging.FileHandler(os.path.join(data_dir, "console.log"))
|
||||
file_handler.setFormatter(formatter)
|
||||
file_handler.addFilter(logging.Filter("lbrynet"))
|
||||
logger.addHandler(file_handler)
|
||||
|
||||
|
||||
console = Console(peer_port, dht_node_port, bootstrap_nodes, fake_wallet=args.fake_wallet,
|
||||
lbrycrd_conf=args.lbrycrd_wallet_conf, lbrycrd_dir=args.lbrycrd_wallet_dir,
|
||||
use_upnp=not args.disable_upnp, data_dir=data_dir,
|
||||
created_data_dir=created_data_dir, lbrycrdd_path=args.lbrycrdd_path)
|
||||
|
||||
d = task.deferLater(reactor, 0, console.start)
|
||||
d.addErrback(lambda _: reactor.stop())
|
||||
reactor.addSystemEventTrigger('before', 'shutdown', console.shut_down)
|
||||
reactor.run()
|
||||
|
||||
if __name__ == "__main__":
|
||||
launch_lbry_console()
|
||||
|
|
@ -1,176 +0,0 @@
|
|||
from twisted.protocols import basic
|
||||
from twisted.internet import defer
|
||||
import logging
|
||||
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class ConsoleControl(basic.LineReceiver):
|
||||
from os import linesep as delimiter
|
||||
|
||||
def __init__(self):
|
||||
self.connected = False
|
||||
self.buffer = []
|
||||
self.command_handlers = {}
|
||||
self.current_handler = None
|
||||
|
||||
def start(self, command_handlers):
|
||||
self.command_handlers = {h.command: h for h in command_handlers}
|
||||
self.current_handler = None
|
||||
self.send_initial_prompt()
|
||||
return defer.succeed(True)
|
||||
|
||||
def connectionMade(self):
|
||||
self.connected = True
|
||||
if self.buffer:
|
||||
self.send(self.buffer)
|
||||
self.buffer = []
|
||||
|
||||
def send_initial_prompt(self):
|
||||
self.sendLine("")
|
||||
self.sendLine("You should have received 1000 LBC the first time you ran\n"
|
||||
"this program. If you did not, let us know! But first give\n"
|
||||
"them a moment to show up. The block time is currently 30\n"
|
||||
"seconds so they should show up within a couple of minutes.\n\n"
|
||||
"Welcome to lbrynet-console!")
|
||||
self.sendLine("")
|
||||
self.sendLine("Enter a command. Try 'get wonderfullife' or 'help' to see more options.")
|
||||
self.show_prompt()
|
||||
|
||||
def send(self, s):
|
||||
self.transport.write(s)
|
||||
|
||||
def write(self, s):
|
||||
if self.connected is False:
|
||||
self.buffer.append(s)
|
||||
else:
|
||||
self.send(s)
|
||||
|
||||
def flush(self):
|
||||
if self.connected is True and self.buffer:
|
||||
self.send(self.buffer)
|
||||
self.buffer = []
|
||||
|
||||
def show_prompt(self):
|
||||
self.send("> ")
|
||||
|
||||
def show_quick_help(self):
|
||||
self.sendLine("Available commands:")
|
||||
self.sendLine("")
|
||||
showed_help_all = False
|
||||
sorted_handlers = sorted(self.command_handlers.items(), key=lambda x: x[0])
|
||||
sorted_handlers = sorted(sorted_handlers, key=lambda x: x[1].priority, reverse=True)
|
||||
for command, handler in sorted_handlers:
|
||||
if handler.priority > 0:
|
||||
if showed_help_all is False and handler.priority < 10:
|
||||
self.sendLine("help-all - Show the full list of available commands")
|
||||
showed_help_all = True
|
||||
self.sendLine(command + " - " + handler.short_help)
|
||||
self.sendLine("")
|
||||
self.sendLine("For more information about any command type 'help <command>'")
|
||||
|
||||
def show_full_help(self):
|
||||
self.sendLine("Available commands:")
|
||||
self.sendLine("")
|
||||
for command, handler in sorted(self.command_handlers.items(), key=lambda x: x[0]):
|
||||
self.sendLine(command + " - " + handler.short_help)
|
||||
self.sendLine("")
|
||||
self.sendLine("For more information about any command type 'help <command>'")
|
||||
|
||||
def handler_done(self):
|
||||
self.current_handler = None
|
||||
self.show_prompt()
|
||||
|
||||
def handler_failed(self, err):
|
||||
self.current_handler = None
|
||||
self.sendLine("An error occurred:")
|
||||
self.sendLine(err.getTraceback())
|
||||
self.show_prompt()
|
||||
|
||||
def lineReceived(self, line):
|
||||
if not self.command_handlers:
|
||||
return
|
||||
if self.current_handler is None:
|
||||
words = line.split()
|
||||
if len(words) == 0:
|
||||
self.show_prompt()
|
||||
return
|
||||
command, args = words[0], words[1:]
|
||||
if command == "help":
|
||||
if len(args) == 0:
|
||||
self.show_quick_help()
|
||||
self.show_prompt()
|
||||
return
|
||||
if args[0] in self.command_handlers:
|
||||
self.sendLine(self.command_handlers[args[0]].full_help)
|
||||
self.show_prompt()
|
||||
return
|
||||
if args[0] == "help-all":
|
||||
self.sendLine("Show the full list of available commands!")
|
||||
self.show_prompt()
|
||||
return
|
||||
self.sendLine("Can't help you with '%s'. Sorry!" % args[0])
|
||||
self.show_prompt()
|
||||
return
|
||||
elif command == "help-all":
|
||||
self.show_full_help()
|
||||
self.show_prompt()
|
||||
return
|
||||
if command in self.command_handlers:
|
||||
command_handler = self.command_handlers[command]
|
||||
else:
|
||||
candidates = [k for k in self.command_handlers.keys() if k.startswith(command)]
|
||||
if len(candidates) == 0:
|
||||
self.sendLine("Unknown command. Type 'help' for a list of commands.")
|
||||
self.show_prompt()
|
||||
return
|
||||
if len(candidates) >= 2:
|
||||
l = "Ambiguous command. Matches: "
|
||||
for candidate in candidates:
|
||||
l += candidate
|
||||
l += ", "
|
||||
l = l[:-2]
|
||||
l += l
|
||||
self.sendLine(l)
|
||||
self.show_prompt()
|
||||
return
|
||||
else:
|
||||
command_handler = self.command_handlers[candidates[0]]
|
||||
try:
|
||||
self.current_handler = command_handler.get_handler(self)
|
||||
except:
|
||||
self.current_handler = None
|
||||
import traceback
|
||||
self.sendLine(traceback.format_exc())
|
||||
log.error(traceback.format_exc())
|
||||
self.show_prompt()
|
||||
return
|
||||
try:
|
||||
self.current_handler.start(*args)
|
||||
except TypeError:
|
||||
self.current_handler = None
|
||||
self.sendLine("Invalid arguments. Type 'help <command>' for the argument list.")
|
||||
import traceback
|
||||
log.error(traceback.format_exc())
|
||||
self.show_prompt()
|
||||
return
|
||||
except:
|
||||
self.current_handler = None
|
||||
import traceback
|
||||
self.sendLine(traceback.format_exc())
|
||||
log.error(traceback.format_exc())
|
||||
self.show_prompt()
|
||||
return
|
||||
self.current_handler.finished_deferred.addCallbacks(lambda _: self.handler_done(),
|
||||
self.handler_failed)
|
||||
else:
|
||||
try:
|
||||
self.current_handler.handle_line(line)
|
||||
except Exception as e:
|
||||
self.current_handler = None
|
||||
import traceback
|
||||
self.sendLine(traceback.format_exc())
|
||||
log.error(traceback.format_exc())
|
||||
self.show_prompt()
|
||||
return
|
||||
File diff suppressed because it is too large
Load diff
|
|
@ -1,13 +0,0 @@
|
|||
from yapsy.IPlugin import IPlugin
|
||||
|
||||
|
||||
class Plugin(IPlugin):
|
||||
|
||||
def __init__(self):
|
||||
IPlugin.__init__(self)
|
||||
|
||||
def setup(self, lbry_console):
|
||||
raise NotImplementedError
|
||||
|
||||
def stop(self):
|
||||
raise NotImplementedError
|
||||
|
|
@ -1,8 +0,0 @@
|
|||
"""
|
||||
A plugin-enabled console application for interacting with the LBRY network called lbrynet-console.
|
||||
|
||||
lbrynet-console can be used to download and upload LBRY Files and includes plugins for streaming
|
||||
LBRY Files to an external application and to download unknown chunks of data for the purpose of
|
||||
re-uploading them. It gives the user some control over how much will be paid for data and
|
||||
metadata and also what types of queries from clients.
|
||||
"""
|
||||
|
|
@ -1,24 +0,0 @@
|
|||
from zope.interface import Interface
|
||||
|
||||
|
||||
class IControlHandlerFactory(Interface):
|
||||
def get_prompt_description(self):
|
||||
pass
|
||||
|
||||
def get_handler(self):
|
||||
pass
|
||||
|
||||
|
||||
class IControlHandler(Interface):
|
||||
def handle_line(self, line):
|
||||
pass
|
||||
|
||||
|
||||
class ICommandHandlerFactory(Interface):
|
||||
def get_handler(self):
|
||||
pass
|
||||
|
||||
|
||||
class ICommandHandler(Interface):
|
||||
def handle_line(self, line):
|
||||
pass
|
||||
|
|
@ -1,15 +0,0 @@
|
|||
from zope.interface import implements
|
||||
from lbrynet.interfaces import IBlobHandler
|
||||
from twisted.internet import defer
|
||||
|
||||
|
||||
class BlindBlobHandler(object):
|
||||
implements(IBlobHandler)
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
######### IBlobHandler #########
|
||||
|
||||
def handle_blob(self, blob, blob_info):
|
||||
return defer.succeed(True)
|
||||
|
|
@ -1,78 +0,0 @@
|
|||
from twisted.internet import defer
|
||||
from ValuableBlobInfo import ValuableBlobInfo
|
||||
import os
|
||||
import sqlite3
|
||||
from twisted.enterprise import adbapi
|
||||
from lbrynet.core.sqlite_helpers import rerun_if_locked
|
||||
|
||||
|
||||
class BlindInfoManager(object):
|
||||
|
||||
def __init__(self, db_dir, peer_manager):
|
||||
self.db_dir = db_dir
|
||||
self.db_conn = None
|
||||
self.peer_manager = peer_manager
|
||||
|
||||
def setup(self):
|
||||
# check_same_thread=False is solely to quiet a spurious error that appears to be due
|
||||
# to a bug in twisted, where the connection is closed by a different thread than the
|
||||
# one that opened it. The individual connections in the pool are not used in multiple
|
||||
# threads.
|
||||
self.db_conn = adbapi.ConnectionPool('sqlite3', os.path.join(self.db_dir, "blind_info.db"),
|
||||
check_same_thread=False)
|
||||
|
||||
def set_up_table(transaction):
|
||||
transaction.execute("create table if not exists valuable_blobs (" +
|
||||
" blob_hash text primary key, " +
|
||||
" blob_length integer, " +
|
||||
" reference text, " +
|
||||
" peer_host text, " +
|
||||
" peer_port integer, " +
|
||||
" peer_score text" +
|
||||
")")
|
||||
return self.db_conn.runInteraction(set_up_table)
|
||||
|
||||
def stop(self):
|
||||
self.db = None
|
||||
return defer.succeed(True)
|
||||
|
||||
def get_all_blob_infos(self):
|
||||
d = self._get_all_blob_infos()
|
||||
|
||||
def make_blob_infos(blob_data):
|
||||
blob_infos = []
|
||||
for blob in blob_data:
|
||||
blob_hash, length, reference, peer_host, peer_port, peer_score = blob
|
||||
peer = self.peer_manager.get_peer(peer_host, peer_port)
|
||||
blob_info = ValuableBlobInfo(blob_hash, length, reference, peer, peer_score)
|
||||
blob_infos.append(blob_info)
|
||||
return blob_infos
|
||||
d.addCallback(make_blob_infos)
|
||||
return d
|
||||
|
||||
def save_blob_infos(self, blob_infos):
|
||||
blobs = []
|
||||
for blob_info in blob_infos:
|
||||
blob_hash = blob_info.blob_hash
|
||||
length = blob_info.length
|
||||
reference = blob_info.reference
|
||||
peer_host = blob_info.peer.host
|
||||
peer_port = blob_info.peer.port
|
||||
peer_score = blob_info.peer_score
|
||||
blobs.append((blob_hash, length, reference, peer_host, peer_port, peer_score))
|
||||
return self._save_blob_infos(blobs)
|
||||
|
||||
@rerun_if_locked
|
||||
def _get_all_blob_infos(self):
|
||||
return self.db_conn.runQuery("select * from valuable_blobs")
|
||||
|
||||
@rerun_if_locked
|
||||
def _save_blob_infos(self, blobs):
|
||||
def save_infos(transaction):
|
||||
for blob in blobs:
|
||||
try:
|
||||
transaction.execute("insert into valuable_blobs values (?, ?, ?, ?, ?, ?)",
|
||||
blob)
|
||||
except sqlite3.IntegrityError:
|
||||
pass
|
||||
return self.db_conn.runInteraction(save_infos)
|
||||
|
|
@ -1,321 +0,0 @@
|
|||
from zope.interface import implements
|
||||
from lbrynet.interfaces import IMetadataHandler, IRequestCreator
|
||||
from lbrynet.core.client.ClientRequest import ClientRequest, ClientPaidRequest
|
||||
from lbrynet.core.Error import InsufficientFundsError, InvalidResponseError, RequestCanceledError
|
||||
from lbrynet.core.Error import NoResponseError, ConnectionClosedBeforeResponseError
|
||||
from ValuableBlobInfo import ValuableBlobInfo
|
||||
import datetime
|
||||
import logging
|
||||
import random
|
||||
from twisted.internet import defer
|
||||
from twisted.python.failure import Failure
|
||||
from collections import defaultdict
|
||||
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class BlindMetadataHandler(object):
|
||||
implements(IMetadataHandler, IRequestCreator)
|
||||
|
||||
def __init__(self, info_manager, peers, peer_finder, approved_peers, payment_rate_manager, wallet,
|
||||
download_manager):
|
||||
self.info_manager = info_manager
|
||||
self.payment_rate_manager = payment_rate_manager
|
||||
self.wallet = wallet
|
||||
self.download_manager = download_manager
|
||||
self._peers = peers # {Peer: score}
|
||||
self.peer_finder = peer_finder
|
||||
self.approved_peers = approved_peers
|
||||
self._valuable_protocol_prices = {}
|
||||
self._info_protocol_prices = {}
|
||||
self._price_disagreements = [] # [Peer]
|
||||
self._incompatible_peers = [] # [Peer]
|
||||
self._last_blob_hashes_from_peers = {} # {Peer: (blob_hash, expire_time)}
|
||||
self._valuable_hashes = {} # {blob_hash: (peer score, reference, peer)}
|
||||
self._blob_infos = {} # {blob_hash: ValuableBlobInfo}
|
||||
self._peer_search_results = defaultdict(list) # {peer: [blob_hash]}
|
||||
|
||||
######### IMetadataHandler #########
|
||||
|
||||
def get_initial_blobs(self):
|
||||
d = self.info_manager.get_all_blob_infos()
|
||||
return d
|
||||
|
||||
def final_blob_num(self):
|
||||
return None
|
||||
|
||||
######### IRequestCreator #########
|
||||
|
||||
def send_next_request(self, peer, protocol):
|
||||
# Basic idea:
|
||||
# If the peer has been sending us blob hashes to download recently (10 minutes?),
|
||||
# send back an example of one (the most recent?) so that it can
|
||||
# keep sending us more like it. Otherwise, just ask for
|
||||
# valuable blobs
|
||||
sent_request = False
|
||||
if self._should_send_request_to(peer):
|
||||
v_r = self._get_valuable_blob_request(peer)
|
||||
if v_r is not None:
|
||||
v_p_r = self._get_valuable_price_request(peer, protocol)
|
||||
reserved_points = self._reserve_points_valuable(peer, protocol, v_r.max_pay_units)
|
||||
if reserved_points is not None:
|
||||
d1 = protocol.add_request(v_r)
|
||||
d1.addCallback(self._handle_valuable_blob_response, peer, v_r)
|
||||
d1.addBoth(self._pay_or_cancel_payment, protocol, reserved_points,
|
||||
self._info_protocol_prices)
|
||||
d1.addErrback(self._request_failed, "valuable blob request", peer)
|
||||
sent_request = True
|
||||
if v_p_r is not None:
|
||||
d2 = protocol.add_request(v_p_r)
|
||||
d2.addCallback(self._handle_valuable_price_response, peer, v_p_r, protocol)
|
||||
d2.addErrback(self._request_failed, "valuable price request", peer)
|
||||
else:
|
||||
return defer.fail(InsufficientFundsError())
|
||||
i_r = self._get_info_request(peer)
|
||||
if i_r is not None:
|
||||
i_p_r = self._get_info_price_request(peer, protocol)
|
||||
reserved_points = self._reserve_points_info(peer, protocol, i_r.max_pay_units)
|
||||
if reserved_points is not None:
|
||||
d3 = protocol.add_request(i_r)
|
||||
d3.addCallback(self._handle_info_response, peer, i_r, protocol, reserved_points)
|
||||
d3.addBoth(self._pay_or_cancel_payment, protocol, reserved_points,
|
||||
self._valuable_protocol_prices)
|
||||
d3.addErrback(self._request_failed, "info request", peer, reserved_points)
|
||||
sent_request = True
|
||||
if i_p_r is not None:
|
||||
d4 = protocol.add_request(i_p_r)
|
||||
d4.addCallback(self._handle_info_price_response, peer, i_p_r, protocol)
|
||||
d4.addErrback(self._request_failed, "info price request", peer)
|
||||
else:
|
||||
return defer.fail(InsufficientFundsError())
|
||||
return defer.succeed(sent_request)
|
||||
|
||||
def get_new_peers(self):
|
||||
peers = None
|
||||
if self._peer_search_results:
|
||||
peers = self._peer_search_results.keys()
|
||||
elif len(self.approved_peers) != 0:
|
||||
peers = random.sample(self.approved_peers, len(self.approved_peers))
|
||||
return defer.succeed(peers)
|
||||
|
||||
######### internal #########
|
||||
|
||||
def _should_send_request_to(self, peer):
|
||||
if peer in self._incompatible_peers:
|
||||
return False
|
||||
if self._peers[peer] >= 0:
|
||||
return True
|
||||
return False
|
||||
|
||||
def _get_valuable_blob_request(self, peer):
|
||||
blob_hash = None
|
||||
if peer in self._last_blob_hashes_from_peers:
|
||||
h, expire_time = self._last_blob_hashes_from_peers[peer]
|
||||
if datetime.datetime.now() > expire_time:
|
||||
del self._last_blob_hashes_from_peers[peer]
|
||||
else:
|
||||
blob_hash = h
|
||||
r_dict = {'valuable_blob_hashes': {'reference': blob_hash, 'max_blob_hashes': 20}}
|
||||
response_identifier = 'valuable_blob_hashes'
|
||||
request = ClientPaidRequest(r_dict, response_identifier, 20)
|
||||
return request
|
||||
|
||||
def _get_valuable_price_request(self, peer, protocol):
|
||||
request = None
|
||||
if not protocol in self._valuable_protocol_prices:
|
||||
self._valuable_protocol_prices[protocol] = self.payment_rate_manager.get_rate_valuable_blob_hash(peer)
|
||||
request_dict = {'valuable_blob_payment_rate': self._valuable_protocol_prices[protocol]}
|
||||
request = ClientRequest(request_dict, 'valuable_blob_payment_rate')
|
||||
return request
|
||||
|
||||
def _get_info_request(self, peer):
|
||||
if peer in self._peer_search_results:
|
||||
blob_hashes = self._peer_search_results[peer]
|
||||
del self._peer_search_results[peer]
|
||||
references = []
|
||||
for blob_hash in blob_hashes:
|
||||
if blob_hash in self._valuable_hashes:
|
||||
references.append(self._valuable_hashes[blob_hash][1])
|
||||
hashes_to_search = [h for h, (s, r, p) in self._valuable_hashes.iteritems() if r in references]
|
||||
if hashes_to_search:
|
||||
r_dict = {'blob_length': {'blob_hashes': hashes_to_search}}
|
||||
response_identifier = 'blob_length'
|
||||
request = ClientPaidRequest(r_dict, response_identifier, len(hashes_to_search))
|
||||
return request
|
||||
if not self._peer_search_results:
|
||||
self._search_for_peers()
|
||||
return None
|
||||
|
||||
def _get_info_price_request(self, peer, protocol):
|
||||
request = None
|
||||
if not protocol in self._info_protocol_prices:
|
||||
self._info_protocol_prices[protocol] = self.payment_rate_manager.get_rate_valuable_blob_info(peer)
|
||||
request_dict = {'blob_length_payment_rate': self._info_protocol_prices[protocol]}
|
||||
request = ClientRequest(request_dict, 'blob_length_payment_rate')
|
||||
return request
|
||||
|
||||
def _update_local_score(self, peer, amount):
|
||||
self._peers[peer] += amount
|
||||
|
||||
def _reserve_points_valuable(self, peer, protocol, max_units):
|
||||
return self._reserve_points(peer, protocol, max_units, self._valuable_protocol_prices)
|
||||
|
||||
def _reserve_points_info(self, peer, protocol, max_units):
|
||||
return self._reserve_points(peer, protocol, max_units, self._info_protocol_prices)
|
||||
|
||||
def _reserve_points(self, peer, protocol, max_units, prices):
|
||||
assert protocol in prices
|
||||
points_to_reserve = 1.0 * max_units * prices[protocol] / 1000.0
|
||||
return self.wallet.reserve_points(peer, points_to_reserve)
|
||||
|
||||
def _pay_or_cancel_payment(self, arg, protocol, reserved_points, protocol_prices):
|
||||
if isinstance(arg, Failure) or arg == 0:
|
||||
self._cancel_points(reserved_points)
|
||||
else:
|
||||
self._pay_peer(protocol, arg, reserved_points, protocol_prices)
|
||||
return arg
|
||||
|
||||
def _pay_peer(self, protocol, num_units, reserved_points, prices):
|
||||
assert num_units != 0
|
||||
assert protocol in prices
|
||||
point_amount = 1.0 * num_units * prices[protocol] / 1000.0
|
||||
self.wallet.send_points(reserved_points, point_amount)
|
||||
|
||||
def _cancel_points(self, reserved_points):
|
||||
self.wallet.cancel_point_reservation(reserved_points)
|
||||
|
||||
def _handle_valuable_blob_response(self, response_dict, peer, request):
|
||||
if not request.response_identifier in response_dict:
|
||||
return InvalidResponseError("response identifier not in response")
|
||||
response = response_dict[request.response_identifier]
|
||||
if 'error' in response:
|
||||
if response['error'] == "RATE_UNSET":
|
||||
return 0
|
||||
else:
|
||||
return InvalidResponseError("Got an unknown error from the peer: %s" %
|
||||
(response['error'],))
|
||||
if not 'valuable_blob_hashes' in response:
|
||||
return InvalidResponseError("Missing the required field 'valuable_blob_hashes'")
|
||||
hashes = response['valuable_blob_hashes']
|
||||
log.info("Handling %s valuable blob hashes from %s", str(len(hashes)), str(peer))
|
||||
expire_time = datetime.datetime.now() + datetime.timedelta(minutes=10)
|
||||
reference = None
|
||||
unique_hashes = set()
|
||||
if 'reference' in response:
|
||||
reference = response['reference']
|
||||
for blob_hash, peer_score in hashes:
|
||||
if reference is None:
|
||||
reference = blob_hash
|
||||
self._last_blob_hashes_from_peers[peer] = (blob_hash, expire_time)
|
||||
if not (blob_hash in self._valuable_hashes or blob_hash in self._blob_infos):
|
||||
self._valuable_hashes[blob_hash] = (peer_score, reference, peer)
|
||||
unique_hashes.add(blob_hash)
|
||||
|
||||
if len(unique_hashes):
|
||||
self._update_local_score(peer, len(unique_hashes))
|
||||
peer.update_stats('downloaded_valuable_blob_hashes', len(unique_hashes))
|
||||
peer.update_score(len(unique_hashes))
|
||||
else:
|
||||
self._update_local_score(peer, -.0001)
|
||||
return len(unique_hashes)
|
||||
|
||||
def _handle_info_response(self, response_dict, peer, request):
|
||||
if not request.response_identifier in response_dict:
|
||||
return InvalidResponseError("response identifier not in response")
|
||||
response = response_dict[request.response_identifier]
|
||||
if 'error' in response:
|
||||
if response['error'] == 'RATE_UNSET':
|
||||
return 0
|
||||
else:
|
||||
return InvalidResponseError("Got an unknown error from the peer: %s" %
|
||||
(response['error'],))
|
||||
if not 'blob_lengths' in response:
|
||||
return InvalidResponseError("Missing the required field 'blob_lengths'")
|
||||
raw_blob_lengths = response['blob_lengths']
|
||||
log.info("Handling %s blob lengths from %s", str(len(raw_blob_lengths)), str(peer))
|
||||
log.debug("blobs: %s", str(raw_blob_lengths))
|
||||
infos = []
|
||||
unique_hashes = set()
|
||||
for blob_hash, length in raw_blob_lengths:
|
||||
if blob_hash in self._valuable_hashes:
|
||||
peer_score, reference, peer = self._valuable_hashes[blob_hash]
|
||||
del self._valuable_hashes[blob_hash]
|
||||
infos.append(ValuableBlobInfo(blob_hash, length, reference, peer, peer_score))
|
||||
unique_hashes.add(blob_hash)
|
||||
elif blob_hash in request.request_dict['blob_length']['blob_hashes']:
|
||||
unique_hashes.add(blob_hash)
|
||||
d = self.info_manager.save_blob_infos(infos)
|
||||
d.addCallback(lambda _: self.download_manager.add_blobs_to_download(infos))
|
||||
|
||||
def pay_or_penalize_peer():
|
||||
if len(unique_hashes):
|
||||
self._update_local_score(peer, len(unique_hashes))
|
||||
peer.update_stats('downloaded_valuable_blob_infos', len(unique_hashes))
|
||||
peer.update_score(len(unique_hashes))
|
||||
else:
|
||||
self._update_local_score(peer, -.0001)
|
||||
return len(unique_hashes)
|
||||
|
||||
d.addCallback(lambda _: pay_or_penalize_peer())
|
||||
|
||||
return d
|
||||
|
||||
def _handle_valuable_price_response(self, response_dict, peer, request, protocol):
|
||||
if not request.response_identifier in response_dict:
|
||||
return InvalidResponseError("response identifier not in response")
|
||||
assert protocol in self._valuable_protocol_prices
|
||||
response = response_dict[request.response_identifier]
|
||||
if response == "RATE_ACCEPTED":
|
||||
return True
|
||||
else:
|
||||
del self._valuable_protocol_prices[protocol]
|
||||
self._price_disagreements.append(peer)
|
||||
return True
|
||||
|
||||
def _handle_info_price_response(self, response_dict, peer, request, protocol):
|
||||
if not request.response_identifier in response_dict:
|
||||
return InvalidResponseError("response identifier not in response")
|
||||
assert protocol in self._info_protocol_prices
|
||||
response = response_dict[request.response_identifier]
|
||||
if response == "RATE_ACCEPTED":
|
||||
return True
|
||||
else:
|
||||
del self._info_protocol_prices[protocol]
|
||||
self._price_disagreements.append(peer)
|
||||
return True
|
||||
|
||||
def _request_failed(self, reason, request_type, peer):
|
||||
if reason.check(RequestCanceledError):
|
||||
return
|
||||
if reason.check(NoResponseError):
|
||||
self._incompatible_peers.append(peer)
|
||||
log.warning("Valuable blob info requester: a request of type %s has failed. Reason: %s",
|
||||
str(request_type), str(reason.getErrorMessage()))
|
||||
self._update_local_score(peer, -10.0)
|
||||
peer.update_score(-5.0)
|
||||
if reason.check(ConnectionClosedBeforeResponseError):
|
||||
return
|
||||
return reason
|
||||
|
||||
def _search_for_peers(self):
|
||||
references_with_sources = set()
|
||||
for h_list in self._peer_search_results.itervalues():
|
||||
for h in h_list:
|
||||
if h in self._valuable_hashes:
|
||||
references_with_sources.add(self._valuable_hashes[h][1])
|
||||
hash_to_search = None
|
||||
used_references = []
|
||||
for h, (s, r, p) in self._valuable_hashes.iteritems():
|
||||
if not r in used_references:
|
||||
used_references.append(r)
|
||||
hash_to_search = h
|
||||
if not r in references_with_sources:
|
||||
break
|
||||
if hash_to_search:
|
||||
d = self.peer_finder.find_peers_for_blob(hash_to_search)
|
||||
d.addCallback(self._set_peer_search_results, hash_to_search)
|
||||
|
||||
def _set_peer_search_results(self, peers, searched_hash):
|
||||
for peer in peers:
|
||||
self._peer_search_results[peer].append(searched_hash)
|
||||
|
|
@ -1,93 +0,0 @@
|
|||
from zope.interface import implements
|
||||
from lbrynet.interfaces import IProgressManager
|
||||
from twisted.internet import defer
|
||||
|
||||
|
||||
class BlindProgressManager(object):
|
||||
implements(IProgressManager)
|
||||
|
||||
def __init__(self, blob_manager, peers, max_space, blob_scorers, download_manager):
|
||||
self.blob_manager = blob_manager
|
||||
self.peers = peers
|
||||
self.max_space = max_space
|
||||
self.blob_scorers = blob_scorers
|
||||
self.download_manager = download_manager
|
||||
self.paused = True
|
||||
self.blobs_to_download = []
|
||||
self._next_manage_downloaded_blobs = None
|
||||
|
||||
def set_max_space(self, max_space):
|
||||
self.max_space = max_space
|
||||
|
||||
######### IProgressManager #########
|
||||
|
||||
def start(self):
|
||||
from twisted.internet import reactor
|
||||
|
||||
self.paused = False
|
||||
self._next_manage_downloaded_blobs = reactor.callLater(0, self._manage_downloaded_blobs)
|
||||
return defer.succeed(True)
|
||||
|
||||
def stop(self):
|
||||
self.paused = True
|
||||
if self._next_manage_downloaded_blobs is not None and self._next_manage_downloaded_blobs.active():
|
||||
self._next_manage_downloaded_blobs.cancel()
|
||||
self._next_manage_downloaded_blobs = None
|
||||
return defer.succeed(True)
|
||||
|
||||
def stream_position(self):
|
||||
return 0
|
||||
|
||||
def needed_blobs(self):
|
||||
needed_blobs = [b for b in self.blobs_to_download if not b.is_validated()]
|
||||
return sorted(needed_blobs, key=lambda x: x.is_downloading(), reverse=True)[:20]
|
||||
|
||||
######### internal #########
|
||||
|
||||
def _manage_downloaded_blobs(self):
|
||||
|
||||
self._next_manage_downloaded_blobs = None
|
||||
|
||||
from twisted.internet import reactor
|
||||
|
||||
blobs = self.download_manager.blobs
|
||||
blob_infos = self.download_manager.blob_infos
|
||||
|
||||
blob_hashes = [b.blob_hash for b in blobs]
|
||||
|
||||
blobs_to_score = [(blobs[blob_hash], blob_infos[blob_hash]) for blob_hash in blob_hashes]
|
||||
|
||||
scores = self._score_blobs(blobs_to_score)
|
||||
|
||||
from future_builtins import zip
|
||||
|
||||
scored_blobs = zip(blobs_to_score, scores)
|
||||
ranked_blobs = sorted(scored_blobs, key=lambda x: x[1], reverse=True)
|
||||
|
||||
space_so_far = 0
|
||||
blobs_to_delete = []
|
||||
blobs_to_download = []
|
||||
|
||||
for (blob, blob_info), score in ranked_blobs:
|
||||
space_so_far += blob.blob_length
|
||||
if blob.is_validated() and space_so_far >= self.max_space:
|
||||
blobs_to_delete.append(blob)
|
||||
elif not blob.is_validated() and space_so_far < self.max_space:
|
||||
blobs_to_download.append(blob)
|
||||
|
||||
self.blob_manager.delete_blobs(blobs_to_delete)
|
||||
self.blobs_to_download = blobs_to_download
|
||||
|
||||
self._next_manage_downloaded_blobs = reactor.callLater(30, self._manage_downloaded_blobs)
|
||||
|
||||
def _score_blobs(self, blobs):
|
||||
scores = []
|
||||
for blob, blob_info in blobs:
|
||||
summands = []
|
||||
multiplicands = []
|
||||
for blob_scorer in self.blob_scorers:
|
||||
s, m = blob_scorer.score_blob(blob, blob_info)
|
||||
summands.append(s)
|
||||
multiplicands.append(m)
|
||||
scores.append(1.0 * sum(summands) * reduce(lambda x, y: x * y, multiplicands, 1))
|
||||
return scores
|
||||
|
|
@ -1,113 +0,0 @@
|
|||
from twisted.internet import defer
|
||||
from twisted.python.failure import Failure
|
||||
from lbrynet.core.client.BlobRequester import BlobRequester
|
||||
from lbrynet.core.client.ConnectionManager import ConnectionManager
|
||||
from lbrynet.core.client.DownloadManager import DownloadManager
|
||||
from BlindMetadataHandler import BlindMetadataHandler
|
||||
from BlindProgressManager import BlindProgressManager
|
||||
from BlindBlobHandler import BlindBlobHandler
|
||||
from collections import defaultdict
|
||||
from interfaces import IBlobScorer
|
||||
from zope.interface import implements
|
||||
|
||||
|
||||
class PeerScoreBasedScorer(object):
|
||||
implements(IBlobScorer)
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def score_blob(self, blob, blob_info):
|
||||
return blob_info.peer_score, 1
|
||||
|
||||
|
||||
class LengthBasedScorer(object):
|
||||
implements(IBlobScorer)
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def score_blob(self, blob, blob_info):
|
||||
return 0, 1.0 * blob.get_length() / 2**21
|
||||
|
||||
|
||||
class BlindRepeater(object):
|
||||
def __init__(self, peer_finder, rate_limiter, blob_manager, info_manager, wallet, payment_rate_manager):
|
||||
self.peer_finder = peer_finder
|
||||
self.rate_limiter = rate_limiter
|
||||
self.blob_manager = blob_manager
|
||||
self.info_manager = info_manager
|
||||
self.wallet = wallet
|
||||
self.payment_rate_manager = payment_rate_manager
|
||||
self.download_manager = None
|
||||
self.progress_manager = None
|
||||
self.max_space = 0
|
||||
self.peers = defaultdict(int)
|
||||
self.approved_peers = set()
|
||||
self.stopped = True
|
||||
|
||||
def setup(self):
|
||||
return defer.succeed(True)
|
||||
|
||||
def start(self):
|
||||
if self.stopped is True:
|
||||
return self._start()
|
||||
else:
|
||||
return defer.fail(Failure(ValueError("The repeater is already running")))
|
||||
|
||||
def stop(self):
|
||||
if self.stopped is False:
|
||||
return self._stop()
|
||||
else:
|
||||
return defer.fail(Failure(ValueError("The repeater is not running")))
|
||||
|
||||
def status(self):
|
||||
if self.stopped is True:
|
||||
return "stopped"
|
||||
else:
|
||||
return "running"
|
||||
|
||||
def set_max_space(self, max_space):
|
||||
self.max_space = max_space
|
||||
if self.progress_manager is not None:
|
||||
self.progress_manager.set_max_space(self.max_space)
|
||||
|
||||
def add_approved_peer(self, peer):
|
||||
self.approved_peers.add(peer)
|
||||
|
||||
def remove_approved_peer(self, peer):
|
||||
self.approved_peers.remove(peer)
|
||||
|
||||
def _start(self):
|
||||
self.download_manager = DownloadManager(self.blob_manager, True)
|
||||
info_finder = BlindMetadataHandler(self.info_manager, self.peers, self.peer_finder,
|
||||
self.approved_peers, self.payment_rate_manager,
|
||||
self.wallet, self.download_manager)
|
||||
self.download_manager.blob_info_finder = info_finder
|
||||
blob_requester = BlobRequester(self.blob_manager, self.peer_finder, self.payment_rate_manager,
|
||||
self.wallet, self.download_manager)
|
||||
self.download_manager.blob_requester = blob_requester
|
||||
self.progress_manager = BlindProgressManager(self.blob_manager, self.peers, self.max_space,
|
||||
[PeerScoreBasedScorer(), LengthBasedScorer()],
|
||||
self.download_manager)
|
||||
self.download_manager.progress_manager = self.progress_manager
|
||||
self.download_manager.blob_handler = BlindBlobHandler()
|
||||
wallet_info_exchanger = self.wallet.get_info_exchanger()
|
||||
self.download_manager.wallet_info_exchanger = wallet_info_exchanger
|
||||
connection_manager = ConnectionManager(self, self.rate_limiter, [info_finder, blob_requester],
|
||||
[wallet_info_exchanger])
|
||||
self.download_manager.connection_manager = connection_manager
|
||||
d = defer.maybeDeferred(self.download_manager.start_downloading)
|
||||
d.addCallback(lambda _: self._update_status(stopped=False))
|
||||
return d
|
||||
|
||||
def _stop(self):
|
||||
d = defer.maybeDeferred(self.download_manager.stop_downloading)
|
||||
d.addCallback(lambda _: self._update_status(stopped=True))
|
||||
return d
|
||||
|
||||
def _update_status(self, stopped=True):
|
||||
self.stopped = stopped
|
||||
|
||||
def insufficient_funds(self, err):
|
||||
return self.stop()
|
||||
|
|
@ -1,335 +0,0 @@
|
|||
from lbrynet.lbrynet_console.ControlHandlers import CommandHandler, CommandHandlerFactory
|
||||
from lbrynet.lbrynet_console.ControlHandlers import RecursiveCommandHandler, ModifyPaymentRate
|
||||
from twisted.internet import defer
|
||||
|
||||
|
||||
class StartRepeater(CommandHandler):
|
||||
prompt_description = "Start the blind repeater"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
self.settings = settings
|
||||
|
||||
def start(self):
|
||||
#assert line is None, "Start repeater should not be passed any arguments"
|
||||
d = self.settings.save_repeater_status(running=True)
|
||||
d.addCallback(lambda _: self.repeater.start())
|
||||
d.addCallback(lambda _: self.console.sendLine("Started the repeater"))
|
||||
d.chainDeferred(self.finished_deferred)
|
||||
|
||||
|
||||
class StartRepeaterFactory(CommandHandlerFactory):
|
||||
control_handler_class = StartRepeater
|
||||
|
||||
|
||||
class StopRepeater(CommandHandler):
|
||||
prompt_description = "Stop the blind repeater"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
self.settings = settings
|
||||
|
||||
def start(self):
|
||||
#assert line is None, "Stop repeater should not be passed any arguments"
|
||||
d = self.settings.save_repeater_status(running=False)
|
||||
d.addCallback(lambda _: self.repeater.stop())
|
||||
d.addCallback(lambda _: self.console.sendLine("Stopped the repeater"))
|
||||
d.chainDeferred(self.finished_deferred)
|
||||
|
||||
|
||||
class StopRepeaterFactory(CommandHandlerFactory):
|
||||
control_handler_class = StopRepeater
|
||||
|
||||
|
||||
class UpdateMaxSpace(CommandHandler):
|
||||
prompt_description = "Set the maximum space to be used by the blind repeater"
|
||||
line_prompt = "Maximum space (in bytes):"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
self.settings = settings
|
||||
|
||||
def start(self):
|
||||
self.console.sendLine(self.line_prompt)
|
||||
|
||||
def handle_line(self, line):
|
||||
d = self._set_max_space(line)
|
||||
d.chainDeferred(self.finished_deferred)
|
||||
|
||||
def _set_max_space(self, line):
|
||||
max_space = int(line)
|
||||
d = self.settings.save_max_space(max_space)
|
||||
d.addCallback(lambda _: self.repeater.set_max_space(max_space))
|
||||
d.addCallback(lambda _: self.console.sendLine("Set the maximum space to " + str(max_space) + " bytes"))
|
||||
return d
|
||||
|
||||
|
||||
class UpdateMaxSpaceFactory(CommandHandlerFactory):
|
||||
control_handler_class = UpdateMaxSpace
|
||||
|
||||
|
||||
class AddApprovedPeer(CommandHandler):
|
||||
prompt_description = "Add a peer to the approved list of peers to check for valuable blob hashes"
|
||||
host_prompt = "Peer host in dotted quad (e.g. 127.0.0.1)"
|
||||
port_prompt = "Peer port (e.g. 4444)"
|
||||
|
||||
def __init__(self, console, repeater, peer_manager, settings):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
self.peer_manager = peer_manager
|
||||
self.settings = settings
|
||||
self.host_to_add = None
|
||||
|
||||
def start(self):
|
||||
self.console.sendLine(self.host_prompt)
|
||||
|
||||
def handle_line(self, line):
|
||||
#if line is None:
|
||||
# return False, defer.succeed(self.host_prompt)
|
||||
if self.host_to_add is None:
|
||||
self.host_to_add = line
|
||||
self.console.sendLine(self.port_prompt)
|
||||
else:
|
||||
self.host_to_add, host = None, self.host_to_add
|
||||
d = self._add_peer(host, line)
|
||||
d.chainDeferred(self.finished_deferred)
|
||||
|
||||
def _add_peer(self, host, port):
|
||||
peer = self.peer_manager.get_peer(host, int(port))
|
||||
d = self.settings.save_approved_peer(host, int(port))
|
||||
d.addCallback(lambda _: self.repeater.add_approved_peer(peer))
|
||||
d.addCallback(lambda _: self.console.sendLine("Successfully added peer"))
|
||||
return d
|
||||
|
||||
|
||||
class AddApprovedPeerFactory(CommandHandlerFactory):
|
||||
control_handler_class = AddApprovedPeer
|
||||
|
||||
|
||||
class ApprovedPeerChooser(RecursiveCommandHandler):
|
||||
|
||||
def __init__(self, console, repeater, factory_class, *args, **kwargs):
|
||||
self.repeater = repeater
|
||||
self.factory_class = factory_class
|
||||
self.args = args
|
||||
RecursiveCommandHandler.__init__(self, console, **kwargs)
|
||||
|
||||
def _get_control_handler_factories(self):
|
||||
control_handler_factories = []
|
||||
for peer in self.repeater.approved_peers:
|
||||
control_handler_factories.append(self.factory_class(peer, *self.args))
|
||||
return control_handler_factories
|
||||
|
||||
|
||||
class ApprovedPeerChooserFactory(CommandHandlerFactory):
|
||||
def get_prompt_description(self):
|
||||
peer = self.args[0]
|
||||
return str(peer)
|
||||
|
||||
|
||||
class DeleteApprovedPeerChooser(ApprovedPeerChooser):
|
||||
prompt_description = "Remove a peer from the approved list of peers to check for valuable blob hashes"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
ApprovedPeerChooser.__init__(self, console, repeater, DeleteApprovedPeerFactory, repeater,
|
||||
settings, exit_after_one_done=True)
|
||||
|
||||
|
||||
class DeleteApprovedPeerChooserFactory(CommandHandlerFactory):
|
||||
control_handler_class = DeleteApprovedPeerChooser
|
||||
|
||||
|
||||
class DeleteApprovedPeer(CommandHandler):
|
||||
prompt_description = "Remove a peer from the approved list of peers to check for valuable blob hashes"
|
||||
|
||||
def __init__(self, console, peer, repeater, settings):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
self.settings = settings
|
||||
self.peer_to_remove = peer
|
||||
|
||||
def start(self):
|
||||
d = self._remove_peer()
|
||||
d.chainDeferred(self.finished_deferred)
|
||||
|
||||
def _remove_peer(self):
|
||||
d = self.settings.remove_approved_peer(self.peer_to_remove.host, int(self.peer_to_remove.port))
|
||||
d.addCallback(lambda _: self.repeater.remove_approved_peer(self.peer_to_remove))
|
||||
d.addCallback(lambda _: self.console.sendLine("Successfully removed peer"))
|
||||
return d
|
||||
|
||||
|
||||
class DeleteApprovedPeerFactory(ApprovedPeerChooserFactory):
|
||||
control_handler_class = DeleteApprovedPeer
|
||||
|
||||
|
||||
class ShowApprovedPeers(CommandHandler):
|
||||
prompt_description = "Show the list of peers approved to be checked for valuable blob hashes"
|
||||
|
||||
def __init__(self, console, repeater):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
|
||||
def start(self):
|
||||
#assert line is None, "Show approved peers should not be passed any arguments"
|
||||
d = self._show_peers()
|
||||
d.chainDeferred(self.finished_deferred)
|
||||
|
||||
def _show_peers(self):
|
||||
peer_string = "Approved peers:\n"
|
||||
for peer in self.repeater.approved_peers:
|
||||
peer_string += str(peer) + "\n"
|
||||
self.console.sendLine(peer_string)
|
||||
return defer.succeed(None)
|
||||
|
||||
|
||||
class ShowApprovedPeersFactory(CommandHandlerFactory):
|
||||
control_handler_class = ShowApprovedPeers
|
||||
|
||||
|
||||
class RepeaterStatus(CommandHandler):
|
||||
prompt_description = "Show the repeater's status"
|
||||
|
||||
def __init__(self, console, repeater):
|
||||
CommandHandler.__init__(self, console)
|
||||
self.repeater = repeater
|
||||
|
||||
def start(self):
|
||||
#assert line is None, "Show repeater status should not be passed any arguments"
|
||||
self._show_status()
|
||||
self.finished_deferred.callback(None)
|
||||
|
||||
def _show_status(self):
|
||||
status_string = "Repeater status: " + self.repeater.status() + "\n"
|
||||
|
||||
if self.repeater.stopped is False:
|
||||
max_space = self.repeater.progress_manager.max_space
|
||||
space_used = 0
|
||||
for blob in self.repeater.download_manager.blobs:
|
||||
if blob.is_validated():
|
||||
space_used += blob.get_length()
|
||||
|
||||
status_string += "Maximum space: " + str(max_space) + " bytes\n"
|
||||
status_string += "Space used: " + str(space_used) + " bytes\n"
|
||||
self.console.sendLine(status_string)
|
||||
|
||||
|
||||
class RepeaterStatusFactory(CommandHandlerFactory):
|
||||
control_handler_class = RepeaterStatus
|
||||
|
||||
|
||||
class ModifyDataPaymentRate(ModifyPaymentRate):
|
||||
prompt_description = "Modify Blind Repeater data payment rate"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
ModifyPaymentRate.__init__(self, console)
|
||||
self._prompt_choices['unset'] = (self._unset, "Use the application default data rate")
|
||||
self.payment_rate_manager = repeater.payment_rate_manager
|
||||
self.settings = settings
|
||||
|
||||
def _unset(self):
|
||||
self._set_rate(None)
|
||||
return defer.succeed("Using the application default data rate")
|
||||
|
||||
def _set_rate(self, rate):
|
||||
|
||||
def set_data_payment_rate(rate):
|
||||
self.payment_rate_manager.min_blob_data_payment_rate = rate
|
||||
|
||||
d = self.settings.save_data_payment_rate(rate)
|
||||
d.addCallback(lambda _: set_data_payment_rate(rate))
|
||||
return d
|
||||
|
||||
def _get_current_status(self):
|
||||
effective_rate = self.payment_rate_manager.get_effective_min_blob_data_payment_rate()
|
||||
if self.payment_rate_manager.min_blob_data_payment_rate is None:
|
||||
status = "The current data payment rate is set to use the application default, "
|
||||
status += str(effective_rate)
|
||||
else:
|
||||
status = "The current data payment rate is "
|
||||
status += str(effective_rate)
|
||||
return status
|
||||
|
||||
|
||||
class ModifyDataPaymentRateFactory(CommandHandlerFactory):
|
||||
control_handler_class = ModifyDataPaymentRate
|
||||
|
||||
|
||||
class ModifyInfoPaymentRate(ModifyPaymentRate):
|
||||
prompt_description = "Modify Blind Repeater valuable info payment rate"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
ModifyPaymentRate.__init__(self, console)
|
||||
self.payment_rate_manager = repeater.payment_rate_manager
|
||||
self.settings = settings
|
||||
|
||||
def _set_rate(self, rate):
|
||||
|
||||
def set_info_payment_rate(rate):
|
||||
self.payment_rate_manager.min_valuable_blob_info_payment_rate = rate
|
||||
|
||||
d = self.settings.save_valuable_info_payment_rate(rate)
|
||||
d.addCallback(lambda _: set_info_payment_rate(rate))
|
||||
return d
|
||||
|
||||
def _get_current_status(self):
|
||||
status = "The current valuable blob info payment rate is "
|
||||
status += str(self.payment_rate_manager.min_valuable_blob_info_payment_rate)
|
||||
return status
|
||||
|
||||
|
||||
class ModifyInfoPaymentRateFactory(CommandHandlerFactory):
|
||||
control_handler_class = ModifyInfoPaymentRate
|
||||
|
||||
|
||||
class ModifyHashPaymentRate(ModifyPaymentRate):
|
||||
prompt_description = "Modify Blind Repeater valuable hash payment rate"
|
||||
|
||||
def __init__(self, console, repeater, settings):
|
||||
ModifyPaymentRate.__init__(self, console)
|
||||
self.payment_rate_manager = repeater.payment_rate_manager
|
||||
self.settings = settings
|
||||
|
||||
def _set_rate(self, rate):
|
||||
|
||||
def set_hash_payment_rate(rate):
|
||||
self.payment_rate_manager.min_valuable_blob_hash_payment_rate = rate
|
||||
|
||||
d = self.settings.save_valuable_hash_payment_rate(rate)
|
||||
d.addCallback(lambda _: set_hash_payment_rate(rate))
|
||||
return d
|
||||
|
||||
def _get_current_status(self):
|
||||
status = "The current valuable blob hash payment rate is "
|
||||
status += str(self.payment_rate_manager.min_valuable_blob_hash_payment_rate)
|
||||
return status
|
||||
|
||||
|
||||
class ModifyHashPaymentRateFactory(CommandHandlerFactory):
|
||||
control_handler_class = ModifyHashPaymentRate
|
||||
|
||||
|
||||
class ModifyRepeaterOptions(RecursiveCommandHandler):
|
||||
prompt_description = "Modify Blind Repeater options"
|
||||
|
||||
def __init__(self, console, repeater, lbry_session, settings):
|
||||
self.repeater = repeater
|
||||
self.lbry_session = lbry_session
|
||||
self.settings = settings
|
||||
RecursiveCommandHandler.__init__(self, console)
|
||||
|
||||
def _get_control_handler_factories(self):
|
||||
return [ModifyDataPaymentRateFactory(self.repeater, self.settings),
|
||||
ModifyInfoPaymentRateFactory(self.repeater, self.settings),
|
||||
ModifyHashPaymentRateFactory(self.repeater, self.settings),
|
||||
UpdateMaxSpaceFactory(self.repeater, self.settings),
|
||||
AddApprovedPeerFactory(self.repeater, self.lbry_session.peer_manager, self.settings),
|
||||
DeleteApprovedPeerChooserFactory(self.repeater, self.settings),
|
||||
]
|
||||
|
||||
|
||||
class ModifyRepeaterOptionsFactory(CommandHandlerFactory):
|
||||
control_handler_class = ModifyRepeaterOptions
|
||||
|
|
@ -1,106 +0,0 @@
|
|||
from twisted.internet import threads, defer
|
||||
import json
|
||||
import unqlite
|
||||
import os
|
||||
from twisted.enterprise import adbapi
|
||||
from lbrynet.core.sqlite_helpers import rerun_if_locked
|
||||
|
||||
class BlindRepeaterSettings(object):
|
||||
|
||||
def __init__(self, db_dir):
|
||||
self.db_dir = db_dir
|
||||
self.unq_db = None
|
||||
self.sql_db = None
|
||||
|
||||
def setup(self):
|
||||
self.unq_db = unqlite.UnQLite(os.path.join(self.db_dir, "blind_settings.db"))
|
||||
# check_same_thread=False is solely to quiet a spurious error that appears to be due
|
||||
# to a bug in twisted, where the connection is closed by a different thread than the
|
||||
# one that opened it. The individual connections in the pool are not used in multiple
|
||||
# threads.
|
||||
self.sql_db = adbapi.ConnectionPool('sqlite3', os.path.join(self.db_dir, "blind_peers.db"),
|
||||
check_same_thread=False)
|
||||
|
||||
return self.sql_db.runQuery("create table if not exists approved_peers (" +
|
||||
" ip_address text, " +
|
||||
" port integer" +
|
||||
")")
|
||||
|
||||
def stop(self):
|
||||
self.unq_db = None
|
||||
self.sql_db = None
|
||||
return defer.succeed(True)
|
||||
|
||||
def save_repeater_status(self, running):
|
||||
def save_status():
|
||||
self.unq_db["running"] = json.dumps(running)
|
||||
|
||||
return threads.deferToThread(save_status)
|
||||
|
||||
def get_repeater_saved_status(self):
|
||||
def get_status():
|
||||
if "running" in self.unq_db:
|
||||
return json.loads(self.unq_db['running'])
|
||||
else:
|
||||
return False
|
||||
|
||||
return threads.deferToThread(get_status)
|
||||
|
||||
def save_max_space(self, max_space):
|
||||
def save_space():
|
||||
self.unq_db['max_space'] = json.dumps(max_space)
|
||||
|
||||
return threads.deferToThread(save_space)
|
||||
|
||||
def get_saved_max_space(self):
|
||||
def get_space():
|
||||
if 'max_space' in self.unq_db:
|
||||
return json.loads(self.unq_db['max_space'])
|
||||
else:
|
||||
return 0
|
||||
|
||||
return threads.deferToThread(get_space)
|
||||
|
||||
@rerun_if_locked
|
||||
def save_approved_peer(self, host, port):
|
||||
return self.sql_db.runQuery("insert into approved_peers values (?, ?)",
|
||||
(host, port))
|
||||
|
||||
@rerun_if_locked
|
||||
def remove_approved_peer(self, host, port):
|
||||
return self.sql_db.runQuery("delete from approved_peers where ip_address = ? and port = ?",
|
||||
(host, port))
|
||||
|
||||
@rerun_if_locked
|
||||
def get_approved_peers(self):
|
||||
return self.sql_db.runQuery("select * from approved_peers")
|
||||
|
||||
def get_data_payment_rate(self):
|
||||
return threads.deferToThread(self._get_rate, "data_payment_rate")
|
||||
|
||||
def save_data_payment_rate(self, rate):
|
||||
return threads.deferToThread(self._save_rate, "data_payment_rate", rate)
|
||||
|
||||
def get_valuable_info_payment_rate(self):
|
||||
return threads.deferToThread(self._get_rate, "valuable_info_rate")
|
||||
|
||||
def save_valuable_info_payment_rate(self, rate):
|
||||
return threads.deferToThread(self._save_rate, "valuable_info_rate", rate)
|
||||
|
||||
def get_valuable_hash_payment_rate(self):
|
||||
return threads.deferToThread(self._get_rate, "valuable_hash_rate")
|
||||
|
||||
def save_valuable_hash_payment_rate(self, rate):
|
||||
return threads.deferToThread(self._save_rate, "valuable_hash_rate", rate)
|
||||
|
||||
def _get_rate(self, rate_type):
|
||||
if rate_type in self.unq_db:
|
||||
return json.loads(self.unq_db[rate_type])
|
||||
else:
|
||||
return None
|
||||
|
||||
def _save_rate(self, rate_type, rate):
|
||||
if rate is not None:
|
||||
self.unq_db[rate_type] = json.dumps(rate)
|
||||
elif rate_type in self.unq_db:
|
||||
del self.unq_db[rate_type]
|
||||
|
|
@ -1,20 +0,0 @@
|
|||
from lbrynet.core.PaymentRateManager import PaymentRateManager
|
||||
|
||||
|
||||
class BlindRepeaterPaymentRateManager(PaymentRateManager):
|
||||
def __init__(self, base, valuable_info_rate, valuable_hash_rate, blob_data_rate=None):
|
||||
PaymentRateManager.__init__(self, base, blob_data_rate)
|
||||
self.min_valuable_blob_info_payment_rate = valuable_info_rate
|
||||
self.min_valuable_blob_hash_payment_rate = valuable_hash_rate
|
||||
|
||||
def get_rate_valuable_blob_info(self, peer):
|
||||
return self.min_valuable_blob_info_payment_rate
|
||||
|
||||
def accept_rate_valuable_blob_info(self, peer, payment_rate):
|
||||
return payment_rate >= self.min_valuable_blob_info_payment_rate
|
||||
|
||||
def get_rate_valuable_blob_hash(self, peer):
|
||||
return self.min_valuable_blob_hash_payment_rate
|
||||
|
||||
def accept_rate_valuable_blob_hash(self, peer, payment_rate):
|
||||
return payment_rate >= self.min_valuable_blob_hash_payment_rate
|
||||
|
|
@ -1,9 +0,0 @@
|
|||
from lbrynet.core.BlobInfo import BlobInfo
|
||||
|
||||
|
||||
class ValuableBlobInfo(BlobInfo):
|
||||
def __init__(self, blob_hash, length, reference, peer, peer_score):
|
||||
BlobInfo.__init__(self, blob_hash, blob_hash, length)
|
||||
self.reference = reference
|
||||
self.peer = peer
|
||||
self.peer_score = peer_score
|
||||
|
|
@ -1,202 +0,0 @@
|
|||
from lbrynet.interfaces import IQueryHandlerFactory, IQueryHandler
|
||||
from zope.interface import implements
|
||||
from twisted.internet import defer
|
||||
import logging
|
||||
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class ValuableQueryHandler(object):
|
||||
implements(IQueryHandler)
|
||||
|
||||
def __init__(self, wallet, payment_rate_manager):
|
||||
self.wallet = wallet
|
||||
self.payment_rate_manager = payment_rate_manager
|
||||
self.peer = None
|
||||
self.payment_rate = None
|
||||
self.query_identifiers = []
|
||||
|
||||
######### IQueryHandler #########
|
||||
|
||||
def register_with_request_handler(self, request_handler, peer):
|
||||
self.peer = peer
|
||||
request_handler.register_query_handler(self, self.query_identifiers)
|
||||
|
||||
def handle_queries(self, queries):
|
||||
pass
|
||||
|
||||
|
||||
class ValuableBlobHashQueryHandlerFactory(object):
|
||||
implements(IQueryHandlerFactory)
|
||||
|
||||
def __init__(self, peer_finder, wallet, payment_rate_manager):
|
||||
self.peer_finder = peer_finder
|
||||
self.wallet = wallet
|
||||
self.payment_rate_manager = payment_rate_manager
|
||||
|
||||
######### IQueryHandlerFactory #########
|
||||
|
||||
def build_query_handler(self):
|
||||
q_h = ValuableBlobHashQueryHandler(self.wallet, self.payment_rate_manager, self.peer_finder)
|
||||
return q_h
|
||||
|
||||
def get_primary_query_identifier(self):
|
||||
return 'valuable_blob_hashes'
|
||||
|
||||
def get_description(self):
|
||||
return "Valuable Hashes - Hashes of blobs that it may be valuable to repeat"
|
||||
|
||||
|
||||
class ValuableBlobHashQueryHandler(ValuableQueryHandler):
|
||||
implements(IQueryHandler)
|
||||
|
||||
def __init__(self, wallet, payment_rate_manager, peer_finder):
|
||||
ValuableQueryHandler.__init__(self, wallet, payment_rate_manager)
|
||||
self.peer_finder = peer_finder
|
||||
self.query_identifiers = ['valuable_blob_hashes', 'valuable_blob_payment_rate']
|
||||
self.valuable_blob_hash_payment_rate = None
|
||||
self.blob_length_payment_rate = None
|
||||
|
||||
######### IQueryHandler #########
|
||||
|
||||
def handle_queries(self, queries):
|
||||
response = {}
|
||||
|
||||
def set_fields(fields):
|
||||
response.update(fields)
|
||||
|
||||
if self.query_identifiers[1] in queries:
|
||||
d = self._handle_valuable_blob_payment_rate(queries[self.query_identifiers[1]])
|
||||
d.addCallback(set_fields)
|
||||
else:
|
||||
d = defer.succeed(True)
|
||||
|
||||
if self.query_identifiers[0] in queries:
|
||||
d.addCallback(lambda _: self._handle_valuable_blob_hashes(queries[self.query_identifiers[0]]))
|
||||
d.addCallback(set_fields)
|
||||
|
||||
d.addCallback(lambda _: response)
|
||||
return d
|
||||
|
||||
######### internal #########
|
||||
|
||||
def _handle_valuable_blob_payment_rate(self, requested_payment_rate):
|
||||
if not self.payment_rate_manager.accept_rate_valuable_blob_hash(self.peer, "VALUABLE_BLOB_HASH",
|
||||
requested_payment_rate):
|
||||
r = "RATE_TOO_LOW"
|
||||
else:
|
||||
self.valuable_blob_hash_payment_rate = requested_payment_rate
|
||||
r = "RATE_ACCEPTED"
|
||||
return defer.succeed({'valuable_blob_payment_rate': r})
|
||||
|
||||
def _handle_valuable_blob_hashes(self, request):
|
||||
# TODO: eventually, look at the request and respond appropriately given the 'reference' field
|
||||
if self.valuable_blob_hash_payment_rate is not None:
|
||||
max_hashes = 20
|
||||
if 'max_blob_hashes' in request:
|
||||
max_hashes = int(request['max_blob_hash'])
|
||||
valuable_hashes = self.peer_finder.get_most_popular_blobs(max_hashes)
|
||||
hashes_and_scores = []
|
||||
for blob_hash, count in valuable_hashes:
|
||||
hashes_and_scores.append((blob_hash, 1.0 * count / 10.0))
|
||||
if len(hashes_and_scores) != 0:
|
||||
log.info("Responding to a valuable blob hashes request with %s blob hashes",
|
||||
str(len(hashes_and_scores)))
|
||||
expected_payment = 1.0 * len(hashes_and_scores) * self.valuable_blob_hash_payment_rate / 1000.0
|
||||
self.wallet.add_expected_payment(self.peer, expected_payment)
|
||||
self.peer.update_stats('uploaded_valuable_blob_hashes', len(hashes_and_scores))
|
||||
return defer.succeed({'valuable_blob_hashes': {'blob_hashes': hashes_and_scores}})
|
||||
return defer.succeed({'valuable_blob_hashes': {'error': "RATE_UNSET"}})
|
||||
|
||||
|
||||
class ValuableBlobLengthQueryHandlerFactory(object):
|
||||
implements(IQueryHandlerFactory)
|
||||
|
||||
def __init__(self, wallet, payment_rate_manager, blob_manager):
|
||||
self.blob_manager = blob_manager
|
||||
self.wallet = wallet
|
||||
self.payment_rate_manager = payment_rate_manager
|
||||
|
||||
######### IQueryHandlerFactory #########
|
||||
|
||||
def build_query_handler(self):
|
||||
q_h = ValuableBlobLengthQueryHandler(self.wallet, self.payment_rate_manager, self.blob_manager)
|
||||
return q_h
|
||||
|
||||
def get_primary_query_identifier(self):
|
||||
return 'blob_length'
|
||||
|
||||
def get_description(self):
|
||||
return "Valuable Blob Lengths - Lengths of blobs that it may be valuable to repeat"
|
||||
|
||||
|
||||
class ValuableBlobLengthQueryHandler(ValuableQueryHandler):
|
||||
|
||||
def __init__(self, wallet, payment_rate_manager, blob_manager):
|
||||
ValuableQueryHandler.__init__(self, wallet, payment_rate_manager)
|
||||
self.blob_manager = blob_manager
|
||||
self.query_identifiers = ['blob_length', 'blob_length_payment_rate']
|
||||
self.valuable_blob_hash_payment_rate = None
|
||||
self.blob_length_payment_rate = None
|
||||
|
||||
######## IQueryHandler #########
|
||||
|
||||
def handle_queries(self, queries):
|
||||
response = {}
|
||||
|
||||
def set_fields(fields):
|
||||
response.update(fields)
|
||||
|
||||
if self.query_identifiers[1] in queries:
|
||||
d = self._handle_blob_length_payment_rate(queries[self.query_identifiers[1]])
|
||||
d.addCallback(set_fields)
|
||||
else:
|
||||
d = defer.succeed(True)
|
||||
|
||||
if self.query_identifiers[0] in queries:
|
||||
d.addCallback(lambda _: self._handle_blob_length(queries[self.query_identifiers[0]]))
|
||||
d.addCallback(set_fields)
|
||||
|
||||
d.addCallback(lambda _: response)
|
||||
return d
|
||||
|
||||
######### internal #########
|
||||
|
||||
def _handle_blob_length_payment_rate(self, requested_payment_rate):
|
||||
if not self.payment_rate_manager.accept_rate_valuable_blob_info(self.peer, "VALUABLE_BLOB_INFO",
|
||||
requested_payment_rate):
|
||||
r = "RATE_TOO_LOW"
|
||||
else:
|
||||
self.blob_length_payment_rate = requested_payment_rate
|
||||
r = "RATE_ACCEPTED"
|
||||
return defer.succeed({'blob_length_payment_rate': r})
|
||||
|
||||
def _handle_blob_length(self, request):
|
||||
if self.blob_length_payment_rate is not None:
|
||||
assert 'blob_hashes' in request
|
||||
ds = []
|
||||
|
||||
def make_response_pair(length, blob_hash):
|
||||
return blob_hash, length
|
||||
|
||||
for blob_hash in request['blob_hashes']:
|
||||
d = self.blob_manager.get_blob_length(blob_hash)
|
||||
d.addCallback(make_response_pair, blob_hash)
|
||||
ds.append(d)
|
||||
|
||||
dl = defer.DeferredList(ds)
|
||||
|
||||
def make_response(response_pairs):
|
||||
lengths = []
|
||||
for success, response_pair in response_pairs:
|
||||
if success is True:
|
||||
lengths.append(response_pair)
|
||||
if len(lengths) > 0:
|
||||
log.info("Responding with %s blob lengths", str(len(lengths)))
|
||||
expected_payment = 1.0 * len(lengths) * self.blob_length_payment_rate / 1000.0
|
||||
self.wallet.add_expected_payment(self.peer, expected_payment)
|
||||
self.peer.update_stats('uploaded_valuable_blob_infos', len(lengths))
|
||||
return {'blob_length': {'blob_lengths': lengths}}
|
||||
|
||||
dl.addCallback(make_response)
|
||||
|
|
@ -1,130 +0,0 @@
|
|||
from lbrynet.lbrynet_console import Plugin
|
||||
from twisted.internet import defer
|
||||
from lbrynet.conf import settings
|
||||
from BlindRepeater import BlindRepeater
|
||||
from BlindInfoManager import BlindInfoManager
|
||||
from BlindRepeaterSettings import BlindRepeaterSettings
|
||||
from BlindRepeaterControlHandlers import StartRepeaterFactory, StopRepeaterFactory, UpdateMaxSpaceFactory
|
||||
from BlindRepeaterControlHandlers import AddApprovedPeerFactory, DeleteApprovedPeerFactory, RepeaterStatusFactory
|
||||
from BlindRepeaterControlHandlers import ShowApprovedPeersFactory, ModifyRepeaterOptionsFactory
|
||||
from ValuableBlobQueryHandler import ValuableBlobLengthQueryHandlerFactory
|
||||
from ValuableBlobQueryHandler import ValuableBlobHashQueryHandlerFactory
|
||||
|
||||
from PaymentRateManager import BlindRepeaterPaymentRateManager
|
||||
|
||||
|
||||
class BlindRepeaterPlugin(Plugin.Plugin):
|
||||
|
||||
def __init__(self):
|
||||
Plugin.Plugin.__init__(self)
|
||||
self.enabled = False
|
||||
self.blind_info_manager = None
|
||||
self.valuable_blob_length_query_handler = None
|
||||
self.valuable_blob_hash_query_handler = None
|
||||
self.repeater = None
|
||||
self.control_handlers = None
|
||||
self.payment_rate_manager = None
|
||||
self.settings = None
|
||||
|
||||
def setup(self, lbry_console):
|
||||
if not self.enabled:
|
||||
return defer.succeed(True)
|
||||
lbry_session = lbry_console.session
|
||||
d = self._setup_settings(lbry_session.db_dir)
|
||||
d.addCallback(lambda _: self._get_payment_rate_manager(lbry_session.base_payment_rate_manager))
|
||||
d.addCallback(lambda _: self._setup_blind_info_manager(lbry_session.peer_manager, lbry_session.db_dir))
|
||||
d.addCallback(lambda _: self._setup_blind_repeater(lbry_session))
|
||||
d.addCallback(lambda _: self._setup_valuable_blob_query_handler(lbry_session))
|
||||
d.addCallback(lambda _: self._create_control_handlers(lbry_session))
|
||||
d.addCallback(lambda _: self._restore_repeater_status(lbry_session))
|
||||
d.addCallback(lambda _: self._add_to_lbry_console(lbry_console))
|
||||
return d
|
||||
|
||||
def stop(self):
|
||||
if self.settings is not None:
|
||||
return self.settings.stop()
|
||||
return defer.succeed(True)
|
||||
|
||||
def _setup_settings(self, db_dir):
|
||||
self.settings = BlindRepeaterSettings(db_dir)
|
||||
return self.settings.setup()
|
||||
|
||||
def _get_payment_rate_manager(self, default_payment_rate_manager):
|
||||
d1 = self.settings.get_data_payment_rate()
|
||||
d2 = self.settings.get_valuable_info_payment_rate()
|
||||
d3 = self.settings.get_valuable_hash_payment_rate()
|
||||
|
||||
dl = defer.DeferredList([d1, d2, d3])
|
||||
|
||||
def get_payment_rate_manager(rates):
|
||||
data_rate = rates[0][1] if rates[0][0] is True else None
|
||||
info_rate = rates[1][1] if rates[1][0] is True else None
|
||||
info_rate = info_rate if info_rate is not None else settings.min_valuable_info_rate
|
||||
hash_rate = rates[2][1] if rates[2][0] is True else None
|
||||
hash_rate = hash_rate if hash_rate is not None else settings.min_valuable_hash_rate
|
||||
self.payment_rate_manager = BlindRepeaterPaymentRateManager(default_payment_rate_manager,
|
||||
info_rate, hash_rate,
|
||||
blob_data_rate=data_rate)
|
||||
|
||||
dl.addCallback(get_payment_rate_manager)
|
||||
return dl
|
||||
|
||||
def _setup_blind_info_manager(self, peer_manager, db_dir):
|
||||
self.blind_info_manager = BlindInfoManager(db_dir, peer_manager)
|
||||
return self.blind_info_manager.setup()
|
||||
|
||||
def _setup_valuable_blob_query_handler(self, lbry_session):
|
||||
self.valuable_blob_length_query_handler = ValuableBlobLengthQueryHandlerFactory(lbry_session.blob_manager,
|
||||
lbry_session.wallet,
|
||||
self.payment_rate_manager)
|
||||
self.valuable_blob_hash_query_handler = ValuableBlobHashQueryHandlerFactory(lbry_session.peer_finder,
|
||||
lbry_session.wallet,
|
||||
self.payment_rate_manager)
|
||||
|
||||
def _setup_blind_repeater(self, lbry_session):
|
||||
self.repeater = BlindRepeater(lbry_session.peer_finder, lbry_session.rate_limiter,
|
||||
lbry_session.blob_manager, self.blind_info_manager,
|
||||
lbry_session.wallet, self.payment_rate_manager)
|
||||
return self.repeater.setup()
|
||||
|
||||
def _restore_repeater_status(self, lbry_session):
|
||||
d = self.settings.get_saved_max_space()
|
||||
|
||||
def set_max_space(max_space):
|
||||
self.repeater.set_max_space(max_space)
|
||||
|
||||
d.addCallback(set_max_space)
|
||||
|
||||
d.addCallback(lambda _: self.settings.get_approved_peers())
|
||||
|
||||
def set_approved_peers(peers):
|
||||
for host, port in peers:
|
||||
peer = lbry_session.peer_manager.get_peer(host, int(port))
|
||||
self.repeater.add_approved_peer(peer)
|
||||
|
||||
d.addCallback(set_approved_peers)
|
||||
|
||||
d.addCallback(lambda _: self.settings.get_repeater_saved_status())
|
||||
|
||||
def restore_running(running):
|
||||
if running:
|
||||
return self.repeater.start()
|
||||
else:
|
||||
return defer.succeed(True)
|
||||
|
||||
d.addCallback(restore_running)
|
||||
return d
|
||||
|
||||
def _create_control_handlers(self, lbry_session):
|
||||
category = "Blind Repeater"
|
||||
control_handlers = [StartRepeaterFactory(self.repeater, self.settings),
|
||||
StopRepeaterFactory(self.repeater, self.settings),
|
||||
RepeaterStatusFactory(self.repeater),
|
||||
ShowApprovedPeersFactory(self.repeater),
|
||||
ModifyRepeaterOptionsFactory(self.repeater, lbry_session, self.settings)]
|
||||
self.control_handlers = zip([category] * len(control_handlers), control_handlers)
|
||||
|
||||
def _add_to_lbry_console(self, lbry_console):
|
||||
lbry_console.add_control_handlers(self.control_handlers)
|
||||
lbry_console.add_query_handlers([self.valuable_blob_length_query_handler,
|
||||
self.valuable_blob_hash_query_handler])
|
||||
|
|
@ -1,6 +0,0 @@
|
|||
from zope.interface import Interface
|
||||
|
||||
|
||||
class IBlobScorer(Interface):
|
||||
def score_blob(self, blob, blob_info):
|
||||
pass
|
||||
|
|
@ -1,8 +0,0 @@
|
|||
[Core]
|
||||
Name = BlindRepeater
|
||||
Module = BlindRepeater
|
||||
|
||||
[Documentation]
|
||||
Author = LBRY
|
||||
Version = 0.1
|
||||
Description = A client which blindly downloads data it judges valuable so that it can be re-uploaded for profit
|
||||
|
|
@ -32,7 +32,7 @@ from lbrynet.lbryfile.EncryptedFileMetadataManager import DBEncryptedFileMetadat
|
|||
from lbrynet.lbryfile.EncryptedFileMetadataManager import TempEncryptedFileMetadataManager
|
||||
from lbrynet.lbryfile.StreamDescriptor import EncryptedFileStreamType
|
||||
from lbrynet.lbryfilemanager.EncryptedFileManager import EncryptedFileManager
|
||||
from lbrynet.lbrynet_console.Settings import Settings
|
||||
from lbrynet.lbrynet_daemon.Settings import Settings
|
||||
from lbrynet.lbrynet_daemon.UIManager import UIManager
|
||||
from lbrynet.lbrynet_daemon.Downloader import GetStream
|
||||
from lbrynet.lbrynet_daemon.Publisher import Publisher
|
||||
|
|
|
|||
Loading…
Reference in a new issue