feat(python-sdk): collection files (#46)

Save, load, delete, tag, list and filter files in collections.

.pre-commit-config.yaml

@@ -17,6 +17,7 @@ repos:
         entry: "mypy --strict ./python"
         pass_filenames: false
         additional_dependencies:
+          - "types-requests"
           - "pytest-asyncio"
           - "pydantic"
           - "marimo"

pyproject.toml

@@ -11,6 +11,7 @@ dependencies = [
     "typing-extensions>=4.12.2",
     "ariadne-codegen[subscriptions]>=0.14.0",
     "plotly>4",
+    "requests>=2.32.3",
 ]
 
 [project.optional-dependencies]
@@ -20,6 +21,7 @@ dev = [
     "pytest-asyncio==0.24.0",
     "ruff==0.6.4",
     "mypy==1.11.2",
+    "types-requests>=2.32.0",
     "coverage==7.6.1",
     "twine==5.1.1",
     "marimo==0.8.14",

python/queries.gql

@@ -17,6 +17,21 @@ fragment CollectionNotFound on CollectionNotFound {
   id
 }
 
+fragment CollectionFileReference on CollectionFile {
+  id
+  key
+  downloadURL
+  uploadURL
+  tags {
+    key
+    value
+  }
+}
+
+fragment CollectionFileNotFound on CollectionFileNotFound {
+  id
+}
+
 mutation CollectionCreate($organizationID: ID!, $key: ID!, $parentID: ID) {
   collectionCreate(
     organizationID: $organizationID
@@ -142,3 +157,80 @@ query CollectionDocuments(
     }
   }
 }
+
+mutation CollectionFileCreate($collectionID: ID!, $key: ID!) {
+  collectionFileCreate(collectionID: $collectionID, key: $key) {
+    __typename
+    ... on CollectionFile {
+      ...CollectionFileReference
+    }
+  }
+}
+
+mutation CollectionFileDelete($id: ID!) {
+  collectionFileDelete(id: $id) {
+    __typename
+    ... on CollectionFile {
+      ...CollectionFileReference
+    }
+    ... on CollectionFileNotFound {
+      ...CollectionFileNotFound
+    }
+  }
+}
+
+query CollectionFiles(
+  $collectionID: ID!
+  $tag: TagInput
+  $after: ID
+  $first: Int
+) {
+  collection(id: $collectionID) {
+    __typename
+    ... on Collection {
+      id
+      key
+      files(after: $after, first: $first, tag: $tag) {
+        edges {
+          node {
+            __typename
+            ... on CollectionFile {
+              ...CollectionFileReference
+            }
+          }
+        }
+        pageInfo {
+          hasNextPage
+          endCursor
+        }
+      }
+    }
+  }
+}
+
+mutation CollectionFileTagAdd($id: ID!, $tag: TagInput!) {
+  collectionFileTagAdd(id: $id, tag: $tag) {
+    __typename
+    ... on CollectionFile {
+      ...CollectionFileReference
+    }
+  }
+}
+
+mutation CollectionFileTagDelete($id: ID!, $tag_key: String!) {
+  collectionFileTagDelete(id: $id, key: $tag_key) {
+    __typename
+    ... on CollectionFile {
+      ...CollectionFileReference
+    }
+  }
+}
+
+query CollectionFile($id: ID!) {
+  collectionFile(id: $id) {
+    __typename
+    ... on CollectionFile {
+      ...CollectionFileReference
+    }
+  }
+}