Require dependencies based on plugin config

CONTRIBUTING.md

@@ -60,7 +60,6 @@ For example:
 - `tox -e coverage` - unit test coverage report.
 - `tox -e mypy` - runs mypy static type checker on the source.
 - `tox -e pyright` - runs pyright static type checker on the source.
-- `tox -e refurb` - runs the [refurb](https://github.com/dosisod/refurb) tool over the source.
 - `tox -e integration` - runs the dockerized integration test suite.
 - `tox` - run everything, you maniac!
 

README.md

@@ -53,6 +53,30 @@ Configuration for a [Starlite](https://github.com/starlite-api/starlite) applica
 - SAQ async worker
 - Lots of features!
 
+## Installation
+
+This will install `starlite-saqlalchemy` with minimal dependencies.
+
+```console
+poetry add starlite-saqlalchemy
+```
+
+You can also install additional dependencies depending on the features you need:
+
+```console
+# Repository implementation, DTOs
+poetry add starlite-saqlalchemy[sqlalchemy]
+# Async worker using saq
+poetry add starlite-saqlalchemy[worker]
+# Redis cache backend
+poetry add starlite-saqlalchemy[cache]
+# Sentry integration for starlite
+poetry add starlite-saqlalchemy[sentry]
+
+# or to install them all:
+poetry add starlite-saqlalchemy[all]
+```
+
 ## Example
 
 ```python

pyproject.toml

@@ -58,20 +58,29 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.10"
 asyncpg = "*"
-hiredis = "*"
 httpx = "*"
 msgspec = "*"
 pydantic = "*"
 python-dotenv = "*"
-redis = "*"
-saq = "^0.9.1"
-sentry-sdk = ">=1.13.0"
-sqlalchemy = "==2.0.0rc2"
 starlite = "^1.40.1"
-structlog = ">=22.2.0"
 tenacity = "*"
 uvicorn = "*"
 uvloop = "*"
+structlog = ">=22.2.0"
+
+# Optionals
+hiredis = { version = "*", optional = true }
+redis = { version = "*", optional = true }
+saq = { version = "^0.9.1", optional = true }
+sentry-sdk = { version = "*", optional = true }
+sqlalchemy = { version = "==2.0.0rc2", optional = true }
+
+[tool.poetry.extras]
+cache = ["redis", "hiredis"]
+worker = ["saq", "hiredis"]
+sentry = ["sentry-sdk"]
+sqlalchemy = ["sqlalchemy"]
+all = ["redis", "hiredis", "saq", "sentry-sdk", "sqlalchemy"]
 
 [tool.poetry.plugins."pytest11"]
 pytest_starlite_saqlalchemy = "pytest_starlite_saqlalchemy"
@@ -92,11 +101,18 @@ add-select = "D401,D404,D417"
 convention = "google"
 
 [tool.pytest.ini_options]
-addopts = ["-ra", "--strict-config"]
+addopts = [
+    "-ra",
+    "--strict-config",
+    # Plugin are enabled in tests/conftest.py to control loading order.
+    "-p",
+    "no:pytest_starlite_saqlalchemy",
+    "-p",
+    "no:pytest_dotenv",
+]
 asyncio_mode = "auto"
 env_files = ["tests.env"]
 testpaths = ["tests/unit"]
-test_app = "tests.utils.app:create_app"
 
 [tool.pylint.main]
 disable = [

requirements.dev-extras.txt

@@ -0,0 +1,7 @@
+-r requirements.dev.txt
+
+sentry-sdk >= "1.13.0"
+hiredis
+redis
+saq >= "0.9.1"
+sqlalchemy == 2.0.0rc2

src/pytest_starlite_saqlalchemy/plugin.py

@@ -2,6 +2,7 @@
 # pylint: disable=import-outside-toplevel
 from __future__ import annotations
 
+import os
 import re
 from typing import TYPE_CHECKING
 from unittest.mock import MagicMock
@@ -12,6 +13,8 @@
 from structlog.testing import CapturingLogger
 from uvicorn.importer import ImportFromStringError, import_from_string
 
+from starlite_saqlalchemy import constants
+
 if TYPE_CHECKING:
     from collections.abc import Generator
 
@@ -35,7 +38,7 @@ def pytest_addoption(parser: Parser) -> None:
         "test_app",
         "Path to application instance, or callable that returns an application instance.",
         type="string",
-        default="app.main:create_app",
+        default=os.environ.get("TEST_APP", "app.main:create_app"),
     )
     parser.addini(
         "unit_test_pattern",
@@ -65,7 +68,7 @@ def _patch_http_close(monkeypatch: MonkeyPatch) -> None:
     monkeypatch.setattr(starlite_saqlalchemy.http, "clients", set())
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture(autouse=constants.IS_SQLALCHEMY_INSTALLED)
 def _patch_sqlalchemy_plugin(is_unit_test: bool, monkeypatch: MonkeyPatch) -> None:
     if is_unit_test:
         from starlite_saqlalchemy import sqlalchemy_plugin
@@ -77,7 +80,7 @@ def _patch_sqlalchemy_plugin(is_unit_test: bool, monkeypatch: MonkeyPatch) -> No
         )
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture(autouse=constants.IS_SAQ_INSTALLED)
 def _patch_worker(is_unit_test: bool, monkeypatch: MonkeyPatch) -> None:
     """We don't want the worker to start for unittests."""
     if is_unit_test:
@@ -94,7 +97,6 @@ def fx_app(pytestconfig: Config, monkeypatch: MonkeyPatch) -> Starlite:
         An application instance, configured via plugin.
     """
     test_app_str = pytestconfig.getini("test_app")
-
     try:
         app_or_callable = import_from_string(test_app_str)
     except (ImportFromStringError, ModuleNotFoundError):

src/starlite_saqlalchemy/__init__.py

@@ -22,48 +22,35 @@ def example_handler() -> dict:
 # this is because pycharm wigs out when there is a module called `exceptions`:
 # noinspection PyCompatibility
 from . import (
-    cache,
     compression,
-    db,
     dependencies,
-    dto,
     exceptions,
     health,
     http,
     log,
     openapi,
-    redis,
     repository,
-    sentry,
     service,
     settings,
-    sqlalchemy_plugin,
     type_encoders,
-    worker,
 )
 from .init_plugin import ConfigureApp, PluginConfig
 
 __all__ = [
     "ConfigureApp",
     "PluginConfig",
-    "cache",
     "compression",
-    "db",
     "dependencies",
-    "dto",
     "exceptions",
     "health",
     "http",
     "log",
     "openapi",
-    "redis",
     "repository",
-    "sentry",
     "service",
     "settings",
-    "sqlalchemy_plugin",
     "type_encoders",
-    "worker",
 ]
 
+
 __version__ = "0.28.1"

src/starlite_saqlalchemy/constants.py

@@ -1,11 +1,50 @@
 """Application constants."""
 from __future__ import annotations
 
+from importlib import import_module
+from typing import TYPE_CHECKING, Any
+
 from starlite_saqlalchemy.settings import app
 from starlite_saqlalchemy.utils import case_insensitive_string_compare
 
+if TYPE_CHECKING:
+    from collections.abc import MutableMapping
+
+    from starlite_saqlalchemy.service import Service
+
+
 IS_TEST_ENVIRONMENT = case_insensitive_string_compare(app.ENVIRONMENT, app.TEST_ENVIRONMENT_NAME)
 """Flag indicating if the application is running in a test environment."""
 
 IS_LOCAL_ENVIRONMENT = case_insensitive_string_compare(app.ENVIRONMENT, app.LOCAL_ENVIRONMENT_NAME)
 """Flag indicating if application is running in local development mode."""
+
+IS_REDIS_INSTALLED = True
+"""Flag indicating if redis module is installed."""
+
+IS_SAQ_INSTALLED = True
+"""Flag indicating if saq module is installed."""
+
+IS_SENTRY_SDK_INSTALLED = True
+"""Flag indicating if sentry_sdk module is installed."""
+
+IS_SQLALCHEMY_INSTALLED = True
+"""Flag indicating if sqlalchemy module is installed."""
+
+
+for package in ("redis", "saq", "sentry_sdk", "sqlalchemy"):
+    try:
+        import_module(package)
+    except ModuleNotFoundError:
+        match package:
+            case "redis":
+                IS_REDIS_INSTALLED = False
+            case "saq":
+                IS_SAQ_INSTALLED = False
+            case "sentry_sdk":
+                IS_SENTRY_SDK_INSTALLED = False
+            case "sqlalchemy":  # pragma: no cover
+                IS_SQLALCHEMY_INSTALLED = False
+
+SERVICE_OBJECT_IDENTITY_MAP: MutableMapping[str, type[Service[Any]]] = {}
+"""Used by the worker to lookup methods for service object callbacks."""

src/starlite_saqlalchemy/exceptions.py

@@ -56,6 +56,17 @@ class AuthorizationError(StarliteSaqlalchemyClientError):
     """A user tried to do something they shouldn't have."""
 
 
+class MissingDependencyError(StarliteSaqlalchemyError, ValueError):
+    """A required dependency is not installed."""
+
+    def __init__(self, module: str, config: str | None = None) -> None:
+        config = config if config else module
+        super().__init__(
+            f"You enabled {config} configuration but package {module!r} is not installed. "
+            f'You may need to run: "poetry install starlite-saqlalchemy[{config}]"'
+        )
+
+
 class HealthCheckConfigurationError(StarliteSaqlalchemyError):
     """An error occurred while registering an health check."""