wi1-bot 1.4.1__py3-none-any.whl → 1.4.3__py3-none-any.whl

wi1_bot/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.4.1'
-__version_tuple__ = version_tuple = (1, 4, 1)
+__version__ = version = '1.4.3'
+__version_tuple__ = version_tuple = (1, 4, 3)

wi1_bot/config.py

@@ -4,20 +4,21 @@ from typing import TypedDict
 
 import yaml
 
-_config_path: str | None = None
+_config_path = os.getenv("WB_CONFIG_PATH")
 
-if home := os.getenv("HOME"):
-    _path = pathlib.Path(home) / ".config" / "wi1-bot" / "config.yaml"
-    if _path.is_file():
-        _config_path = str(_path.resolve())
+if _config_path is None:
+    if home := os.getenv("HOME"):
+        _path = pathlib.Path(home) / ".config" / "wi1-bot" / "config.yaml"
+        if _path.is_file():
+            _config_path = str(_path.resolve())
 
-if xdg_config_home := os.getenv("XDG_CONFIG_HOME"):
-    _path = pathlib.Path(xdg_config_home) / "wi1-bot" / "config.yaml"
-    if _path.is_file():
-        _config_path = str(_path.resolve())
+    if xdg_config_home := os.getenv("XDG_CONFIG_HOME"):
+        _path = pathlib.Path(xdg_config_home) / "wi1-bot" / "config.yaml"
+        if _path.is_file():
+            _config_path = str(_path.resolve())
 
-if pathlib.Path("config.yaml").is_file():
-    _config_path = "config.yaml"
+    if pathlib.Path("config.yaml").is_file():
+        _config_path = "config.yaml"
 
 if _config_path is None:
     raise FileNotFoundError(
@@ -99,8 +100,14 @@ class TranscodingConfig(TranscodingConfigOptional):
     profiles: dict[str, TranscodingProfile]
 
 
+class RemotePathMapping(TypedDict):
+    remote: str
+    local: str
+
+
 class GeneralConfigOptional(TypedDict, total=False):
     log_dir: str
+    remote_path_mappings: list[RemotePathMapping]
 
 
 class GeneralConfig(GeneralConfigOptional):
@@ -121,3 +128,9 @@ class Config(ConfigOptional):
 
 with open(_config_path, "r") as f:
     config: Config = yaml.load(f, Loader=yaml.SafeLoader)
+
+if log_dir := os.getenv("WB_LOG_DIR"):
+    if "general" not in config:
+        config["general"] = {}
+
+    config["general"]["log_dir"] = log_dir

wi1_bot/webhook.py

@@ -8,7 +8,7 @@ from flask import Flask, request
 
 from wi1_bot import push, transcoder
 from wi1_bot.arr import Radarr, Sonarr
-from wi1_bot.config import config
+from wi1_bot.config import RemotePathMapping, config
 
 app = Flask(__name__)
 
@@ -26,6 +26,37 @@ def on_grab(req: dict[str, Any]) -> None:
     )
 
 
+def replace_remote_paths(path: pathlib.Path) -> pathlib.Path:
+    if "general" not in config or "remote_path_mappings" not in config["general"]:
+        return path
+
+    mappings = config["general"]["remote_path_mappings"]
+
+    most_specific: RemotePathMapping | None = None
+
+    for mapping in mappings:
+        if path.is_relative_to(mapping["remote"]):
+            mapping_len = len(pathlib.Path(mapping["remote"]).parts)
+            most_specific_len = (
+                len(pathlib.Path(most_specific["remote"]).parts)
+                if most_specific is not None
+                else 0
+            )
+
+            if mapping_len > most_specific_len:
+                most_specific = mapping
+
+    if most_specific is not None:
+        remote_path = path
+        path = pathlib.Path(most_specific["local"]) / path.relative_to(
+            most_specific["remote"]
+        )
+
+        logger.debug(f"replaced remote path mapping: {remote_path} -> {path}")
+
+    return path
+
+
 def on_download(req: dict[str, Any]) -> None:
     path: pathlib.Path
     content_id: int
@@ -65,6 +96,8 @@ def on_download(req: dict[str, Any]) -> None:
     else:
         raise ValueError("unknown download request")
 
+    path = replace_remote_paths(path)
+
     try:
         quality_options = config["transcoding"]["profiles"][quality_profile]
     except KeyError:
@@ -119,10 +152,10 @@ def index() -> Any:
 def start() -> None:
     logger.info("starting webhook listener")
 
-    t = threading.Thread(target=app.run, kwargs={"host": "localhost", "port": 9000})
+    t = threading.Thread(target=app.run, kwargs={"host": "0.0.0.0", "port": 9000})
     t.daemon = True
     t.start()
 
 
 if __name__ == "__main__":
-    app.run(host="localhost", port=9000)
+    app.run(host="0.0.0.0", port=9000)

