dbt-common 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

dbt_common/__about__.py

@@ -1 +1 @@
-version = "0.1.3"
+version = "0.1.5"

dbt_common/context.py

@@ -1,4 +1,4 @@
-from contextvars import ContextVar
+from contextvars import ContextVar, copy_context
 from typing import List, Mapping, Optional
 
 from dbt_common.constants import SECRET_ENV_PREFIX
@@ -26,10 +26,23 @@ class InvocationContext:
 _INVOCATION_CONTEXT_VAR: ContextVar[InvocationContext] = ContextVar("DBT_INVOCATION_CONTEXT_VAR")
 
 
+def reliably_get_invocation_var() -> ContextVar:
+    invocation_var: Optional[ContextVar] = next(
+        (cv for cv in copy_context() if cv.name == _INVOCATION_CONTEXT_VAR.name), None
+    )
+
+    if invocation_var is None:
+        invocation_var = _INVOCATION_CONTEXT_VAR
+
+    return invocation_var
+
+
 def set_invocation_context(env: Mapping[str, str]) -> None:
-    _INVOCATION_CONTEXT_VAR.set(InvocationContext(env))
+    invocation_var = reliably_get_invocation_var()
+    invocation_var.set(InvocationContext(env))
 
 
 def get_invocation_context() -> InvocationContext:
-    ctx = _INVOCATION_CONTEXT_VAR.get()
+    invocation_var = reliably_get_invocation_var()
+    ctx = invocation_var.get()
     return ctx

dbt_common/events/functions.py

