s/temporalio.bridge.temporal_sdk_bridge/rg-replace temporal_sdk_bridge/

Author: dandavison

temporalio/bridge/Cargo.toml

@@ -3,10 +3,6 @@ name = "temporal-sdk-bridge"
 version = "0.1.0"
 edition = "2021"
 
-[package.metadata.maturin]
-module-name = "temporalio.bridge.temporal_sdk_bridge"
-
-
 [lib]
 name = "temporal_sdk_bridge"
 crate-type = ["cdylib"]

temporalio/bridge/client.py

@@ -12,8 +12,8 @@
 import google.protobuf.message
 
 import temporalio.bridge.runtime
-import temporalio.bridge.temporal_sdk_bridge
-from temporalio.bridge.temporal_sdk_bridge import RPCError
+import temporal_sdk_bridge
+from temporal_sdk_bridge import RPCError
 
 
 @dataclass
@@ -94,15 +94,15 @@ async def connect(
         """Establish connection with server."""
         return Client(
             runtime,
-            await temporalio.bridge.temporal_sdk_bridge.connect_client(
+            await temporal_sdk_bridge.connect_client(
                 runtime._ref, config
             ),
         )
 
     def __init__(
         self,
         runtime: temporalio.bridge.runtime.Runtime,
-        ref: temporalio.bridge.temporal_sdk_bridge.ClientRef,
+        ref: temporal_sdk_bridge.ClientRef,
     ):
         """Initialize client with underlying SDK Core reference."""
         self._runtime = runtime

temporalio/bridge/metric.py

@@ -8,7 +8,7 @@
 from typing import Mapping, Optional, Union
 
 import temporalio.bridge.runtime
-import temporalio.bridge.temporal_sdk_bridge
+import temporal_sdk_bridge
 
 
 class MetricMeter:
@@ -17,13 +17,13 @@ class MetricMeter:
     @staticmethod
     def create(runtime: temporalio.bridge.runtime.Runtime) -> Optional[MetricMeter]:
         """Create optional metric meter."""
-        ref = temporalio.bridge.temporal_sdk_bridge.new_metric_meter(runtime._ref)
+        ref = temporal_sdk_bridge.new_metric_meter(runtime._ref)
         if not ref:
             return None
         return MetricMeter(ref)
 
     def __init__(
-        self, ref: temporalio.bridge.temporal_sdk_bridge.MetricMeterRef
+        self, ref: temporal_sdk_bridge.MetricMeterRef
     ) -> None:
         """Initialize metric meter."""
         self._ref = ref
@@ -161,7 +161,7 @@ class MetricAttributes:
     def __init__(
         self,
         meter: MetricMeter,
-        ref: temporalio.bridge.temporal_sdk_bridge.MetricAttributesRef,
+        ref: temporal_sdk_bridge.MetricAttributesRef,
     ) -> None:
         """Initialize attributes."""
         self._meter = meter

temporalio/bridge/runtime.py

@@ -10,7 +10,7 @@
 
 from typing_extensions import Protocol
 
-import temporalio.bridge.temporal_sdk_bridge
+import temporal_sdk_bridge
 
 
 class Runtime:
@@ -19,13 +19,13 @@ class Runtime:
     @staticmethod
     def _raise_in_thread(thread_id: int, exc_type: Type[BaseException]) -> bool:
         """Internal helper for raising an exception in thread."""
-        return temporalio.bridge.temporal_sdk_bridge.raise_in_thread(
+        return temporal_sdk_bridge.raise_in_thread(
             thread_id, exc_type
         )
 
     def __init__(self, *, telemetry: TelemetryConfig) -> None:
         """Create SDK Core runtime."""
-        self._ref = temporalio.bridge.temporal_sdk_bridge.init_runtime(telemetry)
+        self._ref = temporal_sdk_bridge.init_runtime(telemetry)
 
     def retrieve_buffered_metrics(self, durations_as_seconds: bool) -> Sequence[Any]:
         """Get buffered metrics."""

temporalio/bridge/testing.py

@@ -9,7 +9,7 @@
 from typing import Optional, Sequence
 
 import temporalio.bridge.runtime
-import temporalio.bridge.temporal_sdk_bridge
+import temporal_sdk_bridge
 
 
 @dataclass
@@ -53,7 +53,7 @@ async def start_dev_server(
     ) -> EphemeralServer:
         """Start a dev server instance."""
         return EphemeralServer(
-            await temporalio.bridge.temporal_sdk_bridge.start_dev_server(
+            await temporal_sdk_bridge.start_dev_server(
                 runtime._ref, config
             )
         )
@@ -64,12 +64,12 @@ async def start_test_server(
     ) -> EphemeralServer:
         """Start a test server instance."""
         return EphemeralServer(
-            await temporalio.bridge.temporal_sdk_bridge.start_test_server(
+            await temporal_sdk_bridge.start_test_server(
                 runtime._ref, config
             )
         )
 
-    def __init__(self, ref: temporalio.bridge.temporal_sdk_bridge.EphemeralServerRef):
+    def __init__(self, ref: temporal_sdk_bridge.EphemeralServerRef):
         """Initialize an ephemeral server."""
         self._ref = ref
 

temporalio/bridge/worker.py

@@ -29,13 +29,13 @@
 import temporalio.bridge.proto.workflow_activation
 import temporalio.bridge.proto.workflow_completion
 import temporalio.bridge.runtime
-import temporalio.bridge.temporal_sdk_bridge
+import temporal_sdk_bridge
 import temporalio.converter
 import temporalio.exceptions
-from temporalio.bridge.temporal_sdk_bridge import (
+from temporal_sdk_bridge import (
     CustomSlotSupplier as BridgeCustomSlotSupplier,
 )
-from temporalio.bridge.temporal_sdk_bridge import PollShutdownError
+from temporal_sdk_bridge import PollShutdownError
 
 
 @dataclass
@@ -111,7 +111,7 @@ class Worker:
     def create(client: temporalio.bridge.client.Client, config: WorkerConfig) -> Worker:
         """Create a bridge worker from a bridge client."""
         return Worker(
-            temporalio.bridge.temporal_sdk_bridge.new_worker(
+            temporal_sdk_bridge.new_worker(
                 client._runtime._ref, client._ref, config
             )
         )
@@ -120,17 +120,17 @@ def create(client: temporalio.bridge.client.Client, config: WorkerConfig) -> Wor
     def for_replay(
         runtime: temporalio.bridge.runtime.Runtime,
         config: WorkerConfig,
-    ) -> Tuple[Worker, temporalio.bridge.temporal_sdk_bridge.HistoryPusher]:
+    ) -> Tuple[Worker, temporal_sdk_bridge.HistoryPusher]:
         """Create a bridge replay worker."""
         [
             replay_worker,
             pusher,
-        ] = temporalio.bridge.temporal_sdk_bridge.new_replay_worker(
+        ] = temporal_sdk_bridge.new_replay_worker(
             runtime._ref, config
         )
         return Worker(replay_worker), pusher
 
-    def __init__(self, ref: temporalio.bridge.temporal_sdk_bridge.WorkerRef) -> None:
+    def __init__(self, ref: temporal_sdk_bridge.WorkerRef) -> None:
         """Create SDK core worker from a bridge worker."""
         self._ref = ref
 

temporalio/worker/workflow_sandbox/_restrictions.py

@@ -459,7 +459,7 @@ def with_child_unrestricted(self, *child_path: str) -> SandboxMatcher:
     # Must pass through the entire bridge in even the most minimum causes
     # because PyO3 does not allow re-init since 0.17. See
     # https://github.com/PyO3/pyo3/pull/2523.
-    "temporalio.bridge.temporal_sdk_bridge",
+    "temporal_sdk_bridge",
 }
 
 SandboxRestrictions.passthrough_modules_with_temporal = (