diff options
| -rwxr-xr-x | radio/radio.py | 64 |
1 files changed, 55 insertions, 9 deletions
diff --git a/radio/radio.py b/radio/radio.py index 2c04056..d562627 100755 --- a/radio/radio.py +++ b/radio/radio.py @@ -5,8 +5,11 @@ from travnik import Torrent import inotify.adapters from threading import Thread from random import randrange -from os import stat +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 @@ -26,25 +29,31 @@ def random_file(torrent, save_path, endung=".flac"): 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: - stat(save_path + "/" + torrent.torrent_file().files().file_path(get_index_from_endung(chosen))) + 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: - chosen = (chosen + 1) % endungs; 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, d="/tmp/", s=".", active_torrents=10): +def watcher(session, torrents, s, d="/tmp/", active_torrents=10): i = inotify.adapters.Inotify() i.add_watch(d) for event in i.event_gen(yield_nones=False): @@ -63,33 +72,70 @@ def watcher(session, torrents, d="/tmp/", s=".", active_torrents=10): del torrents[0] break -def alerts(session): +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: - print(f"file completed alert: {a}") - 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]: + 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]: continue print(f"{type(a)} | {a}") +def stream(playstack): + if False: + 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 = [] - w = Thread(target=watcher, args=(session, torrents)) - a = Thread(target=alerts, args=(session,)) + playstack = [] + save_path = "." + w = Thread(target=watcher, args=(session, torrents, save_path)) + a = Thread(target=alerts, args=(session, playstack, save_path, torrents)) p = Thread(target=progress, args=(session, torrents)) + s = Thread(target=stream, args=(playstack,)) w.start() a.start() p.start() + s.start() w.join() a.join() p.join() + s.join() |