class FileManagerTestCase(AbstractTorrentTestCase):
def setUp(self):
self.manager = FileManager()
self.configure_session()
def test_get_torrent_handler_magnet(self):
t_obj = self.manager.get_torrent_handler("Test",
self.session, MAGNET)
self.assertIsInstance(t_obj, Torrent)
self.assertEqual(t_obj.source_type, "MAGNET")
def test_get_torrent_handler_torrent(self):
server = HTTPServer(('0.0.0.0', 8567), TestSimpleHTTPRequestHandler)
th = Thread(target=server.serve_forever)
th.daemon = True
th.start()
t_obj = self.manager.get_torrent_handler("Test",
self.session, None, "http://127.0.0.1:8567/movie.torrent")
self.assertIsNone(t_obj.source)
timeout = 0
while t_obj.source is None:
time.sleep(0.1)
timeout += 1
if timeout == 30:
raise AssertionError
self.assertIsNotNone(t_obj.source)
self.assertIsInstance(t_obj, Torrent)
self.assertEqual(t_obj.source_type, "TORRENT")
class FileManagerTestCase(AbstractTorrentTestCase):
def setUp(self):
self.manager = FileManager()
self.configure_session()
def test_get_torrent_handler_magnet(self):
t_obj = self.manager.get_torrent_handler("Test", self.session, MAGNET)
self.assertIsInstance(t_obj, Torrent)
self.assertEqual(t_obj.source_type, "MAGNET")
def test_get_torrent_handler_torrent(self):
server = HTTPServer(('0.0.0.0', 8567), TestSimpleHTTPRequestHandler)
th = Thread(target=server.serve_forever)
th.daemon = True
th.start()
t_obj = self.manager.get_torrent_handler(
"Test", self.session, None, "http://127.0.0.1:8567/movie.torrent")
self.assertIsNone(t_obj.source)
timeout = 0
while t_obj.source is None:
time.sleep(0.1)
timeout += 1
if timeout == 30:
raise AssertionError
self.assertIsNotNone(t_obj.source)
self.assertIsInstance(t_obj, Torrent)
self.assertEqual(t_obj.source_type, "TORRENT")
def __init__(self):
self.manager = FileManager()
self.services = [
LegitTorrentsService(),
KickAssService(),
TPBService()
]
# self.services = [LegitTorrentsService()]
self._init_session()
threading.Thread(target=self._prefill_cache).start()
self._playing = None
def __init__(self):
self.manager = FileManager()
self.services = [LegitTorrentsService(), KickAssService(), TPBService()]
# self.services = [LegitTorrentsService()]
self._init_session()
threading.Thread(target=self._prefill_cache).start()
self._playing = None
class P2CDaemon(object):
def __init__(self):
self.manager = FileManager()
self.services = [LegitTorrentsService(), KickAssService(), TPBService()]
# self.services = [LegitTorrentsService()]
self._init_session()
threading.Thread(target=self._prefill_cache).start()
self._playing = None
def __del__(self):
self._destroy_session()
@property
def session(self):
if not self._session:
raise SessionNotBindedException
return self._session
def buffer(self, movie:Movie):
self.manager.prioritize_torrents(movie)
def is_playing(self, movie:Movie):
return self._playing == movie
def play(self, movie:Movie):
self._playing = movie
self.manager.prioritize_torrents(movie)
def get_categories(self) -> list:
output = []
[output.extend(service.get_categories()) for service in self.services]
return output
def search(self, query) -> list:
output = []
[output.extend(service.search(query)) for service in self.services]
return output
def get_torrent(self, torrent: TorrentInfo) -> Torrent:
return self.manager.get_torrent_handler(
torrent.label, self.session, torrent.get_magnet(),
torrent.get_torrent_file())
def _init_session(self):
dht_state = self._get_storage_value("dht_state")
if dht_state:
ses = lt.session(dht_state)
else:
ses = lt.session()
ses.listen_on(settings.START_PORT, settings.END_PORT)
self._session = ses
def _destroy_session(self):
ses = self.session
state = None
try:
state = ses.dht_state()
except:
logger.warning("""
UnicodeEncodeError raised by dht_state\nsee
http://code.google.com/p/libtorrent/issues/detail?id=449
for more details
""")
if state:
self._save_storage_value("dht_state", state)
def _get_storage_value(self, key):
data = self._get_storage_data()
return data.get(key, None
)
def _save_storage_value(self, key, value):
data = self._get_storage_data()
data[key] = value
f = open(settings.STORAGE_PATH, "w")
json.dump(data, f)
f.close()
def _get_storage_data(self):
f = None
try:
f = open(settings.STORAGE_PATH, "r")
except IOError:
logger.info("storage file does not exists.")
data = {}
if f:
try:
data = json.load(f)
except TypeError:
logger.error("storage file is corrupted. Overwriting.")
if isinstance(data, dict):
logger.error("storage data is not a dict. Overwriting.")
data = {}
return data
def _prefill_cache(self):
for category in self.get_categories()[:20]:
category.get_torrents()
def setUp(self):
self.manager = FileManager()
self.configure_session()
class P2CDaemon(object):
def __init__(self):
self.manager = FileManager()
self.services = [
LegitTorrentsService(),
KickAssService(),
TPBService()
]
# self.services = [LegitTorrentsService()]
self._init_session()
threading.Thread(target=self._prefill_cache).start()
self._playing = None
def __del__(self):
self._destroy_session()
@property
def session(self):
if not self._session:
raise SessionNotBindedException
return self._session
def buffer(self, movie: Movie):
self.manager.prioritize_torrents(movie)
def is_playing(self, movie: Movie):
return self._playing == movie
def play(self, movie: Movie):
self._playing = movie
self.manager.prioritize_torrents(movie)
def get_categories(self) -> list:
output = []
[output.extend(service.get_categories()) for service in self.services]
return output
def search(self, query) -> list:
output = []
[output.extend(service.search(query)) for service in self.services]
return output
def get_torrent(self, torrent: TorrentInfo) -> Torrent:
return self.manager.get_torrent_handler(torrent.label, self.session,
torrent.get_magnet(),
torrent.get_torrent_file())
def _init_session(self):
dht_state = self._get_storage_value("dht_state")
if dht_state:
ses = lt.session(dht_state)
else:
ses = lt.session()
ses.listen_on(settings.START_PORT, settings.END_PORT)
self._session = ses
def _destroy_session(self):
ses = self.session
state = None
try:
state = ses.dht_state()
except:
logger.warning("""
UnicodeEncodeError raised by dht_state\nsee
http://code.google.com/p/libtorrent/issues/detail?id=449
for more details
""")
if state:
self._save_storage_value("dht_state", state)
def _get_storage_value(self, key):
data = self._get_storage_data()
return data.get(key, None)
def _save_storage_value(self, key, value):
data = self._get_storage_data()
data[key] = value
f = open(settings.STORAGE_PATH, "w")
json.dump(data, f)
f.close()
def _get_storage_data(self):
f = None
try:
f = open(settings.STORAGE_PATH, "r")
except IOError:
logger.info("storage file does not exists.")
data = {}
if f:
try:
data = json.load(f)
except TypeError:
logger.error("storage file is corrupted. Overwriting.")
if isinstance(data, dict):
logger.error("storage data is not a dict. Overwriting.")
data = {}
return data
def _prefill_cache(self):
for category in self.get_categories()[:20]:
category.get_torrents()
def setUp(self):
self.manager = FileManager()
self.configure_session()