#!/usr/bin/env python3 """qbth - qbittorrent helper Usage: qbth (HOSTNAME) (USERNAME) (PASSWORD) qbth -h Examples: qbth "http://localhost:8080" "admin" "adminadmin" qbth "https://cat.seedhost.eu/lol/qbittorrent" "lol" "meow" Options: -h, --help show this help message and exit """ import json import os import subprocess import requests from bs4 import BeautifulSoup from docopt import docopt from qbittorrent import Client args = docopt(__doc__) qb = Client(args["HOSTNAME"]) qb.login(username=args["USERNAME"], password=args["PASSWORD"]) def add_torrents(urls: list[str]): """ Add torrents from their URLs. Params: urls: list of strings that are URLs. """ for url in urls: qb.download_from_link(url, category="distro") print(f"Added {os.path.basename(url)}") def add_torrents_from_html(webpage_url: str, torrent_substring: str): """ Add torrents from an HTML web page. Params: webpage_url: a string that is the URL for the desired webpage. torrent_substring: a string that is a substring of the URLs in the webpage that you want to extract. It serves as a selector. """ reqs = requests.get(webpage_url, timeout=60) soup = BeautifulSoup(reqs.text, "html.parser") for link in soup.find_all("a"): if torrent_substring in link.get("href"): url = f"{webpage_url}/{link.get('href')}" qb.download_from_link(url, category="distro") print(f"Added {link.get('href')}") def remove_torrents(distro_substring: str): """ Remove torrents by selecting a substring that corresponds to the distro's torrent file name. When the substring is found, the torrent is removed by passing the corresponding hash to the method. Params: distro_substring: a string that is a substring of the distro torrent's file name. """ for torrent in qb.torrents(): if distro_substring in torrent.get("name") # type: ignore qb.delete_permanently(torrent.get("hash")) # type: ignore print(f"Removed {torrent.get('name')}") # type: ignore def add_almalinux(rel_ver: str): """ Add AlmaLinux torrents from a list of URLs. These URLs are partially hardcoded for convenience and aren't expected to change frequently. Params: relver: the AlmaLinux release version. """ urls = [ f"https://almalinux-mirror.dal1.hivelocity.net/{rel_ver}/isos/aarch64/AlmaLinux-{rel_ver}-aarch64.torrent", f"https://almalinux-mirror.dal1.hivelocity.net/{rel_ver}/isos/ppc64le/AlmaLinux-{rel_ver}-ppc64le.torrent", f"https://almalinux-mirror.dal1.hivelocity.net/{rel_ver}/isos/s390x/AlmaLinux-{rel_ver}-s390x.torrent", f"https://almalinux-mirror.dal1.hivelocity.net/{rel_ver}/isos/x86_64/AlmaLinux-{rel_ver}-x86_64.torrent", ] add_torrents(urls) def remove_almalinux(rel_ver: str): """ Remove AlmaLinux torrents given their release version. Params: relver: the AlmaLinux release version. """ remove_torrents(f"AlmaLinux-{rel_ver}") def add_debian(rel_ver: str): """ Add Debian torrents from a list of URLs. Params: relver: the Debian release version. """ urls = [ f"https://cdimage.debian.org/debian-cd/current/amd64/bt-dvd/debian-{rel_ver}-amd64-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/arm64/bt-dvd/debian-{rel_ver}-arm64-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/armel/bt-dvd/debian-{rel_ver}-armel-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/armhf/bt-dvd/debian-{rel_ver}-armhf-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/mips64el/bt-dvd/debian-{rel_ver}-mips64el-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/mipsel/bt-dvd/debian-{rel_ver}-mipsel-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/ppc64el/bt-dvd/debian-{rel_ver}-ppc64el-DVD-1.iso.torrent", f"https://cdimage.debian.org/debian-cd/current/s390x/bt-dvd/debian-{rel_ver}-s390x-DVD-1.iso.torrent", ] add_torrents(urls) def remove_debian(rel_ver: str): """ Remove Debian torrents given their release version. Params: relver: the Debian release version. """ remove_torrents(f"debian-{rel_ver}") def add_devuan(rel_ver: str): """ Add Devuan torrents from a URL. Params: relver: the Devuan release version. """ url = f"https://files.devuan.org/devuan_{rel_ver}.torrent" qb.download_from_link(url, category="distro") print(f"Added {os.path.basename(url)}") def remove_devuan(rel_ver: str): """ Remove Devuan torrents given their release version. Params: relver: the Devuan release version. """ remove_torrents(f"devuan_{rel_ver}") def add_fedora(rel_ver: str): """ Add Fedora torrents from URLs extracted from a webpage. Params: relver: the Fedora release version. """ webpage_url = "https://torrent.fedoraproject.org/torrents" torrent_substring = f"{rel_ver}.torrent" add_torrents_from_html(webpage_url, torrent_substring) def remove_fedora(rel_ver: str): """ Remove Fedora torrents given their release version. Params: relver: the Fedora release version. """ torrents = qb.torrents() for torrent in torrents: if torrent.get("name").startswith("Fedora") and torrent.get("name").endswith(rel_ver): # type: ignore qb.delete_permanently(torrent.get("hash")) # type: ignore print(f"Removed {torrent.get('name')}") # type: ignore def add_freebsd(rel_ver: str): """ Add FreeBSD torrents via a text file on the web that contains their magnet links. Params: relver: the FreeBSD release version. """ url = f"https://people.freebsd.org/~jmg/FreeBSD-{rel_ver}-R-magnet.txt" reqs = requests.get(url, timeout=60) data = reqs.text.split("\n") for line in data: if line.startswith("magnet:"): qb.download_from_link(line, category="distro") print(f"Added {line.split('=')[2]}") def remove_freebsd(rel_ver: str): """ Remove FreeBSD torrents given their release version. Params: relver: the FreeBSD release version. """ remove_torrents(f"FreeBSD-{rel_ver}") def add_kali(): """ Add Kali Linux torrents from their URLs extracted from a webpage. This method does not accept any parameters. The latest Kali Linux version is automatically selected. Params: none """ webpage_url = "https://kali.download/base-images/current" torrent_substring = ".torrent" add_torrents_from_html(webpage_url, torrent_substring) def remove_kali(): """ Remove Kali Linux torrents. This method does not accept any parameters. All Kali Linux torrents in the qBittorrent instance will be removed. Params: none """ remove_torrents("kali-linux") def add_netbsd(rel_ver: str): """ Add NetBSD torrents from their URLs extracted from a webpage. Params: relver: the NetBSD release version. """ webpage_url = f"https://cdn.netbsd.org/pub/NetBSD/NetBSD-{rel_ver}/images/" torrent_substring = ".torrent" add_torrents_from_html(webpage_url, torrent_substring) def remove_netbsd(rel_ver: str): """ Remove NetBSD torrents given their release version. Params: relver: the NetBSD release version. """ remove_torrents(f"NetBSD-{rel_ver}") def add_nixos(): """ Add NixOS torrents from their GitHub release at https://github.com/AninMouse/NixOS-ISO-Torrents. This method does not accept any paramters. The latest NixOS torrent is automatically selected. Params: none """ url = "https://api.github.com/repos/AnimMouse/NixOS-ISO-Torrents/releases/latest" reqs = requests.get(url, timeout=60) json_data = json.loads(reqs.text) for item in json_data["assets"]: qb.download_from_link(item["browser_download_url"], category="distro") print(f"Added {os.path.basename(item['browser_download_url'])}") def remove_nixos(): """ Remove NixOS torrents. This method does not accept any parameters. All NixOS torrents in the qBittorrent instance will be removed. Params: none """ remove_torrents("nixos") def add_qubes(rel_ver: str): """ Add QubesOS torrents from their URLs. Params: relver: the QubesOS release version. """ url = f"https://mirrors.edge.kernel.org/qubes/iso/Qubes-R{rel_ver}-x86_64.torrent" qb.download_from_link(url, category="distro") print(f"Added {os.path.basename(url)}") def remove_qubes(rel_ver: str): """ Remove QubesOS torrents given their release version. Params: relver: the Qubes OS release version. """ remove_torrents(f"Qubes-R{rel_ver}") def add_rockylinux(rel_ver: str): """ Add Rocky Linux torrents from their URLs. Params: relver: the Rocky Linux release version. """ urls = [ f"https://download.rockylinux.org/pub/rocky/{rel_ver}/isos/aarch64/Rocky-{rel_ver}-aarch64-dvd.torrent", f"https://download.rockylinux.org/pub/rocky/{rel_ver}/isos/ppc64le/Rocky-{rel_ver}-ppc64le-dvd.torrent", f"https://download.rockylinux.org/pub/rocky/{rel_ver}/isos/s390x/Rocky-{rel_ver}-s390x-dvd.torrent", f"https://download.rockylinux.org/pub/rocky/{rel_ver}/isos/x86_64/Rocky-{rel_ver}-x86_64-dvd.torrent", ] add_torrents(urls) def remove_rockylinux(rel_ver: str): """ Remove Rocky Linux torrents given their release version. Params: relver: the Rocky Linux release version. """ remove_torrents(f"Rocky-{rel_ver}") def add_tails(rel_ver: str): """ Add Tails torrents from their URLs. Params: relver: the Tails release version. """ urls = [ f"https://tails.net/torrents/files/tails-amd64-{rel_ver}.img.torrent", f"https://tails.net/torrents/files/tails-amd64-{rel_ver}.iso.torrent", ] add_torrents(urls) def remove_tails(rel_ver: str): """ Remove Tails torrents given their release version. Params: relver: the Tails release version. """ remove_torrents(f"tails-amd64-{rel_ver}") if __name__ == "__main__": # Run the gum program in a subprocess to allow easy selecting of distro # torrents. distro_selection = subprocess.run( [ "gum", "choose", "--limit=1", "--header='Available torrents'", "--height=13", "AlmaLinux", "Debian", "Devuan", "Fedora", "FreeBSD", "Kali Linux", "NetBSD", "NixOS", "Qubes", "Rocky Linux", "Tails", ], stdout=subprocess.PIPE, text=True, check=True, ).stdout.strip() # After the distro is selected and stored in the distro_selection variable, # choose an action to take on the selected distro. # Add: add the distro torrents to qBittorrent # Remove: remove the distro torrents from qBittorrent action_selection = subprocess.run( ["gum", "choose", "--limit=1", "--header='Choose:'", "Add", "Remove"], stdout=subprocess.PIPE, text=True, check=True, ).stdout.strip() # After the distro is selected and stored in the distro_selection variable, # and after the action is selected and store in the action_selection # variable, enter the release version of the selected distro to execute the # selected action on. relver = subprocess.run( [ "gum", "input", f"--placeholder='Enter {distro_selection} release version'", ], stdout=subprocess.PIPE, text=True, check=True, ).stdout.strip() # Match the distro_selection to execute the action_selection on. match distro_selection: case "AlmaLinux": if action_selection == "Add": add_almalinux(relver) if action_selection == "Remove": remove_almalinux(relver) case "Debian": if action_selection == "Add": add_debian(relver) if action_selection == "Remove": remove_debian(relver) case "Devuan": if action_selection == "Add": add_devuan(relver) if action_selection == "Remove": remove_devuan(relver) case "Fedora": if action_selection == "Add": add_fedora(relver) if action_selection == "Remove": remove_fedora(relver) case "FreeBSD": if action_selection == "Add": add_freebsd(relver) if action_selection == "Remove": remove_freebsd(relver) case "Kali Linux": if action_selection == "Add": add_kali() if action_selection == "Remove": remove_kali() case "NetBSD": if action_selection == "Add": add_netbsd(relver) if action_selection == "Remove": remove_netbsd(relver) case "NixOS": if action_selection == "Add": add_nixos() if action_selection == "Remove": remove_nixos() case "Qubes": if action_selection == "Add": add_qubes(relver) if action_selection == "Remove": remove_qubes(relver) case "Rocky Linux": if action_selection == "Add": add_rockylinux(relver) if action_selection == "Remove": remove_rockylinux(relver) case "Tails": if action_selection == "Add": add_tails(relver) if action_selection == "Remove": remove_tails(relver) case _: print("Nothing to do.")