dbt-common 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

dbt_common/__about__.py

@@ -1 +1 @@
-version = "0.1.4"
+version = "0.1.6"

dbt_common/context.py

@@ -26,7 +26,7 @@ class InvocationContext:
 _INVOCATION_CONTEXT_VAR: ContextVar[InvocationContext] = ContextVar("DBT_INVOCATION_CONTEXT_VAR")
 
 
-def _reliably_get_invocation_var() -> ContextVar:
+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
     )
@@ -38,11 +38,11 @@ def _reliably_get_invocation_var() -> ContextVar:
 
 
 def set_invocation_context(env: Mapping[str, str]) -> None:
-    invocation_var = _reliably_get_invocation_var()
+    invocation_var = reliably_get_invocation_var()
     invocation_var.set(InvocationContext(env))
 
 
 def get_invocation_context() -> InvocationContext:
-    invocation_var = _reliably_get_invocation_var()
+    invocation_var = reliably_get_invocation_var()
     ctx = invocation_var.get()
     return ctx

dbt_common/events/contextvars.py

@@ -58,12 +58,12 @@ def set_task_contextvars(**kwargs: Any) -> Mapping[str, contextvars.Token]:
 def set_contextvars(prefix: str, **kwargs: Any) -> Mapping[str, contextvars.Token]:
     cvar_tokens = {}
     for k, v in kwargs.items():
-        log_key = f"{prefix}{k}"
+        prefix_key = f"{prefix}{k}"
         try:
-            var = _context_vars[log_key]
+            var = _context_vars[prefix_key]
         except KeyError:
-            var = contextvars.ContextVar(log_key, default=Ellipsis)
-            _context_vars[log_key] = var
+            var = contextvars.ContextVar(prefix_key, default=Ellipsis)
+            _context_vars[prefix_key] = var
 
         cvar_tokens[k] = var.set(v)
 
@@ -73,17 +73,17 @@ def set_contextvars(prefix: str, **kwargs: Any) -> Mapping[str, contextvars.Toke
 # reset by Tokens
 def reset_contextvars(prefix: str, **kwargs: contextvars.Token) -> None:
     for k, v in kwargs.items():
-        log_key = f"{prefix}{k}"
-        var = _context_vars[log_key]
+        prefix_key = f"{prefix}{k}"
+        var = _context_vars[prefix_key]
         var.reset(v)
 
 
 # remove from contextvars
 def unset_contextvars(prefix: str, *keys: str) -> None:
     for k in keys:
-        if k in _context_vars:
-            log_key = f"{prefix}{k}"
-            _context_vars[log_key].set(Ellipsis)
+        prefix_key = f"{prefix}{k}"
+        if prefix_key in _context_vars:
+            _context_vars[prefix_key].set(Ellipsis)
 
 
 # Context manager or decorator to set and unset the context vars

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.4.dist-info → dbt_common-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt-common
-Version: 0.1.4
+Version: 0.1.6
 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
@@ -30,7 +30,7 @@ Requires-Dist: jinja2~=3.0
 Requires-Dist: jsonschema~=4.0
 Requires-Dist: mashumaro[msgpack]~=3.9
 Requires-Dist: pathspec<0.12,>=0.9
-Requires-Dist: protobuf>=4.0.0
+Requires-Dist: protobuf<5.0.0,>=4.0.0
 Requires-Dist: python-dateutil~=2.0
 Requires-Dist: requests<3.0.0
 Requires-Dist: typing-extensions~=4.4

{dbt_common-0.1.4.dist-info → dbt_common-0.1.6.dist-info}/RECORD

@@ -1,7 +1,7 @@
-dbt_common/__about__.py,sha256=tVHHGZCNoOKftKv7v8sDg-78N6kJHq-L6dpD2WYhuH8,18
+dbt_common/__about__.py,sha256=zZ37Z2ygBlDoFI8XD8N8oJHJfXG829j4Pj3mtr-lDDM,18
 dbt_common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt_common/constants.py,sha256=DJIVuTBl5OZpjdFINHon8MNj_XT-Rrp41dAoAuQL2QI,38
-dbt_common/context.py,sha256=ZMmamMuz2CAlIgpoLuNrsl0qGdwPOOQ627R5uBkkvhs,1456
+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
@@ -25,12 +25,12 @@ dbt_common/contracts/config/properties.py,sha256=K8Ut_q4U6pPY1ZJoFKEA6FUCLR1zb2k
 dbt_common/events/README.md,sha256=CSwVajoxCAqOug2UCnXldH1EUK7Kjf3rcq7z9ACjrss,3023
 dbt_common/events/__init__.py,sha256=av08vfpxo0ek7PqZNtMxY8FODJ3xwph4ehRxgInx4LA,383
 dbt_common/events/base_types.py,sha256=E--Fl-KMdJd_UZCN93AuLxnJcQoys2MJjiJxkV03HpM,5448
-dbt_common/events/contextvars.py,sha256=cJaN6FyGUG8CZIclX9D7GgnuWi5eZcPI2GjrnHpEurQ,3050
+dbt_common/events/contextvars.py,sha256=Jx2J_Nkx0DH8EE1Sno_mbv7PMLBfCzm7byeFlUeflLA,3079
 dbt_common/events/event_handler.py,sha256=jfi0PyqIOGnXCG9HEa0VIVULqNvXs1RYmAg0b50ChQs,1385
 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.4.dist-info/METADATA,sha256=JL0lNeTie1Lv_UaFX_IZSeJ2YA6xPanXPzHxrVmNR2Q,3648
-dbt_common-0.1.4.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-dbt_common-0.1.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dbt_common-0.1.4.dist-info/RECORD,,
+dbt_common-0.1.6.dist-info/METADATA,sha256=l7QdlkHqicJ_xG4YQRoWBw2yrq1kOUBfPpj9N8ikzLQ,3655
+dbt_common-0.1.6.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+dbt_common-0.1.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dbt_common-0.1.6.dist-info/RECORD,,