external/hydra
1
0
Fork 0
mirror of https://github.com/hydralauncher/hydra.git synced 2025-03-09 15:40:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: crete kill-torrent

Browse source
This commit is contained in:
Zamitto 2024-07-03 12:06:26 -03:00
parent 75c8f69e81
commit 0f5db4f34e
5 changed files with 36 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

9
torrent-client/downloader.py
View file

@ -30,6 +30,15 @@ class Downloader:
self.torrent_handles[game_id] = None
self.downloading_game_id = -1
def cancel_all_downloads(self):
for game_id in self.torrent_handles:
torrent_handle = self.torrent_handles[game_id]
torrent_handle.pause()
self.session.remove_torrent(torrent_handle)
self.torrent_handles = {}
self.downloading_game_id = -1
def get_download_status(self):
if self.downloading_game_id == -1:
return None

14
torrent-client/main.py
View file

@ -33,23 +33,28 @@ class Handler(BaseHTTPRequestHandler):
status = downloader.get_download_status()
self.wfile.write(json.dumps(status).encode('utf-8'))
if self.path == "/healthcheck":
elif self.path == "/healthcheck":
self.send_response(200)
self.end_headers()
if self.path == "/process-list":
elif self.path == "/process-list":
if self.headers.get(self.rpc_password_header) != rpc_password:
self.send_response(401)
self.end_headers()
return
process_path = list(set([proc.info["exe"] for proc in psutil.process_iter(['exe'])]))
self.send_response(200)
self.send_header("Content-type", "application/json")
self.end_headers()
self.wfile.write(json.dumps(process_path).encode('utf-8'))
def do_POST(self):
global downloader
if self.path == "/action":
if self.headers.get(self.rpc_password_header) != rpc_password:
self.send_response(401)
@ -69,7 +74,10 @@ class Handler(BaseHTTPRequestHandler):
downloader.pause_download(data['game_id'])
elif data['action'] == 'cancel':
downloader.cancel_download(data['game_id'])
elif data['action'] == 'kill-torrent':
downloader.cancel_all_downloads()
downloader = None
self.send_response(200)
self.end_headers()