codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add speed

Browse Source
This commit is contained in:
jxxghp 2023-07-02 16:46:22 +08:00
parent d2152c4341
commit 31b0d061b4
3 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/modules/qbittorrent/__init__.py
View File

@ -140,7 +140,9 @@ class QbittorrentModule(_ModuleBase):
year=meta.year, year=meta.year,
season_episode=meta.season_episode, season_episode=meta.season_episode,
progress=torrent.get('progress'), progress=torrent.get('progress'),
size=torrent.get('total_size') size=torrent.get('total_size'),
dlspeed=StringUtils.str_filesize(torrent.get('dlspeed')),
upspeed=StringUtils.str_filesize(torrent.get('upspeed')),
)) ))
else: else:
return None return None

5
app/modules/transmission/__init__.py
View File

@ -8,6 +8,7 @@ from app.modules import _ModuleBase
from app.modules.transmission.transmission import Transmission from app.modules.transmission.transmission import Transmission
from app.schemas import TransferInfo, TransferTorrent, DownloadingTorrent from app.schemas import TransferInfo, TransferTorrent, DownloadingTorrent
from app.schemas.types import TorrentStatus from app.schemas.types import TorrentStatus
from app.utils.string import StringUtils
class TransmissionModule(_ModuleBase): class TransmissionModule(_ModuleBase):
@ -122,7 +123,9 @@ class TransmissionModule(_ModuleBase):
year=meta.year, year=meta.year,
season_episode=meta.season_episode, season_episode=meta.season_episode,
progress=torrent.progress, progress=torrent.progress,
size=torrent.total_size size=torrent.total_size,
dlspeed=StringUtils.str_filesize(torrent.download_speed),
ulspeed=StringUtils.str_filesize(torrent.upload_speed),
)) ))
else: else:
return None return None

2
app/schemas/context.py
View File

@ -189,6 +189,8 @@ class DownloadingTorrent(BaseModel):
size: Optional[float] = 0 size: Optional[float] = 0
progress: Optional[float] = 0 progress: Optional[float] = 0
state: Optional[str] = 'downloading' state: Optional[str] = 'downloading'
upspeed: Optional[str] = None
dlspeed: Optional[str] = None
media: Optional[dict] = {} media: Optional[dict] = {}