4
1
Fork 1

Add qbittorrent api class

This commit is contained in:
trueold89 2024-05-09 17:47:57 +03:00
parent e5857715e8
commit f9681e35f4
Signed by: trueold89
GPG Key ID: C122E85DD49E6B30
4 changed files with 79 additions and 3 deletions

View File

@ -31,3 +31,4 @@ configuration = loadCFG(CFG_PATH)
dirParser, JellyfinConfig = loadParser(configuration) dirParser, JellyfinConfig = loadParser(configuration)
DBPath = configuration["Main"]["DBPath"] DBPath = configuration["Main"]["DBPath"]
tmpDir = configuration["Main"]["TMPDir"]

View File

@ -1 +1,67 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from requests import get, post
from requests.cookies import RequestsCookieJar
from Torrent import TorrentFile, MagnetLink, Torrent, TorrentTypes
class Qbit(object):
def __init__(self, username: str, passwd: str, url: str):
self.username = username
self.passwd = passwd
self.baseurl = url
self.cookies = self.__getcookies()
def __getcookies(self) -> RequestsCookieJar:
creds = {
"username": self.username,
"password": self.passwd
}
resp = post(f"{self.baseurl}/api/v2/auth/login", data=creds)
cookies = resp.cookies
if get(f"{self.baseurl}/api/v2/app/version", cookies=cookies).status_code != 200:
raise Exception("Error connecting to QBitTorrentAPI")
return cookies
def getlist(self) -> list:
json = get(f"{self.baseurl}/api/v2/torrents/info?filter=completed,downloading&sort=progress", cookies=self.cookies).json()
output = []
for element in json:
output.append(TorrentElement(element["name"], element["state"], float(element["progress"])))
return output
def __uploadfile(self, torrent: TorrentFile) -> int:
with open(torrent.url, "rb") as torrent_binary:
torrent_file = {"torrents": (torrent.url, torrent_binary, "application/x-bittorrent")}
params = {"savepath": torrent.path}
resp = post(url=f"{self.baseurl}/api/v2/torrents/add", cookies=self.cookies, files=torrent_file, data=params)
return resp.status_code
def __uploadmagnet(self, torrent: MagnetLink) -> int:
params = {
"urls": torrent.url,
"savepath": torrent.path
}
resp = post(url=f"{self.baseurl}/api/v2/torrents/add", cookies=self.cookies, data=params)
return resp.status_code
def upload(self, torrent: Torrent) -> None:
status = None
match torrent.torrent_type:
case TorrentTypes.FILE:
status = self.__uploadfile(torrent)
case TorrentTypes.MAGNET:
status = self.__uploadmagnet(torrent)
if status != 200:
raise Exception("Error downloading torrent")
class TorrentElement(object):
def __init__(self, name: str, state: str, progress: float):
self.name = name
self.state = state
self.progress = round(progress*100, 1)
def __str__(self):
return f"{self.name} | {self.state} | {self.progress}%"

View File

@ -1,15 +1,19 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from abc import ABCMeta, abstractmethod, abstractproperty from abc import ABCMeta, abstractmethod, abstractproperty
from typing import BinaryIO
from TYPES import TorrentTypes from TYPES import TorrentTypes
from os.path import splitext as getext from os.path import splitext as getext
from re import match from re import match
from Config import tmpDir
class Torrent(object): class Torrent(object):
__metaclass__ = ABCMeta __metaclass__ = ABCMeta
def __init__(self, url: str): def __init__(self, url: str, path: str):
self.url = url self.url = url
self.path = path
if not (self.validate()): if not (self.validate()):
raise ValueError raise ValueError
@ -41,6 +45,10 @@ class MagnetLink(Torrent):
class TorrentFile(Torrent): class TorrentFile(Torrent):
def __init__(self, url: str, path: str):
super().__init__(url, path)
self.url = f"{tmpDir}/{url}"
@property @property
def torrent_type(self) -> TorrentTypes: def torrent_type(self) -> TorrentTypes:
return TorrentTypes.FILE return TorrentTypes.FILE

View File

@ -1,7 +1,8 @@
{ {
"Main": { "Main": {
"DirParser": "Jellyfin", "DirParser": "Jellyfin",
"DBPath": "/etc/qbitbot/botdb.db" "DBPath": "/etc/qbitbot/botdb.db",
"TMPDir": "/tmp/qbitbot"
}, },
"JellyfinConfig": { "JellyfinConfig": {