admin-scripts/list_torrents.py
2024-11-05 16:46:23 -06:00

94 lines
2.7 KiB
Python
Executable File

#!/home/jas/virtualenvs/list_torrents/bin/python3
"""list_torrents.py
Description:
Fetch a list of torrents from a qBittorrent instance running on localhost.
The qBittorrent instance must be configured to allow login on localhost
without authentication. The output is formatted into a plaintext table.
Usage:
list_torrents.py
list_torrents.py -h
Options:
-h, --help show this help message and exit
"""
import qbittorrentapi
from docopt import docopt
from tabulate import tabulate
try:
with qbittorrentapi.Client(
host="localhost", port=8080, username="", password=""
) as qbt_client:
qbt_client.auth_log_in()
except qbittorrentapi.LoginFailed as e:
print(e)
# convert byte units
def human_bytes(input_bytes: int) -> str:
B = float(input_bytes)
KiB = float(1024)
MiB = float(KiB**2)
GiB = float(KiB**3)
TiB = float(KiB**4)
match B:
case B if B < KiB:
return "{0} {1}".format(B, "bytes" if 0 == B > 1 else "byte")
case B if KiB <= B <= MiB:
return "{0:.2f} KiB".format(B / KiB)
case B if MiB <= B <= GiB:
return "{0:.2f} MiB".format(B / MiB)
case B if GiB <= B <= TiB:
return "{0:.2f} GiB".format(B / GiB)
case B if TiB <= B:
return "{0:.2f} TiB".format(B / TiB)
case _:
return ""
def print_table():
table = []
headers = ["Name", "Total Size", "Trackers Count", "Ratio", "Uploaded"]
with qbittorrentapi.Client(
host="localhost", port=8080, username="", password=""
) as qbt_client:
sorted_torrents = sorted(
qbt_client.torrents_info(), key=lambda d: d.ratio, reverse=True
)
for torrent in sorted_torrents:
row = []
row.append(torrent.name)
row.append(human_bytes(torrent.total_size))
row.append(torrent.trackers_count)
row.append(torrent.ratio)
row.append(human_bytes(torrent.uploaded))
table.append(row)
print(tabulate(table, headers, tablefmt="grid"))
def print_ssv():
with qbittorrentapi.Client(
host="localhost", port=8080, username="", password=""
) as qbt_client:
sorted_torrents = sorted(
qbt_client.torrents_info(), key=lambda d: d.ratio, reverse=True
)
for torrent in sorted_torrents:
name = torrent.name
size = human_bytes(torrent.total_size)
trackers = torrent.trackers_count
ratio = torrent.ratio
uploaded = human_bytes(torrent.uploaded)
print(f"{name} {size} {trackers} {ratio} {uploaded}")
if __name__ == "__main__":
args = docopt(__doc__) # type: ignore
print_ssv()