canvas 0.7.1__tar.gz → 0.8.1__tar.gz

{canvas-0.7.1 → canvas-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: canvas
-Version: 0.7.1
+Version: 0.8.1
 Summary: SDK to customize event-driven actions in your Canvas instance
 License: MIT
 Author: Canvas Team
@@ -15,7 +15,7 @@ Requires-Dist: cron-converter (>=1.2.1,<2.0.0)
 Requires-Dist: django (>=5.1.1,<6.0.0)
 Requires-Dist: django-stubs[compatible-mypy] (>=5.1.1,<6.0.0)
 Requires-Dist: django-timezone-utils (>=0.15.0,<0.16.0)
-Requires-Dist: env-tools (>=2.3.0,<3.0.0)
+Requires-Dist: env-tools (>=2.4.0,<3.0.0)
 Requires-Dist: grpcio (>=1.60.1,<2.0.0)
 Requires-Dist: ipython (>=8.21.0,<9.0.0)
 Requires-Dist: jsonschema (>=4.21.1,<5.0.0)

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/apps/auth/tests.py

@@ -8,6 +8,8 @@ from canvas_cli.apps.auth import get_or_request_api_token
 
 @pytest.fixture
 def valid_token_response() -> Any:
+    """Returns a valid token response."""
+
     class TokenResponse:
         status_code = 200
 
@@ -19,6 +21,8 @@ def valid_token_response() -> Any:
 
 @pytest.fixture
 def error_token_response() -> Any:
+    """Returns an error token response."""
+
     class TokenResponse:
         status_code = 500
 
@@ -27,6 +31,8 @@ def error_token_response() -> Any:
 
 @pytest.fixture
 def expired_token_response() -> Any:
+    """Returns an expired token response."""
+
     class TokenResponse:
         status_code = 200
 
@@ -45,6 +51,7 @@ def test_get_or_request_api_token_uses_stored_token(
     mock_get_password: MagicMock,
     valid_token_response: Any,
 ) -> None:
+    """Test that get_or_request_api_token uses a stored token if it is valid."""
     mock_is_token_valid.return_value = True
     mock_get_password.return_value = "a-stored-valid-token"
     mock_post.return_value = valid_token_response
@@ -66,6 +73,7 @@ def test_get_or_request_api_token_requests_token_if_none_stored(
     mock_set_password: MagicMock,
     valid_token_response: Any,
 ) -> None:
+    """Test that get_or_request_api_token requests a new token if none is stored."""
     mock_client_credentials.return_value = "client_id=id&client_secret=secret"
     mock_get_password.return_value = None
     mock_post.return_value = valid_token_response
@@ -95,6 +103,7 @@ def test_get_or_request_api_token_raises_exception_if_error_token_response(
     mock_get_password: MagicMock,
     error_token_response: Any,
 ) -> None:
+    """Test that get_or_request_api_token raises an exception if an error token response is received."""
     mock_client_credentials.return_value = "client_id=id&client_secret=secret"
     mock_get_password.return_value = None
     mock_post.return_value = error_token_response
@@ -123,6 +132,7 @@ def test_get_or_request_api_token_raises_exception_if_expired_token(
     mock_get_password: MagicMock,
     expired_token_response: Any,
 ) -> None:
+    """Test that get_or_request_api_token raises an exception if an expired token is received."""
     mock_client_credentials.return_value = "client_id=id&client_secret=secret"
     mock_get_password.return_value = None
     mock_post.return_value = expired_token_response

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/apps/emit/emit.py

@@ -7,7 +7,6 @@ import grpc
 import typer
 
 from canvas_generated.messages.events_pb2 import Event as PluginRunnerEvent
-from canvas_generated.messages.events_pb2 import EventType as PluginRunnerEventType
 from canvas_generated.services.plugin_runner_pb2_grpc import PluginRunnerStub
 
 

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/apps/logs/logs.py

@@ -1,5 +1,5 @@
 import json
-from typing import Optional, cast
+from typing import cast
 from urllib.parse import urlparse
 
 import typer
@@ -31,9 +31,9 @@ def _on_open(ws: websocket.WebSocket) -> None:
 
 
 def logs(
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host, help="Canvas instance to connect to", default=None
-    )
+    ),
 ) -> None:
     """Listens and prints log streams from the instance."""
     if not host:
