arthexis 0.1.16__py3-none-any.whl → 0.1.26__py3-none-any.whl

nodes/urls.py

@@ -8,8 +8,22 @@ urlpatterns = [
     path("register/", views.register_node, name="register-node"),
     path("screenshot/", views.capture, name="node-screenshot"),
     path("net-message/", views.net_message, name="net-message"),
-    path("last-message/", views.last_net_message, name="last-net-message"),
+    path("net-message/pull/", views.net_message_pull, name="net-message-pull"),
     path("rfid/export/", views.export_rfids, name="node-rfid-export"),
     path("rfid/import/", views.import_rfids, name="node-rfid-import"),
+    path("network/chargers/", views.network_chargers, name="node-network-chargers"),
+    path(
+        "network/chargers/forward/",
+        views.forward_chargers,
+        name="node-network-forward-chargers",
+    ),
+    path(
+        "network/chargers/action/",
+        views.network_charger_action,
+        name="node-network-charger-action",
+    ),
+    path("proxy/session/", views.proxy_session, name="node-proxy-session"),
+    path("proxy/login/<str:token>/", views.proxy_login, name="node-proxy-login"),
+    path("proxy/execute/", views.proxy_execute, name="node-proxy-execute"),
     path("<slug:endpoint>/", views.public_node_endpoint, name="node-public-endpoint"),
 ]

nodes/utils.py

@@ -11,6 +11,11 @@ from selenium import webdriver
 from selenium.webdriver.firefox.options import Options
 from selenium.common.exceptions import WebDriverException
 
+try:  # pragma: no cover - optional dependency may be missing
+    from geckodriver_autoinstaller import install as install_geckodriver
+except Exception:  # pragma: no cover - fallback when installer is unavailable
+    install_geckodriver = None
+
 from .classifiers import run_default_classifiers, suppress_default_classifiers
 from .models import ContentSample
 
@@ -18,6 +23,29 @@ SCREENSHOT_DIR = settings.LOG_DIR / "screenshots"
 CAMERA_DIR = settings.LOG_DIR / "camera"
 logger = logging.getLogger(__name__)
 
+_FIREFOX_BINARY_CANDIDATES = ("firefox", "firefox-esr", "firefox-bin")
+
+
+def _find_firefox_binary() -> str | None:
+    """Return the first available Firefox binary path or ``None``."""
+
+    for candidate in _FIREFOX_BINARY_CANDIDATES:
+        path = shutil.which(candidate)
+        if path:
+            return path
+    return None
+
+
+def _ensure_geckodriver() -> None:
+    """Install geckodriver on demand when possible."""
+
+    if install_geckodriver is None:  # pragma: no cover - dependency not installed
+        return
+    try:
+        install_geckodriver()
+    except Exception as exc:  # pragma: no cover - external failures are rare in tests
+        logger.warning("Unable to ensure geckodriver availability: %s", exc)
+
 
 def capture_screenshot(url: str, cookies=None) -> Path:
     """Capture a screenshot of ``url`` and save it to :data:`SCREENSHOT_DIR`.
@@ -25,8 +53,16 @@ def capture_screenshot(url: str, cookies=None) -> Path:
     ``cookies`` can be an iterable of Selenium cookie mappings which will be
     applied after the initial navigation and before the screenshot is taken.
     """
+    firefox_binary = _find_firefox_binary()
+    if not firefox_binary:
+        raise RuntimeError(
+            "Screenshot capture failed: Firefox is not installed. Install Firefox to enable screenshot capture."
+        )
+
     options = Options()
+    options.binary_location = firefox_binary
     options.add_argument("-headless")
+    _ensure_geckodriver()
     try:
         with webdriver.Firefox(options=options) as browser:
             browser.set_window_size(1280, 720)
@@ -48,7 +84,12 @@ def capture_screenshot(url: str, cookies=None) -> Path:
             return filename
     except WebDriverException as exc:
         logger.error("Failed to capture screenshot from %s: %s", url, exc)
-        raise RuntimeError(f"Screenshot capture failed: {exc}") from exc
+        message = str(exc)
+        if "Unable to obtain driver for firefox" in message:
+            message = (
+                "Firefox WebDriver is unavailable. Install geckodriver or configure the GECKODRIVER environment variable so Selenium can locate it."
+            )
+        raise RuntimeError(f"Screenshot capture failed: {message}") from exc
 
 
 def capture_rpi_snapshot(timeout: int = 10) -> Path:
@@ -90,10 +131,13 @@ def save_screenshot(
     *,
     content: str | None = None,
     user=None,
+    link_duplicates: bool = False,
 ):
     """Save screenshot file info if not already recorded.
 
-    Returns the created :class:`ContentSample` or ``None`` if duplicate.
+    Returns the created :class:`ContentSample`. If ``link_duplicates`` is ``True``
+    and a sample with identical content already exists, the existing record is
+    returned instead of ``None``.
     """
 
     original = path
@@ -101,7 +145,11 @@ def save_screenshot(
         path = settings.LOG_DIR / path
     with path.open("rb") as fh:
         digest = hashlib.sha256(fh.read()).hexdigest()
-    if ContentSample.objects.filter(hash=digest).exists():
+    existing = ContentSample.objects.filter(hash=digest).first()
+    if existing:
+        if link_duplicates:
+            logger.info("Duplicate screenshot content; reusing existing sample")
+            return existing
         logger.info("Duplicate screenshot content; record not created")
         return None
     stored_path = (original if not original.is_absolute() else path).as_posix()