CDMClient 0.1.0__tar.gz

cdmclient-0.1.0/CDMClient.egg-info/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: CDMClient
+Version: 0.1.0
+Summary: CDMClient
+Author-email: Aron Radics <radics.aron.jozsef@gmail.com>
+Requires-Python: >=3.9
+License-File: LICENSE
+Requires-Dist: requests
+Requires-Dist: transmission_rpc
+Requires-Dist: cryptography
+Provides-Extra: dev
+Requires-Dist: types-requests; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Requires-Dist: pylint; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: mypy; extra == "dev"

cdmclient-0.1.0/CDMClient.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+pyproject.toml
+CDMClient.egg-info/PKG-INFO
+CDMClient.egg-info/SOURCES.txt
+CDMClient.egg-info/dependency_links.txt
+CDMClient.egg-info/entry_points.txt
+CDMClient.egg-info/requires.txt
+CDMClient.egg-info/top_level.txt
+cdm_client/__init__.py
+cdm_client/cdm_client.py
+cdm_client/config.py
+cdm_client/transmission_adapter.py

cdmclient-0.1.0/CDMClient.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

cdmclient-0.1.0/CDMClient.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+cdm-client = cdm_client.cdm_client:main

cdmclient-0.1.0/CDMClient.egg-info/requires.txt

@@ -0,0 +1,10 @@
+requests
+transmission_rpc
+cryptography
+
+[dev]
+types-requests
+twine
+pylint
+black
+mypy

cdmclient-0.1.0/CDMClient.egg-info/top_level.txt

@@ -0,0 +1 @@
+cdm_client

cdmclient-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Aron Radics
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

cdmclient-0.1.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: CDMClient
+Version: 0.1.0
+Summary: CDMClient
+Author-email: Aron Radics <radics.aron.jozsef@gmail.com>
+Requires-Python: >=3.9
+License-File: LICENSE
+Requires-Dist: requests
+Requires-Dist: transmission_rpc
+Requires-Dist: cryptography
+Provides-Extra: dev
+Requires-Dist: types-requests; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Requires-Dist: pylint; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: mypy; extra == "dev"

cdmclient-0.1.0/README.md

@@ -0,0 +1,49 @@
+# CDMClient
+CDM - Centralized Download Manager Client
+
+## Installation
+
+### Install/Upgrade package
+``` shell
+python3 -m pip install --upgrade CDMClient --user
+```
+
+### Create service
+``` shell
+echo "[Unit]
+Description=cdm-client service
+After=multi-user.target
+Conflicts=getty@tty1.service
+[Service]
+User=${USER}
+Type=simple
+Environment="LC_ALL=C.UTF-8"
+Environment="LANG=C.UTF-8"
+ExecStart=${HOME}/.local/bin/cdm-client
+Restart=on-failure
+RestartSec=3
+[Install]
+WantedBy=multi-user.target" | sudo tee /etc/systemd/system/cdm-client.service
+```
+``` shell
+sudo systemctl daemon-reload
+sudo systemctl enable cdm-client.service
+sudo systemctl start cdm-client.service
+```
+
+## Configuartion
+Config file path: ```~/.config/cdm_client/config.ini```
+
+This file automatically generated when the service starts. See the example below.
+``` ini
+[connection]
+server_host =
+api_key =
+rpc_user =
+rpc_password =
+```
+
+## Check logs
+``` shell
+journalctl -fu cdm-client
+```

cdmclient-0.1.0/cdm_client/cdm_client.py

@@ -0,0 +1,66 @@
+import logging
+from logging.handlers import SysLogHandler
+from time import sleep
+import requests
+from cdm_client.config import Config
+from cdm_client.transmission_adapter import TransmissionAdapter
+
+
+class CDMClient:
+    def __init__(self) -> None:
+        self._logger = self._init_logger()
+        self._config = Config()
+        self._transmission_adapter = TransmissionAdapter(
+            self._config["rpc_user"] or None, self._config["rpc_password"] or None
+        )
+
+    def _init_logger(self) -> logging.Logger:
+        syslog = SysLogHandler(address="/dev/log")
+        syslog.setFormatter(logging.Formatter("cdm-client %(name)s: %(levelname)s %(message)s"))
+        logger = logging.getLogger("cdm-client")
+        logger.addHandler(syslog)
+        logger.setLevel(logging.INFO)
+        return logger
+
+    def _update_status(self) -> None:
+        data = {"data": self._transmission_adapter.get_status()}
+        resp = requests.post(
+            f"{self._config['host']}/api/client/status/",
+            json=data,
+            headers={"x-api-key": self._config["api_key"]},
+            timeout=5,
+        )
+        resp.raise_for_status()
+
+    def _download_files(self) -> None:
+        resp = requests.get(
+            f"{self._config['host']}/api/client/", headers={"x-api-key": self._config["api_key"]}, timeout=5
+        )
+        resp.raise_for_status()
+
+        files = resp.json()["data"]["files"]
+        for torrent_id, path in files.items():
+            resp = requests.get(
+                f"{self._config['host']}/api/client/download/{torrent_id}/",
+                headers={"x-api-key": self._config["api_key"]},
+                stream=True,
+                timeout=5,
+            )
+            resp.raise_for_status()
+            self._transmission_adapter.add_torrent(resp.content, download_dir=path)
+            self._logger.info("Downloading torrent: %s to %s", torrent_id, path)
+
+    def run(self) -> None:
+        self._logger.info("Starting cdm-client...")
+        while True:
+            try:
+                self._update_status()
+                self._download_files()
+                sleep(30)
+            except Exception:  # pylint: disable=broad-except
+                self._logger.exception("An error occurred.")
+
+
+def main() -> None:
+    cdm_client = CDMClient()
+    cdm_client.run()

