couchbase-connect-py 0.2.0__py3-none-any.whl

couchbase_connect/__init__.py

@@ -0,0 +1,92 @@
+"""Couchbase Server and Capella connection helpers."""
+
+from couchbase_connect.auto import AutoCouchbaseConnect, IllegalStateError
+from couchbase_connect.capella import (
+    CapellaAllowedCIDR,
+    CapellaAPIError,
+    CapellaBucket,
+    CapellaCertificate,
+    CapellaCluster,
+    CapellaConnectivity,
+    CapellaCredentials,
+    CapellaNotFoundError,
+    CapellaOrganization,
+    CapellaProject,
+    CapellaUser,
+    CouchbaseCapella,
+    UserNotConfiguredError,
+    connect_cluster,
+    disconnect_cluster,
+)
+from couchbase_connect.cloud import Capella
+from couchbase_connect.config import (
+    CouchbaseConfig,
+    convert_bucket_type,
+    convert_conflict_resolution_type,
+    convert_storage_backend,
+)
+from couchbase_connect.models import (
+    BucketData,
+    CapellaNodeConfig,
+    ClusterNodeConfig,
+    CollectionData,
+    GroupData,
+    IndexData,
+    RoleData,
+    ScopeData,
+    SearchIndexData,
+    TableData,
+    UserData,
+)
+from couchbase_connect.protocol import (
+    CouchbaseConnect,
+    get_instance,
+    open_connection,
+    resolve,
+)
+from couchbase_connect.server import Server
+from couchbase_connect.stream import CouchbaseStream
+
+__all__ = [
+    "AutoCouchbaseConnect",
+    "BucketData",
+    "Capella",
+    "CapellaAPIError",
+    "CapellaAllowedCIDR",
+    "CapellaBucket",
+    "CapellaCertificate",
+    "CapellaCluster",
+    "CapellaConnectivity",
+    "CapellaCredentials",
+    "CapellaNodeConfig",
+    "CapellaNotFoundError",
+    "CapellaOrganization",
+    "CapellaProject",
+    "CapellaUser",
+    "ClusterNodeConfig",
+    "CollectionData",
+    "CouchbaseCapella",
+    "CouchbaseConfig",
+    "CouchbaseConnect",
+    "CouchbaseStream",
+    "GroupData",
+    "IllegalStateError",
+    "IndexData",
+    "RoleData",
+    "ScopeData",
+    "SearchIndexData",
+    "Server",
+    "TableData",
+    "UserData",
+    "UserNotConfiguredError",
+    "connect_cluster",
+    "convert_bucket_type",
+    "convert_conflict_resolution_type",
+    "convert_storage_backend",
+    "disconnect_cluster",
+    "get_instance",
+    "open_connection",
+    "resolve",
+]
+
+__version__ = "0.2.0"

couchbase_connect/auto.py

@@ -0,0 +1,54 @@
+"""Facade that routes to Server or Capella based on CouchbaseConfig."""
+
+from __future__ import annotations
+
+from typing import Any, Mapping, Optional
+
+from couchbase_connect.config import CouchbaseConfig
+from couchbase_connect.protocol import CouchbaseConnect, resolve
+
+
+class AutoCouchbaseConnect:
+
+    _instance: Optional["AutoCouchbaseConnect"] = None
+
+    def __init__(self) -> None:
+        self._delegate: Optional[CouchbaseConnect] = None
+
+    @classmethod
+    def get_instance(cls) -> "AutoCouchbaseConnect":
+        if cls._instance is None:
+            cls._instance = cls()
+        assert cls._instance is not None
+        return cls._instance
+
+    def _require_delegate(self) -> CouchbaseConnect:
+        if self._delegate is None:
+            raise IllegalStateError(
+                "Call connect(CouchbaseConfig) or create_cluster(CouchbaseConfig) first"
+            )
+        return self._delegate
+
+    def _assign_delegate(self, config: CouchbaseConfig) -> None:
+        self._delegate = resolve(config)
+
+    def connect(self, config: CouchbaseConfig) -> None:
+        self._assign_delegate(config)
+        assert self._delegate is not None
+        self._delegate.connect(config)
+
+    def create_cluster(
+        self,
+        config: CouchbaseConfig,
+        options: Optional[Mapping[str, str]] = None,
+    ) -> None:
+        self._assign_delegate(config)
+        assert self._delegate is not None
+        self._delegate.create_cluster(config, options)
+
+    def __getattr__(self, name: str) -> Any:
+        return getattr(self._require_delegate(), name)
+
+
+class IllegalStateError(RuntimeError):
+    pass