Add qBitTorrent torrent server API implementation

- init torrent.api
- Add qBitTorrent class

setup.py

@@ -7,10 +7,6 @@ setup(
     author="ORUDO",
     author_email="root@orudo.ru",
     description="A simple Telegram bot that will allow you to upload torrent files / magnet links to a remote Torrent server (qBitTorrent, Transmission, etc.)",
-    install_requires=[
-        "aiohttp>=3.10.0",
-        "aiofiles>=24.1.0",
-        "aiofiles>=24.1.0"
-        ],
+    install_requires=["aiohttp>=3.10.0", "aiofiles>=24.1.0", "aiofiles>=24.1.0"],
     packages=["tubot", "tubot.static", "tubot.torrent"],
 )

tubot/torrent/apis.py

@@ -0,0 +1,130 @@
+# -*- coding: utf-8 -*-
+
+########################################
+# Torrent Server API's implementations #
+########################################
+
+# Imports
+from http.cookies import SimpleCookie
+from aiohttp import ClientSession, ClientResponse, FormData
+from tubot.torrent.abc import TorrentAPI
+from tubot.torrent.torrents import TorrentFile, TorrentMagnet, TorrentURL
+from tubot.torrent.types import ServerTypes, TorrentListBuilder
+from tubot.static.functions import validate
+
+
+class qBitTorrent(TorrentAPI):
+    """
+    qBitTorrent API implementation
+    """
+
+    host: str
+    username: str
+    password: str
+    cookie: SimpleCookie | None
+    _atype = ServerTypes.qBitTorrent
+
+    def __init__(self, host: str, username: str, password: str) -> None:
+        """
+        :param host: qBitTorrent remote server adress
+        :param username: qBitTorrent remote username
+        :param password: qBitTorrent remote password
+        """
+        super().__init__()
+        self.cookie = None
+        self.host = host
+        self.username = username
+        self.password = password
+
+    async def _get(
+        self, api: str, cookie: SimpleCookie | None = None
+    ) -> ClientResponse:
+        """
+        Send get request to Torrent server
+
+        :param api: API schema
+        :param cookie: Cookies for auth
+        """
+        async with ClientSession() as session:
+            return await session.get(url=f"{self.host}/{api}", cookies=cookie)
+
+    async def _post(
+        self,
+        api: str,
+        cookie: SimpleCookie | None = None,
+        data: dict | FormData | None = None,
+    ) -> ClientResponse:
+        """
+        Send post request to Torrent server
+
+        :param api: API schema
+        :param cookie: Cookies for auth
+        :param data: Request data
+        """
+        async with ClientSession() as session:
+            return await session.post(
+                url=f"{self.host}/{api}", cookies=cookie, data=data
+            )
+
+    async def auth(self) -> bool:
+        """
+        Generates cookies for auth
+        """
+        creds = {"username": self.username, "password": self.password}
+        resp = await self._post(api="api/v2/auth/login", data=creds)
+        try:
+            if resp.status == 200:
+                cookies = resp.cookies
+                resp = await self._get(api="api/v2/app/version", cookie=cookies)
+                if resp.status != 200:
+                    raise ValueError("Auth error")
+                self.cookie = cookies
+                return True
+        except Exception:
+            pass
+        return False
+
+    async def upload_file(self, torrent: TorrentFile) -> None:
+        await validate(self)
+        await validate(torrent, "Bad .torrent file")
+        bytes = await torrent.getbytes()
+        data = FormData()
+        data.add_field(
+            "torrents",
+            bytes,
+            filename=torrent.content,
+            content_type="application/x-bittorrent",
+        )
+        data.add_field("savepath", torrent.dest)
+        await self._post("api/v2/torrents/add", cookie=self.cookie, data=data)
+
+    async def upload_magnet(self, torrent: TorrentMagnet) -> None:
+        await validate(self)
+        await validate(torrent, "Bad magnet link")
+        data = {"urls": torrent.content, "savepath": torrent.dest}
+        await self._post("api/v2/torrents/add", cookie=self.cookie, data=data)
+
+    async def upload_url(self, torrent: TorrentURL) -> None:
+        await validate(self)
+        await validate(torrent, "Bad url")
+        data = {"urls": torrent.content, "savepath": torrent.dest}
+        await self._post("api/v2/torrents/add", cookie=self.cookie, data=data)
+
+    @property
+    async def torrent_list(self) -> str:
+        await validate(self)
+        responce = await self._get(
+            "api/v2/torrents/info?filter=completed,downloading&sort=progress",
+            cookie=self.cookie,
+        )
+        responce = await responce.json()
+        lst = tuple(
+            map(lambda i: (i["name"], i["state"], float(i["progress"])), responce)
+        )
+        lb = TorrentListBuilder()
+        for torrent in lst:
+            lb.append(torrent)
+        return str(lb)
+
+    async def __validate__(self) -> bool:
+        return await self.auth()

tubot/torrent/types.py

@@ -24,9 +24,10 @@ class ServerTypes(Enum):
     Types of Torrent servers API's
     """
 
+    qBitTorrent = "qBitTorrent Remote API"
+
 
 class TorrentFromServer(object):
-
     name: str
     state: str
     percent: float
@@ -34,10 +35,10 @@ class TorrentFromServer(object):
     def __init__(self, name: str, state: str, percent: float) -> None:
         self.name = name
         self.state = state
-        self.percent = percent
+        self.percent = round(percent * 100, 1)
 
     def __str__(self) -> str:
-        return f"*Torrent:* {self.name}\n*State:* {self.state}\n*Progress:* {self.percent}"
+        return f"*Torrent:* {self.name}\n*State:* {self.state}\n*Progress:* {self.percent}%"
 
 
 class TorrentListBuilder(object):
@@ -52,7 +53,7 @@ class TorrentListBuilder(object):
 
     def append(self, torrent_data: Iterable) -> "TorrentListBuilder":
         item = TorrentFromServer(*torrent_data)
-        self.collection.append(item)
+        self.collection.append(str(item))
         return self
 
     def __str__(self) -> str: