dbt-common 1.0.0b2__tar.gz → 1.0.2__tar.gz

{dbt_common-1.0.0b2 → dbt_common-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dbt-common
-Version: 1.0.0b2
+Version: 1.0.2
 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-1.0.2/dbt_common/__about__.py

@@ -0,0 +1 @@
+version = "1.0.2"

dbt_common-1.0.2/dbt_common/constants.py

@@ -0,0 +1,6 @@
+# Prefix which identifies environment variables which contains secrets.
+SECRET_ENV_PREFIX = "DBT_ENV_SECRET"
+
+# Prefix which identifies environment variables that should not be visible
+# via macros, flags, or other user-facing mechanisms.
+PRIVATE_ENV_PREFIX = "DBT_ENV_PRIVATE"

{dbt_common-1.0.0b2 → dbt_common-1.0.2}/dbt_common/context.py

@@ -1,13 +1,14 @@
 from contextvars import ContextVar, copy_context
 from typing import List, Mapping, Optional
 
-from dbt_common.constants import SECRET_ENV_PREFIX
+from dbt_common.constants import PRIVATE_ENV_PREFIX, SECRET_ENV_PREFIX
 
 
 class InvocationContext:
     def __init__(self, env: Mapping[str, str]):
-        self._env = env
+        self._env = {k: v for k, v in env.items() if not k.startswith(PRIVATE_ENV_PREFIX)}
         self._env_secrets: Optional[List[str]] = None
+        self._env_private = {k: v for k, v in env.items() if k.startswith(PRIVATE_ENV_PREFIX)}
         self.recorder = None
         # This class will also eventually manage the invocation_id, flags, event manager, etc.
 
@@ -15,6 +16,10 @@ class InvocationContext:
     def env(self) -> Mapping[str, str]:
         return self._env
 
+    @property
+    def env_private(self) -> Mapping[str, str]:
+        return self._env_private
+
     @property
     def env_secrets(self) -> List[str]:
         if self._env_secrets is None:

{dbt_common-1.0.0b2 → dbt_common-1.0.2}/dbt_common/events/functions.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 from dbt_common.events.event_manager_client import get_event_manager
+from dbt_common.exceptions import EventCompilationError
 from dbt_common.invocation import get_invocation_id
 from dbt_common.helper_types import WarnErrorOptions
 from dbt_common.utils.encoding import ForgivingJSONEncoder
@@ -114,12 +115,10 @@ def msg_to_dict(msg: EventMsg) -> dict:
 
 
 def warn_or_error(event, node=None) -> None:
-    if WARN_ERROR or WARN_ERROR_OPTIONS.includes(type(event).__name__):
-        # TODO: resolve this circular import when at top
-        from dbt_common.exceptions import EventCompilationError
-
+    event_name = type(event).__name__
+    if WARN_ERROR or WARN_ERROR_OPTIONS.includes(event_name):
         raise EventCompilationError(event.message(), node)
-    else:
+    elif not WARN_ERROR_OPTIONS.silenced(event_name):
         fire_event(event)
 
 

{dbt_common-1.0.0b2 → dbt_common-1.0.2}/dbt_common/helper_types.py

@@ -54,7 +54,7 @@ class IncludeExclude(dbtClassMixin):
         if isinstance(self.exclude, list):
             self._validate_items(self.exclude)
 
-    def includes(self, item_name: str):
+    def includes(self, item_name: str) -> bool:
         return (
             item_name in self.include or self.include in self.INCLUDE_ALL
         ) and item_name not in self.exclude
@@ -69,10 +69,22 @@ class WarnErrorOptions(IncludeExclude):
         include: Union[str, List[str]],
         exclude: Optional[List[str]] = None,
         valid_error_names: Optional[Set[str]] = None,
+        silence: Optional[List[str]] = None,
     ):
+        self.silence = silence or []
         self._valid_error_names: Set[str] = valid_error_names or set()
         super().__init__(include=include, exclude=(exclude or []))
 
+    def __post_init__(self):
+        super().__post_init__()
+        self._validate_items(self.silence)
+
+    def includes(self, item_name: str) -> bool:
+        return super().includes(item_name) and not self.silenced(item_name)
+
+    def silenced(self, item_name: str) -> bool:
+        return item_name in self.silence
+
     def _validate_items(self, items: List[str]):
         for item in items:
             if item not in self._valid_error_names:

{dbt_common-1.0.0b2 → dbt_common-1.0.2}/dbt_common/ui.py

@@ -80,3 +80,7 @@ def line_wrap_message(msg: str, subtract: int = 0, dedent: bool = True, prefix:
 
 def warning_tag(msg: str) -> str:
     return f'[{yellow("WARNING")}]: {msg}'
+
+
+def error_tag(msg: str) -> str:
+    return f'[{red("ERROR")}]: {msg}'

dbt_common-1.0.0b2/dbt_common/__about__.py

@@ -1 +0,0 @@
-version = "1.0.0b2"

dbt_common-1.0.0b2/dbt_common/constants.py

@@ -1 +0,0 @@
-SECRET_ENV_PREFIX = "DBT_ENV_SECRET_"