dbt-common 1.29.0__py3-none-any.whl → 1.31.0__py3-none-any.whl

dbt_common/__about__.py

@@ -1 +1 @@
-version = "1.29.0"
+version = "1.31.0"

dbt_common/clients/system.py

@@ -600,6 +600,9 @@ def rename(from_path: str, to_path: str, force: bool = False) -> None:
     to_path = convert_path(to_path)
     is_symlink = path_is_symlink(to_path)
 
+    if from_path == to_path:
+        return
+
     if os.path.exists(to_path) and force:
         if is_symlink:
             remove_file(to_path)

dbt_common/invocation.py

@@ -1,8 +1,8 @@
 import uuid
-from datetime import datetime
+from datetime import datetime, timezone
 
 _INVOCATION_ID = str(uuid.uuid4())
-_INVOCATION_STARTED_AT = datetime.utcnow()
+_INVOCATION_STARTED_AT = datetime.now(timezone.utc).replace(tzinfo=None)
 
 
 def get_invocation_id() -> str:
@@ -16,4 +16,4 @@ def get_invocation_started_at() -> datetime:
 def reset_invocation_id() -> None:
     global _INVOCATION_ID, _INVOCATION_STARTED_AT
     _INVOCATION_ID = str(uuid.uuid4())
-    _INVOCATION_STARTED_AT = datetime.utcnow()
+    _INVOCATION_STARTED_AT = datetime.now(timezone.utc).replace(tzinfo=None)

dbt_common/record.py

@@ -179,6 +179,7 @@ class Recorder:
 
         self._record_added = False
         self._recording_file: Optional[TextIO] = None
+        self._recording_file_lock = Lock()
         if mode == RecorderMode.RECORD and not in_memory:
             self._recording_file = open(current_recording_path, "w")
             self._recording_file.write("[")
@@ -200,16 +201,19 @@ class Recorder:
             self._counter += 1
 
         if self._recording_file is not None:
-            if self._record_added:
-                self._recording_file.write(",")
-            try:
-                dct = Recorder._get_tagged_dict(record, rec_cls_name)
-                json.dump(dct, self._recording_file)
-                self._record_added = True
-            except Exception:
-                json.dump(
-                    {"type": "RecordingError", "record_type": rec_cls_name}, self._recording_file
-                )
+            # Lock recording file during streamed recording to avoid race conditions across recording threads
+            with self._recording_file_lock:
+                if self._record_added:
+                    self._recording_file.write(",")
+                try:
+                    dct = Recorder._get_tagged_dict(record, rec_cls_name)
+                    json.dump(dct, self._recording_file)
+                    self._record_added = True
+                except Exception:
+                    json.dump(
+                        {"type": "RecordingError", "record_type": rec_cls_name},
+                        self._recording_file,
+                    )
         else:
             if rec_cls_name not in self._records_by_type:
                 self._records_by_type[rec_cls_name] = []
@@ -444,8 +448,9 @@ class AutoValues(DataClassJSONMixin):
     def _to_dict(self):
         return self.to_dict()
 