cdmclient-0.1.0/cdm_client/config.py

@@ -0,0 +1,85 @@
+import os
+import configparser
+from typing import Union
+from cryptography.fernet import Fernet
+
+
+class Config:
+
+    CONFIG_FOLDER_PATH = os.path.join(os.path.expanduser("~"), ".config", "cdm_client")
+    KEY_PATH = os.path.join(CONFIG_FOLDER_PATH, "key.key")
+    CONFIG_PATH = os.path.join(CONFIG_FOLDER_PATH, "config.ini")
+    DEFAULT_CONFIG = {
+        "connection": {
+            "server_host": "",
+            "api_key": "",
+            "rpc_user": "",
+            "rpc_password": "",
+        }
+    }
+    ENCRYPTED_CONFIG = ["rpc_password"]
+
+    def __init__(self) -> None:
+        self._key: Union[bytes, str] = ""
+        self._config = self._load_config()
+        self._check_key()
+
+    def _check_key(self) -> None:
+        if not self._key_exists:
+            self._create_key()
+        else:
+            self._read_key()
+
+    def _create_key(self) -> None:
+        self._key = Fernet.generate_key()
+        with open(self.KEY_PATH, "w", encoding="utf-8") as f:
+            f.write(self._key.decode())
+
+    def _read_key(self) -> None:
+        with open(self.KEY_PATH, "r", encoding="utf-8") as f:
+            self._key = f.read().encode()
+
+    def _load_config(self) -> configparser.ConfigParser:
+        config = configparser.ConfigParser()
+        if config.read(self.CONFIG_PATH):
+            return config
+        return self._create_config()
+
+    def _create_config(self) -> configparser.ConfigParser:
+        if not os.path.exists(self.CONFIG_FOLDER_PATH):
+            os.makedirs(self.CONFIG_FOLDER_PATH)
+        config = configparser.ConfigParser()
+        for section, configs in self.DEFAULT_CONFIG.items():
+            config[section] = {}
+            for config_key, config_value in configs.items():
+                config[section][config_key] = config_value
+        with open(self.CONFIG_PATH, "w", encoding="utf-8") as f:
+            config.write(f)
+        return config
+
+    def _decrypt(self, value: str) -> str:
+        f = Fernet(self._key)
+        return f.decrypt(value.encode()).decode()
+
+    def _encrypt(self, value: str) -> str:
+        f = Fernet(self._key)
+        return f.encrypt(value.encode()).decode()
+
+    def _write_creds(self) -> None:
+        with open(self.CONFIG_PATH, "w", encoding="utf-8") as f:
+            self._config.write(f)
+
+    @property
+    def _key_exists(self) -> bool:
+        return os.path.exists(self.KEY_PATH)
+
+    def __getitem__(self, name: str) -> str:
+        if name in self.ENCRYPTED_CONFIG:
+            try:
+                return self._decrypt(self._config["connection"][name])
+            except Exception:  # pylint: disable=broad-except
+                raw_value = self._config["connection"][name]
+                self._config["connection"][name] = self._encrypt(self._config["connection"][name])
+                self._write_creds()
+                return raw_value
+        return self._config["connection"][name]

cdmclient-0.1.0/cdm_client/transmission_adapter.py

@@ -0,0 +1,31 @@
+import logging
+from typing import Optional
+from transmission_rpc import Client
+
+
+class TransmissionAdapter:
+    def __init__(self, username: Optional[str] = None, password: Optional[str] = None) -> None:
+        self._client = Client(username=username, password=password)
+        self._logger = logging.getLogger("cdm-client")
+
+    def get_status(self) -> list[dict]:
+        torrents = self._client.get_torrents()
+        status = []
+        for torrent in torrents:
+            status.append(
+                {
+                    "id": torrent.id,
+                    "name": torrent.name,
+                    "status": torrent.status.value,
+                    "progress": int(torrent.progress),
+                    "downloadDir": torrent.download_dir,
+                    "addedDate": torrent.added_date.timestamp(),
+                    "totalSize": torrent.total_size,
+                    "eta": torrent.eta.total_seconds() if torrent.eta else None,
+                }
+            )
+        self._logger.info("Retrieved status of %s torrents", len(status))
+        return status
+
+    def add_torrent(self, torrent: bytes, download_dir: str) -> None:
+        self._client.add_torrent(torrent, download_dir=download_dir)

cdmclient-0.1.0/pyproject.toml

@@ -0,0 +1,42 @@
+[project]
+name = "CDMClient"
+version = "0.1.0"
+description = "CDMClient"
+authors = [{name = "Aron Radics", email = "radics.aron.jozsef@gmail.com"}]
+requires-python = ">=3.9"
+dependencies = [
+  "requests",
+  "transmission_rpc",
+  "cryptography",
+]
+
+[project.optional-dependencies]
+dev = [
+  "types-requests",
+  "twine",
+  "pylint",
+  "black",
+  "mypy",
+]
+
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools]
+packages = ["cdm_client"]
+
+[project.scripts]
+cdm-client = "cdm_client.cdm_client:main"
+
+[tool.pylint]
+disable = ["missing-docstring", "too-few-public-methods"]
+max-line-length = 120
+
+[tool.mypy]
+disallow_untyped_calls = true
+disallow_incomplete_defs = true
+disallow_untyped_defs = true
+
+[tool.black]
+line-length = 120

cdmclient-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+