PublicMirrors/hydra
1
0
Fork 0
mirror of https://github.com/hydralauncher/hydra.git synced 2025-02-03 08:43:48 +03:00
Code Issues Actions 1 Packages Projects Releases 1 Wiki Activity

feat: return seeding status in downloader

Browse Source
This commit is contained in:
Hachi-R 2024-11-05 17:10:25 -03:00
parent 3a3c0a2b35
commit 5daad057e7
1 changed files with 41 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
torrent-client/torrent_downloader.py
View File

@ -140,12 +140,18 @@ class TorrentDownloader:
if self.downloading_game_id == -1: if self.downloading_game_id == -1:
return None return None
torrent_handle = self.torrent_handles.get(self.downloading_game_id)
status = torrent_handle.status()
info = torrent_handle.get_torrent_info()
response = { response = {
'downloading': {},
'seeding': []
}
if self.downloading_game_id != -1:
torrent_handle = self.torrent_handles.get(self.downloading_game_id)
if torrent_handle:
status = torrent_handle.status()
info = torrent_handle.torrent_file()
if status.progress < 1.0:
response['downloading'] = {
'folderName': info.name() if info else "", 'folderName': info.name() if info else "",
'fileSize': info.total_size() if info else 0, 'fileSize': info.total_size() if info else 0,
'gameId': self.downloading_game_id, 'gameId': self.downloading_game_id,
@ -158,9 +164,24 @@ class TorrentDownloader:
'bytesDownloaded': status.progress * info.total_size() if info else status.all_time_download, 'bytesDownloaded': status.progress * info.total_size() if info else status.all_time_download,
} }
# if status.progress == 1: for game_id, torrent_handle in self.torrent_handles.items():
# torrent_handle.pause() if torrent_handle:
# self.session.remove_torrent(torrent_handle) status = torrent_handle.status()
# self.downloading_game_id = -1 if status.state == 5:
info = torrent_handle.torrent_file()
seed_info = {
'folderName': info.name() if info else "",
'fileSize': info.total_size() if info else 0,
'gameId': self.downloading_game_id,
'progress': status.progress,
'downloadSpeed': status.download_rate,
'uploadSpeed': status.upload_rate,
'numPeers': status.num_peers,
'numSeeds': status.num_seeds,
'status': status.state,
'bytesDownloaded': status.progress * info.total_size() if info else status.all_time_download,
}
return response response['seeding'].append(seed_info)
print(response)