inventree-import-plugin 0.2.13__tar.gz

inventree_import_plugin-0.2.13/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: inventree-import-plugin
+Version: 0.2.13
+Summary: Combined InvenTree supplier import and enrich plugin for Mouser and LCSC
+License: MIT
+Requires-Python: >=3.10
+Requires-Dist: requests>=2.28
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Requires-Dist: mypy; extra == "dev"

inventree_import_plugin-0.2.13/inventree_import_plugin/__init__.py

@@ -0,0 +1,6 @@
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    PLUGIN_VERSION = version('inventree-import-plugin')
+except PackageNotFoundError:
+    PLUGIN_VERSION = '0.0.0-dev'

inventree_import_plugin-0.2.13/inventree_import_plugin/api/__init__.py

@@ -0,0 +1,3 @@
+from inventree_import_plugin.api.views import build_urlpatterns
+
+__all__ = ["build_urlpatterns"]

inventree_import_plugin-0.2.13/inventree_import_plugin/api/views.py

@@ -0,0 +1,155 @@
+from __future__ import annotations
+
+import json
+import pathlib
+from typing import Any
+
+from inventree_import_plugin.services import (
+    bulk_enrich,
+    enrich_part_for_provider,
+    get_provider_state,
+    parse_bulk_operations,
+    parse_bulk_payload,
+)
+
+
+def build_urlpatterns(plugin: Any) -> list[Any]:
+    from django.middleware.csrf import get_token
+    from django.shortcuts import render
+    from django.urls import path
+    from InvenTree.permissions import RolePermission
+    from rest_framework.response import Response
+    from rest_framework.views import APIView
+
+    class _BaseRoleView(APIView):  # type: ignore[misc]
+        permission_classes = [RolePermission]
+        role_required = "part.change"
+
+    class _ProviderStateView(_BaseRoleView):
+        def get(inner_self, request: Any, part_id: int) -> Any:  # noqa: N805
+            return Response(get_provider_state(plugin, part_id))
+
+    class _PreviewView(_BaseRoleView):
+        def get(inner_self, request: Any, part_id: int, provider_slug: str) -> Any:  # noqa: N805
+            return Response(
+                enrich_part_for_provider(
+                    plugin,
+                    provider_slug,
+                    part_id,
+                    dry_run=True,
+                    user=getattr(request, "user", None),
+                )
+            )
+
+    class _ApplyView(_BaseRoleView):
+        def post(inner_self, request: Any, part_id: int, provider_slug: str) -> Any:  # noqa: N805
+            selected_keys = request.data.get("selected_keys")
+            if selected_keys is not None:
+                selected_keys = set(selected_keys)
+            return Response(
+                enrich_part_for_provider(
+                    plugin,
+                    provider_slug,
+                    part_id,
+                    dry_run=False,
+                    selected_keys=selected_keys,
+                    user=getattr(request, "user", None),
+                )
+            )
+
+    class _BulkPreviewView(_BaseRoleView):
+        def post(inner_self, request: Any) -> Any:  # noqa: N805
+            try:
+                part_ids, provider_slugs = parse_bulk_payload(plugin, request)
+            except ValueError as exc:
+                return Response({"detail": str(exc)}, status=400)
+            return Response(
+                bulk_enrich(
+                    plugin,
+                    part_ids,
+                    provider_slugs,
+                    dry_run=True,
+                    user=getattr(request, "user", None),
+                )
+            )
+
+    class _BulkApplyView(_BaseRoleView):
+        def post(inner_self, request: Any) -> Any:  # noqa: N805
+            try:
+                if "operations" in request.data:
+                    operations = parse_bulk_operations(plugin, request)
+                    return Response(
+                        bulk_enrich(
+                            plugin,
+                            dry_run=False,
+                            operations=operations,
+                            user=getattr(request, "user", None),
+                        )
+                    )
+                part_ids, provider_slugs = parse_bulk_payload(plugin, request)
+                return Response(
+                    bulk_enrich(
+                        plugin,
+                        part_ids,
+                        provider_slugs,
+                        dry_run=False,
+                        user=getattr(request, "user", None),
+                    )
+                )
+            except ValueError as exc:
+                return Response({"detail": str(exc)}, status=400)
+
+    class _BulkPageView(_BaseRoleView):
+        _bundle_path = (
+            pathlib.Path(__file__).resolve().parent.parent
+            / "static"
+            / "bulk"
+            / "StandaloneBulkPage-v3.js"
+        )
+
+        def get(inner_self, request: Any) -> Any:  # noqa: N805
+            active_providers = [
+                {
+                    "slug": adapter.definition.slug,
+                    "name": adapter.definition.name,
+                }
+                for adapter in plugin._get_active_provider_adapters(require_complete_config=True)
+            ]
+
+            bundle_js = ""
+            if inner_self._bundle_path.exists():
+                bundle_js = inner_self._bundle_path.read_text(encoding="utf-8")
+
+            context = {
+                "bundle_js": bundle_js,
+                "mount_context_json": json.dumps(
+                    {
+                        "pluginSlug": plugin.SLUG,
+                        "previewUrl": f"/plugin/{plugin.SLUG}/api/bulk/preview/",
+                        "applyUrl": f"/plugin/{plugin.SLUG}/api/bulk/apply/",
+                        "csrfToken": get_token(request),
+                        "providers": active_providers,
+                    }
+                ),
+            }
+            return render(request, "inventree_import_plugin/bulk_page.html", context)
+
+    return [
+        path(
+            "api/part/<int:part_id>/providers/", _ProviderStateView.as_view(), name="provider-state"
+        ),
+        path(
+            "api/part/<int:part_id>/preview/<str:provider_slug>/",
+            _PreviewView.as_view(),
+            name="preview-enrich",
+        ),
+        path(
+            "api/part/<int:part_id>/apply/<str:provider_slug>/",
+            _ApplyView.as_view(),
+            name="apply-enrich",
+        ),
+        path("api/bulk/preview/", _BulkPreviewView.as_view(), name="bulk-preview"),
+        path("api/bulk/apply/", _BulkApplyView.as_view(), name="bulk-apply"),
+        path("bulk/", _BulkPageView.as_view(), name="bulk-page"),
+        path("bulk-v2/", _BulkPageView.as_view(), name="bulk-page-v2"),
+    ]