feat: support contextual_tuples in expand requests

openfga · Jan 23, 2025 · 7c42412 · 7c42412
1 parent be9320e
commit 7c42412
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 1 deletion.
diff --git a/config/clients/python/template/src/client/client.py.mustache b/config/clients/python/template/src/client/client.py.mustache
@@ -713,6 +713,10 @@ class OpenFgaClient:
             authorization_model_id=self._get_authorization_model_id(options),
             consistency=self._get_consistency(options),
         )
+        if body.contextual_tuples:
+            req_body.contextual_tuples = ContextualTupleKeys(
+                tuple_keys=convert_tuple_keys(body.contextual_tuples)
+            )
         api_response = {{#asyncio}}await {{/asyncio}}self._api.expand(
             body=req_body,
             **kwargs

diff --git a/config/clients/python/template/src/client/models/expand_request.py.mustache b/config/clients/python/template/src/client/models/expand_request.py.mustache
@@ -1,13 +1,22 @@
 {{>partial_header}}
 
+from {{packageName}}.client.models.tuple import ClientTuple
+
+
 class ClientExpandRequest:
     """
     ClientExpandRequest encapsulates the parameters required to expand request
     """
 
-    def __init__(self, relation: str, object: str):
+    def __init__(
+        self,
+        relation: str,
+        object: str,
+        contextual_tuples: list[ClientTuple] = None,
+    ):
         self._relation = relation
         self._object = object
+        self._contextual_tuples = contextual_tuples
 
     @property
     def relation(self):
@@ -16,9 +25,37 @@ class ClientExpandRequest:
         """
         return self._relation
 
+    @relation.setter
+    def relation(self, value):
+        """
+        Set relation
+        """
+        self._relation = value
+
     @property
     def object(self):
         """
         Return object
         """
         return self._object
+
+    @object.setter
+    def object(self, value):
+        """
+        Set object
+        """
+        self._object = value
+
+    @property
+    def contextual_tuples(self):
+        """
+        Return contextual_tuples
+        """
+        return self._contextual_tuples
+
+    @contextual_tuples.setter
+    def contextual_tuples(self, value):
+        """
+        Set contextual tuples
+        """
+        self._contextual_tuples = value
diff --git a/config/clients/python/template/src/sync/client/client.py.mustache b/config/clients/python/template/src/sync/client/client.py.mustache
@@ -691,6 +691,10 @@ class OpenFgaClient:
             authorization_model_id=self._get_authorization_model_id(options),
                 consistency=self._get_consistency(options),
         )
+        if body.contextual_tuples:
+            req_body.contextual_tuples = ContextualTupleKeys(
+                tuple_keys=convert_tuple_keys(body.contextual_tuples)
+            )
         api_response = self._api.expand(
             body=req_body,
             **kwargs