dl/commands/nyaa-404.py
Michel Roux 99162297ee
All checks were successful
continuous-integration/drone/push Build is passing
Add 404 to nyaa
2023-04-23 21:58:57 +02:00

44 lines
1.2 KiB
Python

from argparse import ArgumentParser
from logging import INFO, StreamHandler
from os import path
from sys import stdout
from time import sleep
from requests import head
from transmissionrpc import Client, DefaultHTTPHandler, LOGGER
parser = ArgumentParser()
parser.add_argument('-u', '--username', required=True)
parser.add_argument('-p', '--password', required=True)
args = parser.parse_args()
LOGGER.setLevel(INFO)
LOGGER.addHandler(StreamHandler(stdout))
class CustomHTTPHandler(DefaultHTTPHandler):
def request(self, url, query, headers, timeout):
headers[
"User-Agent"
] = "Mozilla/5.0 (X11; Linux x86_64; rv:111.0) Gecko/20100101 Firefox/111.0"
return super().request(url, query, headers, timeout)
client = Client(
"https://torrent.crystalyx.net/transmission/rpc",
port=443,
user=args.username,
password=args.password,
http_handler=CustomHTTPHandler(),
)
for torrent in client.get_torrents():
if "nyaa" in torrent.comment:
response = head(torrent.comment)
if response.status_code != 200:
print(
f"{response.status_code} - {path.join(torrent.downloadDir, torrent.name)}"
)
sleep(1)