Feature/async

Pipfile

@@ -23,6 +23,7 @@ asyncmock = "~=0.4.2"
 oauthlib = "~=3.1.0"
 requests-oauthlib = "~=1.3.0"
 zeebe-grpc = "~=1.0.0"
+aiofiles = "~=0.7.0"
 
 [pipenv]
 allow_prereleases = true

README.md

@@ -60,7 +60,7 @@ def example_task(input: str) -> dict:
 async def another_example_task(name: str) -> dict: # Tasks can also be async
     return {"output": f"Hello world, {name} from async task!"}
 
-asyncio.run(worker.work()) # Now every time that a task with type example is called example_task will be called
+asyncio.run(worker.work()) # Now every time that a task with type `example` or `example2` is called, the corresponding function will be called
 ```
 
 Stop a worker:

docs/client_reference.rst

@@ -5,3 +5,7 @@ Client Reference
 .. autoclass:: pyzeebe.ZeebeClient
    :members:
    :undoc-members:
+
+.. autoclass:: pyzeebe.SyncZeebeClient
+   :members:
+   :undoc-members:

docs/exceptions.rst → docs/errors.rst

@@ -22,13 +22,13 @@ All ``pyzeebe`` exceptions inherit from :py:class:`PyZeebeError`
 
 .. autoexception:: pyzeebe.errors.MessageAlreadyExistsError
 
-.. autoexception:: pyzeebe.exceptions.ProcessDefinitionNotFoundError
+.. autoexception:: pyzeebe.errors.ProcessDefinitionNotFoundError
 
-.. autoexception:: pyzeebe.exceptions.ProcessInstanceNotFoundError
+.. autoexception:: pyzeebe.errors.ProcessInstanceNotFoundError
 
-.. autoexception:: pyzeebe.exceptions.ProcessDefinitionHasNoStartEventError
+.. autoexception:: pyzeebe.errors.ProcessDefinitionHasNoStartEventError
 
-.. autoexception:: pyzeebe.exceptions.ProcessInvalidError
+.. autoexception:: pyzeebe.errors.ProcessInvalidError
 
 .. autoexception:: pyzeebe.errors.InvalidJSONError
 

docs/index.rst

@@ -62,4 +62,4 @@ Table Of Contents
     Client <client>
     Worker <worker>
     Decorators <decorators>
-    Exceptions <exceptions>
+    Exceptions <errors>

pyzeebe/__init__.py

@@ -2,8 +2,8 @@
 
 from pyzeebe import errors
 from pyzeebe.client.client import ZeebeClient
-from pyzeebe.credentials.camunda_cloud_credentials import \
-    CamundaCloudCredentials
+from pyzeebe.client.sync_client import SyncZeebeClient  # type: ignore
+from pyzeebe.credentials.camunda_cloud_credentials import CamundaCloudCredentials
 from pyzeebe.credentials.oauth_credentials import OAuthCredentials
 from pyzeebe.job.job import Job
 from pyzeebe.job.job_status import JobStatus

pyzeebe/client/sync_client.py

@@ -0,0 +1,31 @@
+# type: ignore
+
+import asyncio
+from typing import Dict, List, Tuple
+
+from pyzeebe import ZeebeClient
+from pyzeebe.credentials.base_credentials import BaseCredentials
+
+
+class SyncZeebeClient(ZeebeClient):
+    def __init__(self, hostname: str = None, port: int = None, credentials: BaseCredentials = None,
+                 secure_connection: bool = False, max_connection_retries: int = 10):
+        super().__init__(hostname, port, credentials,
+                         secure_connection, max_connection_retries)
+        self.loop = asyncio.get_event_loop()
+
+    def run_process(self, bpmn_process_id: str, variables: Dict = None, version: int = -1) -> int:
+        return self.loop.run_until_complete(super().run_process(bpmn_process_id, variables, version))
+
+    def run_process_with_result(self, bpmn_process_id: str, variables: Dict = None, version: int = -1,
+                                timeout: int = 0, variables_to_fetch: List[str] = None) -> Tuple[int, Dict]:
+        return self.loop.run_until_complete(super().run_process_with_result(bpmn_process_id, variables, version, timeout, variables_to_fetch))
+
+    def cancel_process_instance(self, process_instance_key: int) -> int:
+        return self.loop.run_until_complete(super().cancel_process_instance(process_instance_key))
+
+    def deploy_process(self, *process_file_path: str) -> None:
+        return self.loop.run_until_complete(super().deploy_process(*process_file_path))
+
+    def publish_message(self, name: str, correlation_key: str, variables: Dict = None, time_to_live_in_milliseconds: int = 60000, message_id: str = None) -> None:
+        return self.loop.run_until_complete(super().publish_message(name, correlation_key, variables, time_to_live_in_milliseconds, message_id))

pyzeebe/function_tools/dict_tools.py

@@ -1,7 +1,10 @@
+import functools
+
 from pyzeebe.function_tools import AsyncFunction, DictFunction
 
 
 def convert_to_dict_function(single_value_function: AsyncFunction, variable_name: str) -> DictFunction:
+    @functools.wraps(single_value_function)
     async def inner_fn(*args, **kwargs):
         return {variable_name: await single_value_function(*args, **kwargs)}