Add update_tracker.py

This commit is contained in:
Jeffrey Serio 2024-08-05 20:14:16 -05:00
parent b1345934ef
commit d5c186a389

97
bin/update_tracker.py Executable file
View File

@ -0,0 +1,97 @@
#!/usr/bin/env python3
"""update_tracker.py
Description:
This script collects infohashes of all torrents in each qBittorrent instance,
updates opentracker, and reannounces all torrents to their trackers.
Expectations:
- A JSON qBittorrent authentication file at ~/.config/qbittorrent_auth.json
- SSH pubkey access to torrent tracker server root@bttracker.nirn.quest
- rsync installed on the host system running this script
Usage:
update_tracker.py
update_tracker.py -h
Options:
-h, --help show this help message and exit
"""
import json
import subprocess
import tempfile
from pathlib import Path
from docopt import docopt
from qbittorrent import Client
if __name__ == "__main__":
args = docopt(__doc__) # type: ignore
# JSON file containing authentication info for each qBittorrent instance
QBITTORRENT_AUTH_FILE = Path.home().joinpath(".config/qbittorrent_auth.json")
# Open authentication file and load JSON data
with open(QBITTORRENT_AUTH_FILE, "r") as qbt_auth:
auth_data = json.load(qbt_auth)
# Collect infohashes of all torrents in each qBittorrent instance
torrent_infohashes = []
for item in auth_data["instances"]:
qb = Client(item["hostname"])
qb.login(username=item["username"], password=item["password"])
for torrent in qb.torrents():
torrent_infohashes.append(torrent.get("hash")) # type: ignore
# Format the infohashes to have a \n at the end
format_infohashes = [f"{infohash}\n" for infohash in torrent_infohashes]
# Create a NamedTemporaryFile and write all infohashes to it, one per line
with tempfile.NamedTemporaryFile() as ntf:
with open(ntf.name, "w") as tf:
tf.writelines(format_infohashes)
# Use scp to copy the infohashes file to the torrent tracker's config
# directory on the remote torrent tracker server, overwriting the
# whitelist.txt file
subprocess.run(
[
"scp",
ntf.name,
"root@bttracker.nirn.quest:/etc/opentracker/whitelist.txt",
]
)
# Use SSH to run `systemctl restart opentracker.service` on the remote
# torrent tracker server
subprocess.run(
[
"ssh",
"root@bttracker.nirn.quest",
"systemctl",
"restart",
"opentracker.service",
]
)
# Ensure bttracker.nirn.quest:6969/announce is added to each torrent's
# tracker list.
for item in auth_data["instances"]:
qb = Client(item["hostname"])
qb.login(username=item["username"], password=item["password"])
for torrent in qb.torrents():
qb.add_trackers(
torrent.get("hash"), # type: ignore
"http://bttracker.nirn.quest:6969/announce\nudp://bttracker.nirn.quest:6969/announce",
)
print(f"Updated {torrent.get('name')}") # type: ignore
# Reannounce all torrents in each qBittorrent instance to their trackers
for item in auth_data["instances"]:
qb = Client(item["hostname"])
qb.login(username=item["username"], password=item["password"])
torrent_infohashes = [torrent.get("hash") for torrent in qb.torrents()] # type: ignore
qb.reannounce(torrent_infohashes)