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: apply suggestions

Browse source
This commit is contained in:
Zamitto 2024-07-03 16:10:23 -03:00
parent ae3daa4c79
commit 153291f89f
10 changed files with 32 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

5
torrent-client/downloader.py
View file

@ -30,12 +30,13 @@ class Downloader:
self.torrent_handles[game_id] = None
self.downloading_game_id = -1
def cancel_all_downloads(self):
def abort_session(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.session.abort()
self.torrent_handles = {}
self.downloading_game_id = -1

7
torrent-client/main.py
View file

@ -8,11 +8,12 @@ from downloader import Downloader
torrent_port = sys.argv[1]
http_port = sys.argv[2]
rpc_password = sys.argv[3]
start_download_payload = sys.argv[4]
downloader = None
if sys.argv[4]:
initial_download = json.loads(urllib.parse.unquote(sys.argv[4]))
if start_download_payload:
initial_download = json.loads(urllib.parse.unquote(start_download_payload))
downloader = Downloader(torrent_port)
downloader.start_download(initial_download['game_id'], initial_download['magnet'], initial_download['save_path'])
@ -75,7 +76,7 @@ class Handler(BaseHTTPRequestHandler):
elif data['action'] == 'cancel':
downloader.cancel_download(data['game_id'])
elif data['action'] == 'kill-torrent':
downloader.cancel_all_downloads()
downloader.abort_session()
downloader = None
self.send_response(200)