@@ -62,4 +62,4 @@ def logs(
         ws.run_forever()
 
     except KeyboardInterrupt:
-        raise typer.Exit(0)
+        raise typer.Exit(0) from None

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/apps/plugin/plugin.py

@@ -2,8 +2,9 @@ import ast
 import json
 import tarfile
 import tempfile
+from collections.abc import Iterable
 from pathlib import Path
-from typing import Any, List, Optional, cast
+from typing import Any, cast
 from urllib.parse import urljoin
 
 import requests
@@ -53,14 +54,15 @@ def _build_package(package: Path) -> Path:
 def _get_name_from_metadata(host: str, token: str, package: Path) -> str | None:
     """Extract metadata from a provided package and return the package name if it exists in the metadata."""
     try:
-        metadata_response = requests.post(
-            plugin_url(host, "extract-metadata"),
-            headers={"Authorization": f"Bearer {token}"},
-            files={"package": open(package, "rb")},
-        )
+        with open(package) as package_file:
+            metadata_response = requests.post(
+                plugin_url(host, "extract-metadata"),
+                headers={"Authorization": f"Bearer {token}"},
+                files={"package": package_file},
+            )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if metadata_response.status_code != requests.codes.ok:
         print(f"Status code {metadata_response.status_code}: {metadata_response.text}")
@@ -116,8 +118,8 @@ def _get_meta_properties(protocol_path: Path, classname: str) -> dict[str, str]:
 
 
 def _get_protocols_with_new_cqm_properties(
-    protocol_classes: List[dict[str, Any]], plugin: Path
-) -> List[dict[str, Any]] | None:
+    protocol_classes: Iterable[dict[str, Any]], plugin: Path
+) -> Iterable[dict[str, Any]] | None:
     """Extract the meta properties of any ClinicalQualityMeasure Protocols included in the plugin if they have changed."""
     has_updates = False
     protocol_props = []
@@ -146,14 +148,14 @@ def init() -> None:
     try:
         project_dir = cookiecutter(str(template))
     except OutputDirExistsException:
