indico_toolkit 10.1.0__tar.gz → 10.2.0__tar.gz

{indico_toolkit-10.1.0 → indico_toolkit-10.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: indico-toolkit
-Version: 10.1.0
+Version: 10.2.0
 Summary: Indico Toolkit for Intake and Insights
 License-Expression: MIT
 Requires-Dist: asyncstdlib>=3.14.0,<4.0.0

{indico_toolkit-10.1.0 → indico_toolkit-10.2.0}/pyproject.toml

@@ -8,7 +8,7 @@ description = "Indico Toolkit for Intake and Insights"
 license = "MIT"
 readme = "README.md"
 requires-python = ">=3.11,<4"
-version = "10.1.0"
+version = "10.2.0"
 dependencies = [
     "asyncstdlib (>=3.14.0,<4.0.0)",
     "httpx (>=0.28.1,<0.29.0)",

{indico_toolkit-10.1.0 → indico_toolkit-10.2.0}/pyproject.toml.orig

@@ -9,7 +9,7 @@ license = "MIT"
 readme = "README.md"
 urls = { source = "https://github.com/IndicoDelivery/indico-toolkit-python" }
 requires-python = ">=3.11,<4"
-version = "10.1.0"
+version = "10.2.0"
 dependencies = [
     "asyncstdlib (>=3.14.0,<4.0.0)",
     "httpx (>=0.28.1,<0.29.0)",

{indico_toolkit-10.1.0 → indico_toolkit-10.2.0}/src/indico_toolkit/orm/submissions.py

@@ -46,25 +46,30 @@ def load(submission_class: type[Submission], submission_id: int) -> GraphqlItera
     Yield GraphQL queries to load the fields specified on dataclass `submission_class`
     for submission ID `submission_id`.
     """
+    fields = get_fields(submission_class)
     yield (
         textwrap.dedent("""
             query load(
                 $submissionId: ID!
                 $fieldIds: [String!]!
+                $limit: Int
             ) {
                 submission(id: $submissionId) {
-                    fields(fieldIds: $fieldIds) {
+                    fields(
+                        fieldIds: $fieldIds
+                        limit: $limit
+                    ) {
                         field {
                             fieldId
                         }
                         values(
-                            filters: [
-                                {
-                                    field: "status"
-                                    op: EQUAL
+                            filters: {
+                                match: {
+                                    property: "status"
+                                    operator: EQUAL
                                     textValues: ["ACCEPTED", "BUILTIN"]
                                 }
-                            ]
+                            }
                         ) {
                             id
                             value
@@ -75,7 +80,8 @@ def load(submission_class: type[Submission], submission_id: int) -> GraphqlItera
         """).strip(),
         {
             "submissionId": submission_id,
-            "fieldIds": [field.id for field in get_fields(submission_class)],
+            "fieldIds": [field.id for field in fields],
+            "limit": len(fields),
         },
     )
 

{indico_toolkit-10.1.0 → indico_toolkit-10.2.0}/src/indico_toolkit/router/__init__.py

@@ -8,11 +8,13 @@ from .annotations import (
     ExportId,
     FieldIds,
     FileNameBytes,
+    MergedSubmissionId,
     OrmIter,
     SubmissionId,
     TableId,
 )
 from .create_submission import CreateSubmission, on_create_submission
+from .merge_submission import MergeSubmission, on_merge_submission
 from .start_export import (
     StartExport,
     on_start_custom_export,
@@ -30,7 +32,10 @@ __all__ = (
     "ExportId",
     "FieldIds",
     "FileNameBytes",
+    "MergedSubmissionId",
+    "MergeSubmission",
     "on_create_submission",
+    "on_merge_submission",
     "on_start_custom_export",
     "on_start_export",
     "on_start_json_export",
@@ -52,6 +57,7 @@ _ROUTE_MAPPING = {
     "new_submission": CreateSubmission,
     "status_change": UpdateStatus,
     "table_update": UpdateTable,
+    "submission_merge": MergeSubmission,
 }
 
 

{indico_toolkit-10.1.0 → indico_toolkit-10.2.0}/src/indico_toolkit/router/annotations.py

@@ -14,6 +14,7 @@ _AsyncKwargsIter: TypeAlias = AsyncIterator[tuple[str, Any]]
 DocumentId = NewType("DocumentId", int)
 ExportId = NewType("ExportId", int)
 FieldIds = NewType("FieldIds", frozenset[str])
+MergedSubmissionId = NewType("MergedSubmissionId", int)
 SubmissionId = NewType("SubmissionId", int)
 TableId = NewType("TableId", str)
 
@@ -48,6 +49,12 @@ def resolve_field(callback: Callback, signal: Any) -> _KwargsIter:
             )
 
 
+def resolve_merged_submission(callback: Callback, signal: Any) -> _KwargsIter:
+    for annotation_name, annotation_type in _annotations(callback):
+        if annotation_type is MergedSubmissionId:
+            yield annotation_name, signal["merged_submission_id"]
+
+
 def resolve_microclient(callback: Callback, client: MicroClient) -> _KwargsIter:
     for annotation_name, annotation_type in _annotations(callback):
         if annotation_type is MicroClient:

indico_toolkit-10.2.0/src/indico_toolkit/router/merge_submission.py

@@ -0,0 +1,43 @@
+from collections.abc import Callable
+from dataclasses import dataclass
+from typing import Any, Awaitable, Self, TypeVar
+
+import asyncstdlib as a
+
+from .. import orm
+from ..microclient import MicroClient
+from . import annotations
+from .annotations import AsyncOrmIter, OrmIter
+from .routes import Callback, Route, register
+
+
+@dataclass(frozen=True)
+class MergeSubmission(Route):
+    @classmethod
+    def from_signal(cls, signal: Any) -> Self:
+        return cls()
+
+    def matches(self, other: Any) -> bool:
+        return isinstance(other, type(self))
+
+    async def run(self, callback: Callback, client: MicroClient, signal: Any) -> None:
+        kwargs = await a.dict(
+            a.chain(
+                annotations.resolve_merged_submission(callback, signal),
+                annotations.resolve_microclient(callback, client),
+                annotations.resolve_orm(callback, client, signal),
+                annotations.resolve_submission(callback, signal),
+            )
+        )
+
+        changes = await a.list((await a.sync(callback)(**kwargs)) or [])
+
+        for query, variables in orm.save(changes, signal["submission_id"]):
+            await client.graphql(query, variables)
+
+
+T = TypeVar("T", bound=Callable[..., OrmIter | AsyncOrmIter | None | Awaitable[None]])
+
+
+def on_merge_submission() -> Callable[[T], T]:
+    return register(MergeSubmission())