Add qbittorrent api class

Config.py

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

QBitAPI.py

@@ -1 +1,67 @@
 # -*- 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}%"

TYPES.py

@@ -7,6 +7,11 @@ class DirParserTypes(Enum):
 
 
 class LogTypes(Enum):
-    LOG = "Log"
-    ERROR = "Error"
-    WARN = "Warning"
+    LOG = "LOG"
+    ERROR = "ERROR"
+    WARN = "WARNING"
+
+
+class TorrentTypes(Enum):
+    MAGNET = "Magnet Link"
+    FILE = "Torrent File"

Torrent.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+from abc import ABCMeta, abstractmethod, abstractproperty
+from typing import BinaryIO
+
+from TYPES import TorrentTypes
+from os.path import splitext as getext
+from re import match
+from Config import tmpDir
+
+
+class Torrent(object):
+    __metaclass__ = ABCMeta
+
+    def __init__(self, url: str, path: str):
+        self.url = url
+        self.path = path
+        if not (self.validate()):
+            raise ValueError
+
+    @property
+    @abstractmethod
+    def torrent_type(self) -> TorrentTypes:
+        pass
+
+    @abstractmethod
+    def validate(self) -> bool:
+        pass
+
+    @property
+    def value(self) -> str:
+        return self.value
+
+
+class MagnetLink(Torrent):
+
+    @property
+    def torrent_type(self) -> TorrentTypes:
+        return TorrentTypes.MAGNET
+
+    def validate(self) -> bool:
+        pattern = r"^magnet:\?xt=urn:btih:[a-fA-F0-9]{40}.*$"
+        if match(pattern, self.url):
+            return True
+
+
+class TorrentFile(Torrent):
+
+    def __init__(self, url: str, path: str):
+        super().__init__(url, path)
+        self.url = f"{tmpDir}/{url}"
+
+    @property
+    def torrent_type(self) -> TorrentTypes:
+        return TorrentTypes.FILE
+
+    def validate(self) -> bool:
+        if getext(self.url)[1].lower() == ".torrent":
+            return True

config.json

@@ -1,11 +1,12 @@
 {
   "Main": {
     "DirParser": "Jellyfin",
-    "DBPath": "/etc/qbitbot/botdb.db"
+    "DBPath": "/etc/qbitbot/botdb.db",
+    "TMPDir": "/tmp/qbitbot"
   },
 
   "JellyfinConfig": {
     "ServerURL": "https://yourdomain.com",
     "APIKey": "youaractualjellyfintoken"
   }
-}
+}