#!/usr/bin/python3
from sys import path, argv
path.append("..")
from travnik import Torrent
import inotify.adapters
from threading import Thread
from random import randrange
from os import stat, path
from time import sleep
from pydub import AudioSegment
from pydub.playback import play
from socket import socket, AF_UNIX, AF_INET, SOCK_STREAM
import libtorrent
def random_file(torrent, save_path, endung=".flac"):
def get_index_from_endung(chosen):
for i in range(torrent.torrent_file().files().num_files()):
if (torrent.torrent_file().files().file_path(i).endswith(endung)):
if chosen == 0:
return i
break
chosen -= 1
prios = []
for i in torrent.get_file_priorities():
prios.append(0)
endungs = 0
for i in range(torrent.torrent_file().files().num_files()):
if (torrent.torrent_file().files().file_path(i).endswith(endung)):
endungs += 1
if endungs == 0:
print("no more endungs")
return False
chosen = randrange(endungs)
tries_left = endungs
print(f"there are {endungs} {endung} files")
while True:
if tries_left == 0:
print("no more tries")
return False
tries_left -= 1
try:
p = save_path + "/" + torrent.torrent_file().files().file_path(get_index_from_endung(chosen))
print(f"trying path {p} with stat")
stat(p)
except FileNotFoundError:
break
else:
chosen = (chosen + 1) % endungs;
continue
prios[get_index_from_endung(chosen)] = 4
torrent.prioritize_files(prios)
print(f"random file {torrent.torrent_file().files().file_path(get_index_from_endung(chosen))} chosen to download")
return True
def watcher(session, torrents, s, d, active_torrents=10):
i = inotify.adapters.Inotify()
i.add_watch(d)
for event in i.event_gen(yield_nones=False):
(_, type_names, path, filename) = event
if "IN_CLOSE_WRITE" in type_names and filename.endswith(".torrent"):
t = Torrent()
t.file(path + filename)
for pth, size in t.paths():
if pth[-1].endswith(b'.flac'):
info = libtorrent.torrent_info(path + filename)
torr = session.add_torrent({"ti": info, "save_path": s})
random_file(torr, save_path=s);
torrents.append(torr)
if len(torrents) > active_torrents:
session.remove_torrent(torrents[0])
del torrents[0]
break
def alerts(session, playstack, save_path, torrents):
while True:
session.wait_for_alert(86400*1000*21)
for a in session.pop_alerts():
if a.category() & libtorrent.alert.category_t.error_notification:
print(f"error notification alert from libtorrent: {a}")
if type(a) == libtorrent.file_completed_alert:
if a.handle.get_file_priorities()[a.index] != 4:
continue
playstack.append(a.handle.torrent_file().files().file_path(a.index))
if not random_file(a.handle, save_path):
session.remove_torrent(a.handle)
torrents.remove(a.handle)
continue
else:
a.handle.resume()
if type(a) in [libtorrent.external_ip_alert, libtorrent.listen_succeeded_alert, libtorrent.dht_get_peers_alert, libtorrent.dht_outgoing_get_peers_alert, libtorrent.dht_reply_alert, libtorrent.torrent_added_alert, libtorrent.state_changed_alert, libtorrent.torrent_resumed_alert, libtorrent.stats_alert, libtorrent.torrent_checked_alert, libtorrent.peer_connect_alert, libtorrent.alert, libtorrent.peer_disconnected_alert, libtorrent.incoming_connection_alert, libtorrent.block_downloading_alert, libtorrent.dht_bootstrap_alert, libtorrent.dht_announce_alert, libtorrent.block_finished_alert, libtorrent.piece_finished_alert, libtorrent.peer_unsnubbed_alert, libtorrent.peer_snubbed_alert, libtorrent.portmap_error_alert, libtorrent.block_timeout_alert, libtorrent.block_uploaded_alert, libtorrent.performance_alert, libtorrent.cache_flushed_alert, libtorrent.dht_log_alert, libtorrent.portmap_log_alert, libtorrent.dht_pkt_alert, libtorrent.log_alert, libtorrent.peer_log_alert]:
continue
print(f"{type(a)} | {a}")
def stream(playstack, unix=None):
if type(unix) is str:
while True:
# sock = socket(AF_INET, SOCK_STREAM)
# sock.connect("tranzistor", 1234)
sock = socket(AF_UNIX, SOCK_STREAM)
sock.connect("/tmp/ls")
sock.sendall(b"request.alive\r\n")
sleep(1)
data = sock.recv(256)
a = data.split(b"\r")[0]
if a == b"":
sock.sendall(b"request.push " + bytes(playstack.pop(), encoding="utf-8"))
sleep(1)
else:
while True:
if len(playstack) == 0:
sleep(1)
continue
song = AudioSegment.from_file(path.abspath(playstack.pop()), "flac")
play(song)
sleep(1)
def progress(session, torrents):
while True:
for torr in torrents:
s = torr.status()
print(torr.torrent_file().name() + '\t%.2f%% complete (down: %.1f kB/s up: %.1f kB/s peers: %d) %s' % (s.progress * 100, s.download_rate / 1000, s.upload_rate / 1000, s.num_peers, s.state))
sleep(1)
if __name__ == "__main__":
session = libtorrent.session({"alert_mask": libtorrent.alert.category_t.all_categories});
torrents = []
playstack = []
save_path = "."
torrents_path = "."
unix = None
if len(argv) > 1:
save_path = argv[1]
if len(argv) > 2:
torrents_path = argv[2]
if len(argv) > 3:
unix = argv[3]
w = Thread(target=watcher, args=(session, torrents, save_path, torrents_path))
a = Thread(target=alerts, args=(session, playstack, save_path, torrents))
p = Thread(target=progress, args=(session, torrents))
s = Thread(target=stream, args=(playstack, unix))
w.start()
a.start()
p.start()
s.start()
w.join()
a.join()
p.join()
s.join()