-    def _from_dict(self, data):
-        return self.from_dict(data)
+    @classmethod
+    def _from_dict(cls, data):
+        return cls.from_dict(data)
 
 
 def _record_function_inner(
@@ -530,16 +535,37 @@ def _record_function_inner(
             else:
                 param_args = (getattr(args[0], id_field_name),) + param_args
 
-        params = record_type.params_cls(*param_args, **kwargs)
+        # Build params - this can be dangerous if a subclass overrides the method in such a way that
+        # changes the signature of the base recorded method, and so is wrapped in a try/except.
+        params = None
+        try:
+            # Omits any additional properties that are not fields of the params class
+            params_dict = {
+                field.name: value
+                for field, value in zip(dataclasses.fields(record_type.params_cls), param_args)
+            }
+            params_dict.update(kwargs)
+            try:
+                params = record_type.params_cls._from_dict(params_dict)
+            except (TypeError, AttributeError):
+                params = record_type.params_cls(*param_args, **kwargs)
+            except Exception:
+                # Unfortunately it is not possible to fire an event here because it would cause a circular import
+                # This means we lose visibility into the issue, but it is better than crashing the entire node or command
+                pass
+        except Exception:
+            # Unfortunately it is not possible to fire an event here because it would cause a circular import
+            # This means we lose visibility into the issue, but it is better than crashing the entire node or command
+            pass
 
         include = True
-        if hasattr(params, "_include"):
+        if params is not None and hasattr(params, "_include"):
             include = params._include()
 
         if not include:
             return func_to_record(*call_args, **kwargs)
 
-        if recorder.mode == RecorderMode.REPLAY:
+        if recorder.mode == RecorderMode.REPLAY and params is not None:
             return recorder.expect_record(params)
         if RECORDED_BY_HIGHER_FUNCTION.get():
             return func_to_record(*call_args, **kwargs)
@@ -554,7 +580,8 @@ def _record_function_inner(
             else record_type.result_cls(r)
         )
         RECORDED_BY_HIGHER_FUNCTION.set(False)
-        recorder.add_record(record_type(params=params, result=result))
+        if params is not None:
+            recorder.add_record(record_type(params=params, result=result))
         return r
 
     setattr(

{dbt_common-1.29.0.dist-info → dbt_common-1.31.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbt-common
-Version: 1.29.0
+Version: 1.31.0
 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.29.0.dist-info → dbt_common-1.31.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
-dbt_common/__about__.py,sha256=_rctfPDph1ATziwQ_Kw-ESD20nyypxJbuBbphV5sTK8,19
+dbt_common/__about__.py,sha256=GY1J5mpmws_U4-5YJ7wyu27HoxTMfH6V135Ll8xlp6k,19
 dbt_common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt_common/behavior_flags.py,sha256=hQzxCqQSweJbRp_xoQqNnlUF77PBuOdCdLOSdcBlkxk,4885
 dbt_common/constants.py,sha256=6ATQPXft7vw_vyhnmkGy3geKfuzWC3ZiUH4kuNuGQ5w,381
 dbt_common/context.py,sha256=w0XfTvQNxtuctHm8hxsdySFoInggBMQm2-7faa_7ObU,3317
 dbt_common/dataclass_schema.py,sha256=u2S0dxwxIghv8RMqC91HlWZJVxmsC_844yZQaGyOwdY,5563
 dbt_common/helper_types.py,sha256=wxbbjfsvHVraWexK88PEY3q_dAYqlPl6yOIXnpxCcyY,11569
-dbt_common/invocation.py,sha256=xw0NBIE-6LHd135cx4non-MkGGsia0mYN0lMkmNEucE,435
+dbt_common/invocation.py,sha256=PzWsS7TJSfLzG1_Y3dq1fmfTDuZAjUN7794DglJqTTM,505
 dbt_common/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dbt_common/record.py,sha256=DTVNnQhBOJvhpPF1xpoEdxbS7N7grPbaFHiFE-vjlMA,22375
+dbt_common/record.py,sha256=ybe4Kyy9FJ_uc1K-0tk6Gv2cbD8kpdVIAy5APbrBWfQ,23965
 dbt_common/semver.py,sha256=Znewz6tc_NBpXr4mZf20bK_RayPL4ODrnxDbkUZrrRo,15034
 dbt_common/tests.py,sha256=6lC_JuRtoYO6cbAF8-R5aTM4HtQiM_EH8X5m_97duGY,315
 dbt_common/ui.py,sha256=geIk3ElFj5y_SG5EDc6RMJFEvYGdo-6C21l0pB8I-EU,2934
@@ -15,7 +15,7 @@ dbt_common/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 dbt_common/clients/_jinja_blocks.py,sha256=eUjZLBtLCASJRRj1G8b9u15A3PQMDCKvytIdWkTLSP4,15451
 dbt_common/clients/agate_helper.py,sha256=anKKgKV5PSnFRuFeBwRMdHK3JCaQoUqB2ZXHD0su0Wo,9123
 dbt_common/clients/jinja.py,sha256=wKwBCtIGKXOwU-KWWF2sz77nST-JUP26U_Zo_a6mNe8,22756
-dbt_common/clients/system.py,sha256=aoUBtOuXVmkOyj6IhhJ3Y4a7JFzPO2F_zKyOtz3xy44,23932
+dbt_common/clients/system.py,sha256=fn1bpskrOnp0fFz6QeOWzVD2jmLX7ShhueUe-2bgqEE,23977
 dbt_common/contracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt_common/contracts/constraints.py,sha256=_f1q3Rkcg2UwA7zI5XBbUMXnPUg6pw17UC7l1HyhyQM,1352
 dbt_common/contracts/metadata.py,sha256=K_M06Rue0wmrQhFP_mq3uvQszq10CIt93oGiAVgbRfE,1293
@@ -56,7 +56,7 @@ dbt_common/utils/encoding.py,sha256=6_kSY2FvGNYMg7oX7PrbvVioieydih3Kl7Ii802LaHI,
 dbt_common/utils/executor.py,sha256=pNY0UbPlwQmTE69Vt_Rj91YGCIOEaqeYU3CjAds0T70,2454
 dbt_common/utils/formatting.py,sha256=JUn5rzJ-uajs9wPCN0-f2iRFY1pOJF5YjTD9dERuLoc,165
 dbt_common/utils/jinja.py,sha256=JXgNmJArGGy0h7qkbNLA3zaEQmoF1CxsNBYTlIwFXDw,1101
-dbt_common-1.29.0.dist-info/METADATA,sha256=WZmO2MH7Ay_LGjHIr-i6wNdcDOJnyt4jdsdk3VLYjm0,4988
-dbt_common-1.29.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dbt_common-1.29.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dbt_common-1.29.0.dist-info/RECORD,,
+dbt_common-1.31.0.dist-info/METADATA,sha256=x-HIsYSHb3wM0wLFUwpa6dvgQLJGRmUzcj4qBHAfZBA,4988
+dbt_common-1.31.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dbt_common-1.31.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dbt_common-1.31.0.dist-info/RECORD,,