{wi1_bot-1.4.1.dist-info → wi1_bot-1.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wi1-bot
-Version: 1.4.1
+Version: 1.4.3
 Summary: Discord bot for Radarr/Sonarr integration
 Home-page: https://github.com/wthueb/wi1-bot
 Author-email: William Huebner <wilhueb@gmail.com>
@@ -34,20 +34,18 @@ Classifier: Programming Language :: Python :: 3.10
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: discord.py ==2.3.1
-Requires-Dist: Flask ==2.3.2
-Requires-Dist: mongoengine ==0.27.0
+Requires-Dist: discord.py ==2.3.2
+Requires-Dist: Flask ==3.0.2
+Requires-Dist: mongoengine ==0.28.1
 Requires-Dist: pyarr ==5.2.0
 Requires-Dist: PyYAML ==6.0.1
 Requires-Dist: requests ==2.31.0
 Provides-Extra: dev
-Requires-Dist: black ==23.3.0 ; extra == 'dev'
-Requires-Dist: flake8 ==6.0.0 ; extra == 'dev'
-Requires-Dist: isort ==5.12.0 ; extra == 'dev'
 Requires-Dist: mongo-types ==0.15.1 ; extra == 'dev'
-Requires-Dist: mypy ==1.2.0 ; extra == 'dev'
-Requires-Dist: pre-commit ==3.2.2 ; extra == 'dev'
-Requires-Dist: types-PyYAML ==6.0.12.9 ; extra == 'dev'
+Requires-Dist: mypy ==1.3.0 ; extra == 'dev'
+Requires-Dist: pre-commit ==3.6.2 ; extra == 'dev'
+Requires-Dist: ruff ==0.3.0 ; extra == 'dev'
+Requires-Dist: types-PyYAML ==6.0.12.12 ; extra == 'dev'
 
 # wi1-bot
 

{wi1_bot-1.4.1.dist-info → wi1_bot-1.4.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
 wi1_bot/__init__.py,sha256=11ozJKiUsqDCZ3_mcAHhGYUyGK_Unl54djVSBBExFB4,59
-wi1_bot/_version.py,sha256=oFZsPxoSsCY6D2DiWMSueNvMDRRQN5ssWrPdQtlLJ_o,411
-wi1_bot/config.py,sha256=cWgpI7GUwqUed43RZhD1ruZT0A5RoN8VESbSo8uwr5A,2797
+wi1_bot/_version.py,sha256=cUUlK35wR7qLNaS4SMOrkRP_HG0nSHq0YEIezhPHr3c,411
+wi1_bot/config.py,sha256=AtzOXvCeat_lb-w_Wy37XyWIpV3LmKh38OGigluJ2nM,3139
 wi1_bot/push.py,sha256=-Az8c21R3IZAkJVRQmWsbNXMfvBzzpc9pFTWsDy1mJE,789
-wi1_bot/webhook.py,sha256=SEKd11Hkcewf9Y_DvMKfo2TEP6wgqAGWUlGls7t0gso,3685
+wi1_bot/webhook.py,sha256=cVp6USg1vbfL05TPQSDIUUYPHWDjuPGb1IGsqImao6s,4713
 wi1_bot/arr/__init__.py,sha256=ZIgkW24GBdS4sJmaiEIhueHOB6s2L8y2s9ahgp1USRI,86
 wi1_bot/arr/download.py,sha256=02AYFglnFdWSG8xj_TaJc6l2wjybyhUW6F97CnoyUFw,1381
 wi1_bot/arr/episode.py,sha256=j25ljy9hyTXFAEBeUQ4iozKP2YXpZyzauphaXa2oqh8,1057
@@ -22,9 +22,9 @@ wi1_bot/scripts/transcode_item.py,sha256=21MeeIZ9wIRhAY-FOEgOdPsg6YOLlSUj59r8NkT
 wi1_bot/transcoder/__init__.py,sha256=B4xr82UtIFc3tyy_MEZdZKMukYW0yejPnfsGowaTIM0,105
 wi1_bot/transcoder/transcode_queue.py,sha256=W7r2I7OD8QuxseCEb7_ddOvYXiBBLmKLvCs2IgJJ92I,1856
 wi1_bot/transcoder/transcoder.py,sha256=3MiOnQaXHmS6oQsNEygd9-SeGH7nIipnTK2f7BJFB6Q,8684
-wi1_bot-1.4.1.dist-info/LICENSE,sha256=6V4_mQoPoLJl77_WMsQRQMDG2mnIhDUdfCmMkqM9Qwc,1072
-wi1_bot-1.4.1.dist-info/METADATA,sha256=lh8bpMXqsnTC3910f3MSWoH9bxsLaRhi4l5e3kzmVBI,4612
-wi1_bot-1.4.1.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-wi1_bot-1.4.1.dist-info/entry_points.txt,sha256=G_NEW2uMHHw4pbmVwGZ6cZaBwGpdkv741jNYtuX-1vM,147
-wi1_bot-1.4.1.dist-info/top_level.txt,sha256=Q7mTnPLk80Td82YbjlBMO5tvXJoTFHhheHdmwc-c-7I,8
-wi1_bot-1.4.1.dist-info/RECORD,,
+wi1_bot-1.4.3.dist-info/LICENSE,sha256=6V4_mQoPoLJl77_WMsQRQMDG2mnIhDUdfCmMkqM9Qwc,1072
+wi1_bot-1.4.3.dist-info/METADATA,sha256=niCQgmOgYsWFR6wp5Ya6aZkhV0MHoaOHJ_CzA5mut3Y,4517
+wi1_bot-1.4.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+wi1_bot-1.4.3.dist-info/entry_points.txt,sha256=G_NEW2uMHHw4pbmVwGZ6cZaBwGpdkv741jNYtuX-1vM,147
+wi1_bot-1.4.3.dist-info/top_level.txt,sha256=Q7mTnPLk80Td82YbjlBMO5tvXJoTFHhheHdmwc-c-7I,8
+wi1_bot-1.4.3.dist-info/RECORD,,

{wi1_bot-1.4.1.dist-info → wi1_bot-1.4.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.41.2)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any