-        raise typer.BadParameter(f"The supplied directory already exists")
+        raise typer.BadParameter("The supplied directory already exists") from None
 
     print(f"Project created in {project_dir}")
 
 
 def install(
     plugin_name: Path = typer.Argument(..., help="Path to plugin to install"),
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host,
         help="Canvas instance to connect to",
         default=None,
@@ -181,15 +183,16 @@ def install(
     print(f"Posting {built_package_path.absolute()} to {url}")
 
     try:
-        r = requests.post(
-            url,
-            data={"is_enabled": True},
-            files={"package": open(built_package_path, "rb")},
-            headers={"Authorization": f"Bearer {token}"},
-        )
+        with open(built_package_path, "rb") as package:
+            r = requests.post(
+                url,
+                data={"is_enabled": True},
+                files={"package": package},
+                headers={"Authorization": f"Bearer {token}"},
+            )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if r.status_code == requests.codes.created:
         print(f"Plugin {plugin_name} successfully installed!")
@@ -207,7 +210,7 @@ def install(
 
 def uninstall(
     name: str = typer.Argument(..., help="Plugin name to uninstall"),
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host,
         help="Canvas instance to connect to",
         default=None,
@@ -232,7 +235,7 @@ def uninstall(
         )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if r.status_code == requests.codes.no_content:
         print(f"Plugin {name} successfully uninstalled!")
@@ -243,7 +246,7 @@ def uninstall(
 
 def enable(
     name: str = typer.Argument(..., help="Plugin name to enable"),
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host,
         help="Canvas instance to connect to",
         default=None,
@@ -269,7 +272,7 @@ def enable(
         )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if r.ok:
         print(f"Plugin {name} successfully enabled!")
@@ -280,7 +283,7 @@ def enable(
 
 def disable(
     name: str = typer.Argument(..., help="Plugin name to disable"),
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host,
         help="Canvas instance to connect to",
         default=None,
@@ -306,7 +309,7 @@ def disable(
         )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if r.ok:
         print(f"Plugin {name} successfully disabled!")
@@ -316,11 +319,11 @@ def disable(
 
 
 def list(
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host,
         help="Canvas instance to connect to",
         default=None,
-    )
+    ),
 ) -> None:
     """List all plugins from a Canvas instance."""
     if not host:
@@ -337,7 +340,7 @@ def list(
         )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if r.status_code == requests.codes.ok:
         plugins = r.json().get("results", [])
@@ -382,7 +385,7 @@ def validate_manifest(
 
     except json.JSONDecodeError:
         print("There was a problem loading the manifest file, please ensure it's valid JSON")
-        raise typer.Abort()
+        raise typer.Abort() from None
 
     validate_manifest_file(manifest_json)
 
@@ -391,14 +394,14 @@ def validate_manifest(
 
 def update(
     name: str = typer.Argument(..., help="Plugin name to update"),
-    package: Optional[Path] = typer.Option(
+    package_path: Path | None = typer.Option(
         help="Path to a wheel or sdist file containing the python package to install",
         default=None,
     ),
-    is_enabled: Optional[bool] = typer.Option(
+    is_enabled: bool | None = typer.Option(
         None, "--enable/--disable", show_default=False, help="Enable/disable the plugin"
     ),
-    host: Optional[str] = typer.Option(
+    host: str | None = typer.Option(
         callback=get_default_host,
         help="Canvas instance to connect to",
         default=None,
@@ -408,27 +411,36 @@ def update(
     if not host:
         raise typer.BadParameter("Please specify a host or set a default via the `auth` command")
 
-    if package:
-        validate_package(package)
+    if package_path:
+        validate_package(package_path)
 
     token = get_or_request_api_token(host)
 
-    print(f"Updating plugin {name} from {host} with {is_enabled=}, {package=}")
-
-    binary_package = {"package": open(package, "rb")} if package else None
+    print(f"Updating plugin {name} from {host} with {is_enabled=}, {package_path=}")
 
     url = plugin_url(host, name)
 
     try:
-        r = requests.patch(
-            url,
-            data={"is_enabled": is_enabled} if is_enabled is not None else {},
-            files=binary_package,
-            headers={"Authorization": f"Bearer {token}"},
-        )
+        data = {"is_enabled": is_enabled} if is_enabled is not None else {}
+        headers = {"Authorization": f"Bearer {token}"}
+
+        if package_path:
+            with open(package_path, "rb") as package:
+                r = requests.patch(
+                    url,
+                    data=data,
+                    headers=headers,
+                    files={"package": package},
+                )
+        else:
+            r = requests.patch(
+                url,
+                data=data,
+                headers=headers,
+            )
     except requests.exceptions.RequestException:
         print(f"Failed to connect to {host}")
-        raise typer.Exit(1)
+        raise typer.Exit(1) from None
 
     if r.status_code == requests.codes.ok:
         print("Plugin successfully updated!")

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/apps/plugin/tests.py

@@ -1,7 +1,8 @@
 import shutil
+from collections.abc import Generator
 from datetime import datetime
 from pathlib import Path
-from typing import Any, Generator
+from typing import Any
 
 import pytest
 import typer
@@ -39,12 +40,13 @@ def test_validate_package_valid_file(tmp_path: Path) -> None:
 
 @pytest.fixture(scope="session")
 def init_plugin_name() -> str:
-    """The plugin name to be used for the canvas cli init test"""
+    """The plugin name to be used for the canvas cli init test."""
     return f"testing_init-{datetime.now().timestamp()}".replace(".", "")
 
 
 @pytest.fixture(autouse=True, scope="session")
 def clean_up_plugin(init_plugin_name: str) -> Generator[Any, Any, Any]:
+    """Cleans up the plugin directory after the test."""
     yield
     if Path(f"./{init_plugin_name}").exists():
         shutil.rmtree(Path(f"./{init_plugin_name}"))

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/conftest.py

@@ -19,6 +19,7 @@ def monkeypatch_app_dir(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None
 @pytest.fixture(autouse=True)
 def reset_context_variables() -> None:
     """Reset the context properties to their default value.
+
     This is needed because we cannot build a `reset` method in the CLIContext class,
     because `load_from_file` loads properties dynamically.
     Also since this is a CLI, it's not expected to keep the global context in memory for more than a run,

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/main.py

@@ -1,6 +1,5 @@
 import importlib.metadata
 from pathlib import Path
-from typing import Optional
 
 import typer
 
@@ -62,7 +61,7 @@ def get_or_create_config_file() -> Path:
 
 @app.callback()
 def main(
-    version: Optional[bool] = typer.Option(
+    version: bool | None = typer.Option(
         None, "--version", callback=version_callback, is_eager=True
     ),
 ) -> None:

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/templates/plugins/default/{{ cookiecutter.__project_slug }}/protocols/my_protocol.py

@@ -50,5 +50,4 @@ class Protocol(BaseProtocol):
 
         # Return zero, one, or many effects.
         # Example:
-        # return [Effect(type=EffectType.LOG, payload=json.dumps(payload))]
-        return []
+        return [Effect(type=EffectType.LOG, payload=json.dumps(payload))]

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/tests.py

@@ -1,8 +1,9 @@
 import os
 import shutil
+from collections.abc import Callable, Generator
 from datetime import datetime
 from pathlib import Path
-from typing import Any, Callable, Generator, cast
+from typing import Any, cast
 from unittest.mock import MagicMock, patch
 from urllib.parse import urlparse
 
@@ -20,14 +21,13 @@ runner = CliRunner()
 
 @pytest.fixture(scope="session")
 def plugin_name() -> str:
-    """The plugin name to be used for the canvas cli test"""
+    """The plugin name to be used for the canvas cli test."""
     return f"cli-{datetime.now().timestamp()}".replace(".", "")
 
 
 @pytest.fixture(scope="session")
 def create_or_update_config_auth_file_for_testing(plugin_name: str) -> Generator[None, None, None]:
     """Creates the necessary config file for auth before performing cli tests."""
-
     if not settings.INTEGRATION_TEST_URL:
         raise ImproperlyConfigured("INTEGRATION_TEST_URL is not set")
 
@@ -45,19 +45,27 @@ def create_or_update_config_auth_file_for_testing(plugin_name: str) -> Generator
 
     temp_path = path.parent / "temp_credentials.ini"
 
-    original_content = open(path, "r").read()
-    open(path, "w").writelines(
-        [
-            f"[{host}]\n",
-            f"client_id={client_id}\n",
-            f"client_secret={client_secret}\n",
-        ]
-    )
-    open(temp_path, "a").write(original_content)
+    # Read the original content
+    with open(path) as original_file:
+        original_content = original_file.read()
+
+    # Write new content to the original file
+    with open(path, "w") as original_file:
+        original_file.writelines(
+            [
+                f"[{host}]\n",
+                f"client_id={client_id}\n",
+                f"client_secret={client_secret}\n",
+            ]
+        )
+
+    # Append original content to the temp file
+    with open(temp_path, "a") as temp_file:
+        temp_file.write(original_content)
 
     yield
 
-    with open(temp_path, "r") as temp:
+    with open(temp_path) as temp:
         original_content = temp.read()
         with open(path, "w") as f:
             f.write(original_content)
@@ -66,9 +74,10 @@ def create_or_update_config_auth_file_for_testing(plugin_name: str) -> Generator
 
 @pytest.fixture(autouse=True, scope="session")
 def write_plugin(plugin_name: str) -> Generator[Any, Any, Any]:
+    """Writes a plugin to the file system."""
     runner.invoke(app, "init", input=plugin_name)
-    protocol = open(f"./{plugin_name}/protocols/my_protocol.py", "w")
-    p = """
+
+    protocol_code = """
 from canvas_sdk.events import EventType
 from canvas_sdk.protocols import BaseProtocol
 from logger import log
@@ -81,8 +90,9 @@ class Protocol(BaseProtocol):
         log.info(self.NARRATIVE_STRING)
         return []
 """
-    protocol.write(p)
-    protocol.close()
+
+    with open(f"./{plugin_name}/protocols/my_protocol.py", "w") as protocol:
+        protocol.write(protocol_code)
 
     yield
 
@@ -91,12 +101,12 @@ class Protocol(BaseProtocol):
 
 
 def list_empty_plugins(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 1 - list all plugins"""
+    """Step 1 - list all plugins."""
     return ("list", 0, [], [f"{plugin_name}"])
 
 
 def install_new_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 2 - install a new plugin"""
+    """Step 2 - install a new plugin."""
     return (
         f"install {plugin_name}",
         0,
@@ -111,12 +121,12 @@ def install_new_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]
 
 
 def list_newly_installed_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 3 - list all plugins, including newly installed one"""
+    """Step 3 - list all plugins, including newly installed one."""
     return ("list", 0, [f"{plugin_name}@0.0.1	enabled"], [])
 
 
 def disable_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 4 - disable plugin"""
+    """Step 4 - disable plugin."""
     return (
         f"disable {plugin_name}",
         0,
@@ -126,12 +136,12 @@ def disable_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
 
 
 def list_disabled_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 5 - list disabled plugin"""
+    """Step 5 - list disabled plugin."""
     return ("list", 0, [f"{plugin_name}@0.0.1	disabled"], [])
 
 
 def enable_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 6 - enable the disabled plugin"""
+    """Step 6 - enable the disabled plugin."""
     return (
         f"enable {plugin_name}",
         0,
@@ -141,7 +151,7 @@ def enable_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
 
 
 def uninstall_enabled_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 7 - try to uninstall the enabled plugin"""
+    """Step 7 - try to uninstall the enabled plugin."""
     return (
         f"uninstall {plugin_name}",
         1,
@@ -154,7 +164,7 @@ def uninstall_enabled_plugin(plugin_name: str) -> tuple[str, int, list[str], lis
 
 
 def uninstall_disabled_plugin(plugin_name: str) -> tuple[str, int, list[str], list[str]]:
-    """Step 8 - disable and then uninstall the plugin"""
+    """Step 8 - disable and then uninstall the plugin."""
     runner.invoke(app, f"disable {plugin_name}")
 
     return (
@@ -192,7 +202,7 @@ def test_canvas_list_install_disable_enable_uninstall(
     create_or_update_config_auth_file_for_testing: None,
     step: Callable,
 ) -> None:
-    """Tests that the Canvas CLI can list, install, disable, enable, and uninstall a plugin"""
+    """Tests that the Canvas CLI can list, install, disable, enable, and uninstall a plugin."""
     mock_get_password.return_value = None
     mock_set_password.return_value = None
 

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/utils/context/context.py

@@ -1,7 +1,8 @@
 import functools
 import json
+from collections.abc import Callable
 from pathlib import Path
-from typing import Any, Callable, TypeVar, cast
+from typing import Any, TypeVar, cast
 
 import typer
 
@@ -12,6 +13,7 @@ F = TypeVar("F", bound=Callable)
 
 class CLIContext:
     """Class that handles configuration across the CLI.
+
     Includes methods for:
     * Loading a JSON file with configuration keys into memory.
     * Making a property transient (default, value is not persisted) or persistent (via decorators)
@@ -118,7 +120,7 @@ class CLIContext:
                 success=False,
                 path=str(file),
             )
-            raise typer.Abort()
+            raise typer.Abort() from None
 
         self._config_file_path = file
 

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/utils/context/tests.py

@@ -1,7 +1,7 @@
 import json
 import os
+from collections.abc import Generator
 from pathlib import Path
-from typing import Generator
 
 import pytest
 import typer

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/utils/print/tests.py

@@ -18,7 +18,7 @@ def test_print_json_outputs_valid_json(message: Any, capfd: pytest.CaptureFixtur
     try:
         json.loads(output)
     except ValueError as exc:
-        assert False, f"{output} is not valid json: {exc}"
+        assert AssertionError(f"{output} is not valid json: {exc}")
 
 
 def test_print_json_outputs_kwargs(capfd: pytest.CaptureFixture[str]) -> None:
@@ -33,7 +33,7 @@ def test_print_json_outputs_kwargs(capfd: pytest.CaptureFixture[str]) -> None:
         assert json_dict.get("a_dict") == {"one": 2}
 
     except ValueError as exc:
-        assert False, f"{output} is not valid json: {exc}"
+        assert AssertionError(f"{output} is not valid json: {exc}")
 
 
 def test_print_overrides_default(capfd: pytest.CaptureFixture[str]) -> None:

{canvas-0.7.1 → canvas-0.8.1}/canvas_cli/utils/validators/tests.py

@@ -5,6 +5,7 @@ from canvas_cli.utils.validators import validate_manifest_file
 
 @pytest.fixture
 def protocol_manifest_example() -> dict:
+    """Return a valid protocol manifest example."""
     return {
         "sdk_version": "0.3.1",
         "plugin_version": "1.0.1",
@@ -32,5 +33,5 @@ def protocol_manifest_example() -> dict:
 
 
 def test_manifest_file_schema(protocol_manifest_example: dict) -> None:
-    """Test that no exception raised when a valid manifest file is validated"""
+    """Test that no exception raised when a valid manifest file is validated."""
     validate_manifest_file(protocol_manifest_example)