class TriblerSession(BaseManager):
_sconfig = None
_dispersy = None
_running = False
def _connect(self):
"""
Copies the libswift and ffmpeg binaries when on Android.
:return:
"""
if not self._connected:
self._connected = True
# Copy the swift and ffmpeg binaries
if is_android(strict=True):
binaries = ['swift', 'ffmpeg']
for binary in binaries:
_logger.info("Setting up the %s binary.." % binary)
if not self._copy_binary(binary):
_logger.error("Unable to find or copy the %s binary!" %
binary)
else:
raise RuntimeError('TriblerSession already connected')
def _xmlrpc_register(self, xmlrpc):
"""
Register the public functions in this manager with an XML-RPC Manager.
:param xmlrpc: The XML-RPC Manager it should register to.
:return: Nothing.
"""
xmlrpc.register_function(self.start_session, "tribler.start_session")
xmlrpc.register_function(self.stop_session, "tribler.stop_session")
def _copy_binary(self, binary_name):
"""
Copy a binary, such as swift, from the sdcard (which is mounted with noexec) to the ANDROID_PRIVATE folder which
does allow it. If the binary already exists, do nothing.
:param binary_name: The name of the binary that should be copied.
:return: Boolean indicating success.
"""
# We are on android, setup the swift binary!
sdcard_path = os.getcwd()
binary_source = os.path.join(sdcard_path, binary_name)
binary_dest = os.path.join(os.environ['ANDROID_PRIVATE'], binary_name)
if not os.path.exists(binary_dest):
if not os.path.exists(binary_source):
_logger.error(
"Looked at %s and %s, but couldn't find a '%s' binary!" %
(binary_source, binary_dest, binary_name))
return False
_logger.warn("Copy '%s' binary (%s -> %s)" %
(binary_name, binary_source, binary_dest))
shutil.copy2(binary_source, binary_dest)
# TODO: Set a more conservative permission
os.chmod(binary_dest, 0777)
return True
def get_session(self):
"""
Get the current Tribler session.
:return: Tribler session, or None if no session is started yet.
"""
return self._session
def start_session(self):
"""
This function loads any previous configuration files from the TRIBLER_STATE_DIR environment variable and then
starts a Tribler session.
:return: Nothing.
"""
if self._running:
return False
_logger.info("Set tribler_state_dir to %s" %
os.environ['TRIBLER_STATE_DIR'])
# Load configuration file (if exists)
cfgfilename = Session.get_default_config_filename(
os.environ['TRIBLER_STATE_DIR'])
try:
self._sconfig = SessionStartupConfig.load(cfgfilename)
_logger.info("Loaded previous configuration file from %s" %
cfgfilename)
except:
self._sconfig = SessionStartupConfig()
self._sconfig.set_state_dir(os.environ['TRIBLER_STATE_DIR'])
_logger.info(
"No previous configuration file found, creating one in %s" %
os.environ['TRIBLER_STATE_DIR'])
dlcfgfilename = get_default_dscfg_filename(
self._sconfig.get_state_dir())
_logger.debug("main: Download config %s", dlcfgfilename)
try:
defaultDLConfig = DefaultDownloadStartupConfig.load(dlcfgfilename)
except:
defaultDLConfig = DefaultDownloadStartupConfig.getInstance()
if not defaultDLConfig.get_dest_dir():
defaultDLConfig.set_dest_dir(os.environ['TRIBLER_DOWNLOAD_DIR'])
self._sconfig.set_torrent_collecting_dir(
os.path.join(os.environ['TRIBLER_DOWNLOAD_DIR'],
".%s" % STATEDIR_TORRENTCOLL_DIR))
self._sconfig.set_swift_meta_dir(
os.path.join(os.environ['TRIBLER_DOWNLOAD_DIR'],
".%s" % STATEDIR_SWIFTRESEED_DIR))
if not os.path.isdir(defaultDLConfig.get_dest_dir()):
try:
_logger.info("Creating download directory: %s" %
defaultDLConfig.get_dest_dir())
os.makedirs(defaultDLConfig.get_dest_dir())
except:
_logger.error("Couldn't create download directory! (%s)" %
defaultDLConfig.get_dest_dir())
# Disable unneeded dependencies
self._sconfig.set_torrent_checking(False)
self._sconfig.set_multicast_local_peer_discovery(False)
#self._sconfig.set_megacache(False)
#self._sconfig.set_swift_proc(False)
self._sconfig.set_mainline_dht(False)
#self._sconfig.set_torrent_collecting(False)
#self._sconfig.set_libtorrent(False)
self._sconfig.set_dht_torrent_collecting(False)
#self._sconfig.set_videoplayer(False)
self._sconfig.set_dispersy_tunnel_over_swift(False)
self._sconfig.set_torrent_collecting_max_torrents(5000)
# Start session
_logger.info("Starting tribler session")
self._session = Session(self._sconfig)
self._session.start()
self._swift = self._session.get_swift_proc(
) and self._session.get_swift_process()
self._dispersy = self._session.get_dispersy_instance()
_logger.info("libTribler session started!")
self.define_communities()
# Dispersy init communitites callback function
@call_on_reactor_thread
def define_communities(self):
"""
Load the dispersy communities. This function must be run on the Twisted reactor thread.
:return: Nothing.
"""
integrate_with_tribler = True
comm_args = {'integrate_with_tribler': integrate_with_tribler}
from Tribler.community.search.community import SearchCommunity
from Tribler.community.allchannel.community import AllChannelCommunity
from Tribler.community.channel.community import ChannelCommunity
from Tribler.community.channel.preview import PreviewChannelCommunity
from Tribler.community.metadata.community import MetadataCommunity
_logger.info("Preparing to load dispersy communities...")
comm = self._dispersy.define_auto_load(SearchCommunity,
self._session.dispersy_member,
load=True,
kargs=comm_args)
_logger.debug("Currently loaded dispersy communities: %s" % comm)
#comm = self._dispersy.define_auto_load(AllChannelCommunity, self._session.dispersy_member, load=True,
# kargs=comm_args)
#_logger.debug("Currently loaded dispersy communities: %s" % comm)
# load metadata community
comm = self._dispersy.define_auto_load(MetadataCommunity,
self._session.dispersy_member,
load=True,
kargs=comm_args)
_logger.info("@@@@@@@@@@ Loaded dispersy communities: %s" % comm)
# 17/07/13 Boudewijn: the missing-member message send by the BarterCommunity on the swift port is crashing
# 6.1 clients. We will disable the BarterCommunity for version 6.2, giving people some time to upgrade
# their version before enabling it again.
# if swift_process:
# dispersy.define_auto_load(BarterCommunity,
# s.dispersy_member,
# (swift_process,),
# load=True)
#comm = self._dispersy.define_auto_load(ChannelCommunity, self._session.dispersy_member, load=True,
# kargs=comm_args)
#_logger.debug("Currently loaded dispersy communities: %s" % comm)
#comm = self._dispersy.define_auto_load(PreviewChannelCommunity, self._session.dispersy_member, kargs=comm_args)
#_logger.debug("Currently loaded dispersy communities: %s" % comm)
self._running = True
def is_running(self):
return self._running
def stop_session(self):
"""
Unloads the Tribler session.
:return: Nothing.
"""
_logger.info("Create checkpoint..")
self._session.checkpoint()
_logger.info("Shutting down Tribler..")
self._session.shutdown()
# Just a tad of extra time
time.sleep(1)
self._running = False
_logger.info("Bye bye")
return True
class TriblerSession(BaseManager):
_sconfig = None
_dispersy = None
_running = False
def _connect(self):
"""
Copies the libswift and ffmpeg binaries when on Android.
:return:
"""
if not self._connected:
self._connected = True
# Copy the swift and ffmpeg binaries
if is_android(strict=True):
binaries = ['swift', 'ffmpeg']
for binary in binaries:
_logger.info("Setting up the %s binary.." % binary)
if not self._copy_binary(binary):
_logger.error("Unable to find or copy the %s binary!" % binary)
else:
raise RuntimeError('TriblerSession already connected')
def _xmlrpc_register(self, xmlrpc):
"""
Register the public functions in this manager with an XML-RPC Manager.
:param xmlrpc: The XML-RPC Manager it should register to.
:return: Nothing.
"""
xmlrpc.register_function(self.start_session, "tribler.start_session")
xmlrpc.register_function(self.stop_session, "tribler.stop_session")
def _copy_binary(self, binary_name):
"""
Copy a binary, such as swift, from the sdcard (which is mounted with noexec) to the ANDROID_PRIVATE folder which
does allow it. If the binary already exists, do nothing.
:param binary_name: The name of the binary that should be copied.
:return: Boolean indicating success.
"""
# We are on android, setup the swift binary!
sdcard_path = os.getcwd()
binary_source = os.path.join(sdcard_path, binary_name)
binary_dest = os.path.join(os.environ['ANDROID_PRIVATE'], binary_name)
if not os.path.exists(binary_dest):
if not os.path.exists(binary_source):
_logger.error(
"Looked at %s and %s, but couldn't find a '%s' binary!" % (binary_source, binary_dest, binary_name))
return False
_logger.warn("Copy '%s' binary (%s -> %s)" % (binary_name, binary_source, binary_dest))
shutil.copy2(binary_source, binary_dest)
# TODO: Set a more conservative permission
os.chmod(binary_dest, 0777)
return True
def get_session(self):
"""
Get the current Tribler session.
:return: Tribler session, or None if no session is started yet.
"""
return self._session
def start_session(self):
"""
This function loads any previous configuration files from the TRIBLER_STATE_DIR environment variable and then
starts a Tribler session.
:return: Nothing.
"""
if self._running:
return False
_logger.info("Set tribler_state_dir to %s" % os.environ['TRIBLER_STATE_DIR'])
# Load configuration file (if exists)
cfgfilename = Session.get_default_config_filename(os.environ['TRIBLER_STATE_DIR'])
try:
self._sconfig = SessionStartupConfig.load(cfgfilename)
_logger.info("Loaded previous configuration file from %s" % cfgfilename)
except:
self._sconfig = SessionStartupConfig()
self._sconfig.set_state_dir(os.environ['TRIBLER_STATE_DIR'])
_logger.info("No previous configuration file found, creating one in %s" % os.environ['TRIBLER_STATE_DIR'])
dlcfgfilename = get_default_dscfg_filename(self._sconfig.get_state_dir())
_logger.debug("main: Download config %s", dlcfgfilename)
try:
defaultDLConfig = DefaultDownloadStartupConfig.load(dlcfgfilename)
except:
defaultDLConfig = DefaultDownloadStartupConfig.getInstance()
if not defaultDLConfig.get_dest_dir():
defaultDLConfig.set_dest_dir(os.environ['TRIBLER_DOWNLOAD_DIR'])
self._sconfig.set_torrent_collecting_dir(os.path.join(os.environ['TRIBLER_DOWNLOAD_DIR'], ".%s" % STATEDIR_TORRENTCOLL_DIR))
self._sconfig.set_swift_meta_dir(os.path.join(os.environ['TRIBLER_DOWNLOAD_DIR'], ".%s" % STATEDIR_SWIFTRESEED_DIR))
if not os.path.isdir(defaultDLConfig.get_dest_dir()):
try:
_logger.info("Creating download directory: %s" % defaultDLConfig.get_dest_dir())
os.makedirs(defaultDLConfig.get_dest_dir())
except:
_logger.error("Couldn't create download directory! (%s)" % defaultDLConfig.get_dest_dir())
# Disable unneeded dependencies
self._sconfig.set_torrent_checking(False)
self._sconfig.set_multicast_local_peer_discovery(False)
#self._sconfig.set_megacache(False)
#self._sconfig.set_swift_proc(False)
self._sconfig.set_mainline_dht(False)
#self._sconfig.set_torrent_collecting(False)
#self._sconfig.set_libtorrent(False)
self._sconfig.set_dht_torrent_collecting(False)
#self._sconfig.set_videoplayer(False)
self._sconfig.set_dispersy_tunnel_over_swift(False)
self._sconfig.set_torrent_collecting_max_torrents(5000)
# Start session
_logger.info("Starting tribler session")
self._session = Session(self._sconfig)
self._session.start()
self._swift = self._session.get_swift_proc() and self._session.get_swift_process()
self._dispersy = self._session.get_dispersy_instance()
_logger.info("libTribler session started!")
self.define_communities()
# Dispersy init communitites callback function
@call_on_reactor_thread
def define_communities(self):
"""
Load the dispersy communities. This function must be run on the Twisted reactor thread.
:return: Nothing.
"""
integrate_with_tribler = True
comm_args = {'integrate_with_tribler': integrate_with_tribler}
from Tribler.community.search.community import SearchCommunity
from Tribler.community.allchannel.community import AllChannelCommunity
from Tribler.community.channel.community import ChannelCommunity
from Tribler.community.channel.preview import PreviewChannelCommunity
from Tribler.community.metadata.community import MetadataCommunity
_logger.info("Preparing to load dispersy communities...")
comm = self._dispersy.define_auto_load(SearchCommunity, self._session.dispersy_member, load=True,
kargs=comm_args)
_logger.debug("Currently loaded dispersy communities: %s" % comm)
#comm = self._dispersy.define_auto_load(AllChannelCommunity, self._session.dispersy_member, load=True,
# kargs=comm_args)
#_logger.debug("Currently loaded dispersy communities: %s" % comm)
# load metadata community
comm = self._dispersy.define_auto_load(MetadataCommunity, self._session.dispersy_member, load=True,
kargs=comm_args)
_logger.info("@@@@@@@@@@ Loaded dispersy communities: %s" % comm)
# 17/07/13 Boudewijn: the missing-member message send by the BarterCommunity on the swift port is crashing
# 6.1 clients. We will disable the BarterCommunity for version 6.2, giving people some time to upgrade
# their version before enabling it again.
# if swift_process:
# dispersy.define_auto_load(BarterCommunity,
# s.dispersy_member,
# (swift_process,),
# load=True)
#comm = self._dispersy.define_auto_load(ChannelCommunity, self._session.dispersy_member, load=True,
# kargs=comm_args)
#_logger.debug("Currently loaded dispersy communities: %s" % comm)
#comm = self._dispersy.define_auto_load(PreviewChannelCommunity, self._session.dispersy_member, kargs=comm_args)
#_logger.debug("Currently loaded dispersy communities: %s" % comm)
self._running = True
def is_running(self):
return self._running
def stop_session(self):
"""
Unloads the Tribler session.
:return: Nothing.
"""
_logger.info("Create checkpoint..")
self._session.checkpoint()
_logger.info("Shutting down Tribler..")
self._session.shutdown()
# Just a tad of extra time
time.sleep(1)
self._running = False
_logger.info("Bye bye")
return True
