arthexis 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

core/widgets.py

@@ -0,0 +1,51 @@
+from django import forms
+import json
+
+
+class CopyColorWidget(forms.TextInput):
+    input_type = "color"
+    template_name = "widgets/copy_color.html"
+
+    class Media:
+        js = ["core/copy_color.js"]
+
+
+class CodeEditorWidget(forms.Textarea):
+    """Simple code editor widget for editing recipes."""
+
+    def __init__(self, attrs=None):
+        default_attrs = {"class": "code-editor"}
+        if attrs:
+            default_attrs.update(attrs)
+        super().__init__(attrs=default_attrs)
+
+    class Media:
+        css = {"all": ["core/code_editor.css"]}
+        js = ["core/code_editor.js"]
+
+
+class OdooProductWidget(forms.Select):
+    """Widget for selecting an Odoo product."""
+
+    template_name = "widgets/odoo_product.html"
+
+    class Media:
+        js = ["core/odoo_product.js"]
+
+    def get_context(self, name, value, attrs):
+        attrs = attrs or {}
+        if isinstance(value, dict):
+            attrs["data-current-id"] = str(value.get("id", ""))
+            value = json.dumps(value)
+        elif not value:
+            value = ""
+        return super().get_context(name, value, attrs)
+
+    def value_from_datadict(self, data, files, name):
+        raw = data.get(name)
+        if not raw:
+            return {}
+        try:
+            return json.loads(raw)
+        except Exception:
+            return {}

core/workgroup_urls.py

@@ -1,4 +1,4 @@
-"""URL routes for chat profile endpoints."""
+"""URL routes for assistant profile endpoints."""
 
 from django.urls import path
 
@@ -7,7 +7,11 @@ from . import workgroup_views as views
 app_name = "workgroup"
 
 urlpatterns = [
-    path("chat-profiles/<int:user_id>/", views.issue_key, name="chatprofile-issue"),
+    path(
+        "assistant-profiles/<int:user_id>/",
+        views.issue_key,
+        name="assistantprofile-issue",
+    ),
     path("assistant/test/", views.assistant_test, name="assistant-test"),
+    path("chat/", views.chat, name="chat"),
 ]
-

core/workgroup_views.py

@@ -1,15 +1,17 @@
-"""REST endpoints for ChatProfile issuance and authentication."""
+"""REST endpoints for AssistantProfile issuance and authentication."""
 
 from __future__ import annotations
 
 from functools import wraps
 
+from django.apps import apps
 from django.contrib.auth import get_user_model
+from django.forms.models import model_to_dict
 from django.http import HttpResponse, JsonResponse
 from django.views.decorators.csrf import csrf_exempt
 from django.views.decorators.http import require_GET, require_POST
 
-from .models import ChatProfile, hash_key
+from .models import AssistantProfile, hash_key
 
 
 @csrf_exempt
@@ -21,7 +23,7 @@ def issue_key(request, user_id: int) -> JsonResponse:
     """
 
     user = get_user_model().objects.get(pk=user_id)
-    profile, key = ChatProfile.issue_key(user)
+    profile, key = AssistantProfile.issue_key(user)
     return JsonResponse({"user_id": user_id, "user_key": key})
 
 
@@ -36,11 +38,14 @@ def authenticate(view_func):
 
         key_hash = hash_key(header.split(" ", 1)[1])
         try:
-            profile = ChatProfile.objects.get(user_key_hash=key_hash, is_active=True)
-        except ChatProfile.DoesNotExist:
+            profile = AssistantProfile.objects.get(
+                user_key_hash=key_hash, is_active=True
+            )
+        except AssistantProfile.DoesNotExist:
             return HttpResponse(status=401)
 
         profile.touch()
+        request.assistant_profile = profile
         request.chat_profile = profile
         return view_func(request, *args, **kwargs)
 
@@ -52,6 +57,38 @@ def authenticate(view_func):
 def assistant_test(request):
     """Return a simple greeting to confirm authentication."""
 
-    user_id = request.chat_profile.user_id
+    profile = getattr(request, "assistant_profile", None)
+    user_id = profile.user_id if profile else None
     return JsonResponse({"message": f"Hello from user {user_id}"})
 
+
+@require_GET
+@authenticate
+def chat(request):
+    """Return serialized data from any model.
+
+    Clients must provide ``model`` as ``app_label.ModelName`` and may include a
+    ``pk`` to fetch a specific record. When ``pk`` is omitted, the view returns
+    up to 100 records.
+    """
+
+    model_label = request.GET.get("model")
+    if not model_label:
+        return JsonResponse({"error": "model parameter required"}, status=400)
+    try:
+        model = apps.get_model(model_label)
+    except LookupError:
+        return JsonResponse({"error": "unknown model"}, status=400)
+
+    qs = model.objects.all()
+    pk = request.GET.get("pk")
+    if pk is not None:
+        try:
+            obj = qs.get(pk=pk)
+        except model.DoesNotExist:
+            return JsonResponse({"error": "object not found"}, status=404)
+        data = model_to_dict(obj)
+    else:
+        data = [model_to_dict(o) for o in qs[:100]]
+
+    return JsonResponse({"data": data})

nodes/actions.py

@@ -68,5 +68,3 @@ class CaptureScreenshotAction(NodeAction):
         path = capture_screenshot(url)
         save_screenshot(path, node=node, method="NODE_ACTION")
         return path
-
-