@@ -132,13 +132,6 @@ def fire_event_if(
         fire_event(lazy_e(), level=level)
 
 
-# a special case of fire_event_if, to only fire events in our unit/functional tests
-def fire_event_if_test(
-    lazy_e: Callable[[], BaseEvent], level: Optional[EventLevel] = None
-) -> None:
-    fire_event_if(conditional=("pytest" in sys.modules), lazy_e=lazy_e, level=level)
-
-
 # top-level method for accessing the new eventing system
 # this is where all the side effects happen branched by event type
 # (i.e. - mutating the event history, printing to stdout, logging

dbt_common/utils/executor.py

@@ -1,7 +1,10 @@
 import concurrent.futures
 from contextlib import contextmanager
+from contextvars import ContextVar
 from typing import Protocol, Optional
 
+from dbt_common.context import get_invocation_context, reliably_get_invocation_var
+
 
 class ConnectingExecutor(concurrent.futures.Executor):
     def submit_connected(self, adapter, conn_name, func, *args, **kwargs):
@@ -60,8 +63,17 @@ class HasThreadingConfig(Protocol):
     threads: Optional[int]
 
 
+def _thread_initializer(invocation_context: ContextVar) -> None:
+    invocation_var = reliably_get_invocation_var()
+    invocation_var.set(invocation_context)
+
+
 def executor(config: HasThreadingConfig) -> ConnectingExecutor:
     if config.args.single_threaded:
         return SingleThreadedExecutor()
     else:
-        return MultiThreadedExecutor(max_workers=config.threads)
+        return MultiThreadedExecutor(
+            max_workers=config.threads,
+            initializer=_thread_initializer,
+            initargs=(get_invocation_context(),),
+        )

{dbt_common-0.1.3.dist-info → dbt_common-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt-common
-Version: 0.1.3
+Version: 0.1.5
 Summary: The shared common utilities that dbt-core and adapter implementations use
 Project-URL: Homepage, https://github.com/dbt-labs/dbt-common
 Project-URL: Repository, https://github.com/dbt-labs/dbt-common.git

{dbt_common-0.1.3.dist-info → dbt_common-0.1.5.dist-info}/RECORD

@@ -1,7 +1,7 @@
-dbt_common/__about__.py,sha256=-nFBox60-Jsr0CGYZKCL_Z7JdKnR04wJVcsqACtoUaI,18
+dbt_common/__about__.py,sha256=P4QVYS_PjOFh5NqXToCnBO5ztkaEXiPmgf0LfaGzTcM,18
 dbt_common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt_common/constants.py,sha256=DJIVuTBl5OZpjdFINHon8MNj_XT-Rrp41dAoAuQL2QI,38
-dbt_common/context.py,sha256=r4LJR09L4zw6uYGN8DPvE7IGwOqxl3FPM5G5HnnDCok,1054
+dbt_common/context.py,sha256=XaxjooEU8uHp61TZqZ9jGmKtjjQm81nxJnSjdJmEo8E,1453
 dbt_common/dataclass_schema.py,sha256=CRSQQmzY4QIBzL54aomPT34o3u8T6Zu6W1XZmWfKs-A,5254
 dbt_common/helper_types.py,sha256=jW5Yv0Gwx4M14IEX4RID7k4s0hJiKHIzU8CxACQ1Y7k,3491
 dbt_common/invocation.py,sha256=Zw8jRPn75oi2VrUD6qGvaCDtSyIfqm5pJlPpRjs3s1E,202
@@ -30,7 +30,7 @@ dbt_common/events/event_handler.py,sha256=jfi0PyqIOGnXCG9HEa0VIVULqNvXs1RYmAg0b5
 dbt_common/events/event_manager.py,sha256=ZXJ-N4NsfQTnkEpvuedTjHD8Gt8DgYHbz601eTVCm-o,2175
 dbt_common/events/event_manager_client.py,sha256=etDlUv-3iIwM5IXXuzc1cmQID2fHOXw8dE9snp39-jk,1014
 dbt_common/events/format.py,sha256=x1RWDZ8G7ZMHmxdld6Q4VXca4kvnhiQOIaQXkC6Uo0Q,1609
-dbt_common/events/functions.py,sha256=z6QVFRAv7HdtzyOxcdkWdOeY_tUkkx_HPDLKCM44hkY,5034
+dbt_common/events/functions.py,sha256=TfHUn8nfosEGfIik4pyDDnJDI0HrQ899eH9DiVdINJQ,4756
 dbt_common/events/helpers.py,sha256=CfsWwNDjsLJkPIgOtAfuLEnZ3rGUKeYsH8aDtCW12OA,410
 dbt_common/events/interfaces.py,sha256=hEDeDoB0FW2RYHVZBG7gebEt_mUVBzkn1yPubpaxs-s,147
 dbt_common/events/logger.py,sha256=6LLH8aLoRziFLBmwu-TCR5-EMpBtD28pSaL0FmKpeGA,6159
@@ -51,10 +51,10 @@ dbt_common/utils/casting.py,sha256=TVWHO5ct-foNBH-pj4vDZ84jVU-hPQll52X9YSjz2Zs,6
 dbt_common/utils/connection.py,sha256=J5zxWFpGHpUbe-F04S9bVLLO3HLvHB6INOkAey0k76c,1373
 dbt_common/utils/dict.py,sha256=unI-kJs2yvw7ic2U71-6H8koVRknkb0jCk7W_FCq6_I,4014
 dbt_common/utils/encoding.py,sha256=6_kSY2FvGNYMg7oX7PrbvVioieydih3Kl7Ii802LaHI,1705
-dbt_common/utils/executor.py,sha256=f8TqNi8PpRI-lC0Qu2Mlfi5BeJEn-Vgk4PM_FnO94dU,2016
+dbt_common/utils/executor.py,sha256=W-DtjSPdbnm5bssT3EUo6SbGVi3VePD4LcvgKH1NhZM,2414
 dbt_common/utils/formatting.py,sha256=JUn5rzJ-uajs9wPCN0-f2iRFY1pOJF5YjTD9dERuLoc,165
 dbt_common/utils/jinja.py,sha256=XNfZHuZhLM_R_yPmzYojPm6bF7QOoxIjSWrkJRw6wks,965
-dbt_common-0.1.3.dist-info/METADATA,sha256=n4mFnZzRhtMK8I9_uQY07szZ3Aiuny-TttOdJHXXeBY,3648
-dbt_common-0.1.3.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-dbt_common-0.1.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dbt_common-0.1.3.dist-info/RECORD,,
+dbt_common-0.1.5.dist-info/METADATA,sha256=EUQG5E4N0_IWbAqrYj3gjoA6Fq2LD0tguh-qzyzb7pI,3648
+dbt_common-0.1.5.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+dbt_common-0.1.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dbt_common-0.1.5.dist-info/RECORD,,