chore: Enable B (flake8-bugbear) Ruff checks (#1551)

pyproject.toml

@@ -230,6 +230,7 @@ select = [
     "YTT",  # flake8-2020
     "ANN",  # flake8-annotations
     "BLE",  # flake8-blind-except
+    "B",    # flake8-bugbear
     "COM",  # flake8-commas
     "C4",   # flake8-comprehensions
     "T10",  # flake8-debugger

singer_sdk/authenticators.py

@@ -502,7 +502,7 @@ def update_access_token(self) -> None:
         except requests.HTTPError as ex:
             raise RuntimeError(
                 f"Failed OAuth login, response was '{token_response.json()}'. {ex}",
-            )
+            ) from ex
 
         self.logger.info("OAuth authorization attempt was successful.")
 

singer_sdk/connectors/sql.py

@@ -102,6 +102,7 @@ def create_sqlalchemy_connection(self) -> sqlalchemy.engine.Connection:
             "on the connector currently, make a child class and "
             "add your required method on that connector.",
             DeprecationWarning,
+            stacklevel=2,
         )
         return self._engine.connect().execution_options(stream_results=True)
 
@@ -118,6 +119,7 @@ def create_sqlalchemy_engine(self) -> Engine:
             "`SQLConnector.create_sqlalchemy_engine` is deprecated. Override"
             "`_engine` or sqlalchemy_url` instead.",
             DeprecationWarning,
+            stacklevel=2,
         )
         return self._engine
 
@@ -137,6 +139,7 @@ def connection(self) -> sqlalchemy.engine.Connection:
             "that isn't available on the connector currently, make a child "
             "class and add your required method on that connector.",
             DeprecationWarning,
+            stacklevel=2,
         )
         return self.create_sqlalchemy_connection()
 
@@ -331,7 +334,7 @@ def quote(self, name: str) -> str:
             ],
         )
 
-    @lru_cache()
+    @lru_cache()  # noqa: B019
     def _warn_no_view_detection(self) -> None:
         """Print a warning, but only the first time."""
         self.logger.warning(
@@ -670,10 +673,10 @@ def create_empty_table(
         primary_keys = primary_keys or []
         try:
             properties: dict = schema["properties"]
-        except KeyError:
+        except KeyError as e:
             raise RuntimeError(
                 f"Schema for '{full_table_name}' does not define properties: {schema}",
-            )
+            ) from e
         for property_name, property_jsonschema in properties.items():
             is_primary_key = property_name in primary_keys
             columns.append(
@@ -1042,7 +1045,7 @@ def remove_collation(
         """
         if hasattr(column_type, "collation") and column_type.collation:
             column_type_collation: str = column_type.collation
-            setattr(column_type, "collation", None)
+            column_type.collation = None
             return column_type_collation
         return None
 
@@ -1058,7 +1061,7 @@ def update_collation(
             collation: The colation
         """
         if hasattr(column_type, "collation") and collation:
-            setattr(column_type, "collation", collation)
+            column_type.collation = collation
 
     def _adapt_column_type(
         self,

singer_sdk/helpers/_typing.py

@@ -149,10 +149,7 @@ def is_date_or_datetime_type(type_dict: dict) -> bool:
         True if date or date-time, else False.
     """
     if "anyOf" in type_dict:
-        for type_dict in type_dict["anyOf"]:
-            if is_date_or_datetime_type(type_dict):
-                return True
-        return False
+        return any(is_date_or_datetime_type(option) for option in type_dict["anyOf"])
 
     if "type" in type_dict:
         return type_dict.get("format") in {"date", "date-time"}

singer_sdk/io_base.py

@@ -129,4 +129,4 @@ def _process_unknown_message(self, message_dict: dict) -> None:
         raise ValueError(f"Unknown message type '{record_type}' in message.")
 
     def _process_endofpipe(self) -> None:
-        pass
+        logger.debug("End of pipe reached")

singer_sdk/sinks/core.py

@@ -355,6 +355,7 @@ def _after_process_record(self, context: dict) -> None:
         Args:
             context: Stream partition or context dictionary.
         """
+        self.logger.debug("Processed record: %s", context)
 
     # SDK developer overrides:
 
@@ -447,6 +448,7 @@ def setup(self) -> None:
         Setup is executed once per Sink instance, after instantiation. If a Schema
         change is detected, a new Sink is instantiated and this method is called again.
         """
+        self.logger.info("Setting up %s", self.stream_name)
 
     def clean_up(self) -> None:
         """Perform any clean up actions required at end of a stream.
@@ -455,6 +457,7 @@ def clean_up(self) -> None:
         that may be in use from other instances of the same sink. Stream name alone
         should not be relied on, it's recommended to use a uuid as well.
         """
+        self.logger.info("Cleaning up %s", self.stream_name)
 
     def process_batch_files(
         self,

singer_sdk/streams/rest.py

@@ -493,6 +493,7 @@ def get_new_paginator(self) -> BaseAPIPaginator:
                 "in a future version of the Meltano Singer SDK. "
                 "Override `RESTStream.get_new_paginator` instead.",
                 DeprecationWarning,
+                stacklevel=2,
             )
             return LegacyStreamPaginator(self)  # type: ignore
 

singer_sdk/testing/tap_tests.py

@@ -71,7 +71,7 @@ def test(self) -> None:
             or self.stream.name in self.config.ignore_no_records_for_streams
         ):
             # only warn if this or all streams are set to ignore no records
-            warnings.warn(UserWarning(no_records_message))
+            warnings.warn(UserWarning(no_records_message), stacklevel=2)
         else:
             record_count = len(self.stream_records)
             assert record_count > 0, no_records_message
@@ -90,6 +90,7 @@ def test(self) -> None:
         if diff:
             warnings.warn(
                 UserWarning(f"Fields in catalog but not in records: ({diff})"),
+                stacklevel=2,
             )
 
 
@@ -123,7 +124,7 @@ def test(self) -> None:
                 (r[k] for k in primary_keys or []) for r in self.stream_records
             ]
         except KeyError as e:
-            raise AssertionError(f"Record missing primary key: {str(e)}")
+            raise AssertionError(f"Record missing primary key: {str(e)}") from e
         count_unique_records = len(set(record_ids))
         count_records = len(self.stream_records)
         assert count_unique_records == count_records, (

singer_sdk/testing/templates.py

@@ -232,7 +232,10 @@ def non_null_attribute_values(self) -> list[Any]:
             if r.get(self.attribute_name) is not None
         ]
         if not values:
-            warnings.warn(UserWarning("No records were available to test."))
+            warnings.warn(
+                UserWarning("No records were available to test."),
+                stacklevel=2,
+            )
         return values
 
     @classmethod

tests/external/test_tap_gitlab.py

@@ -23,6 +23,7 @@
             "Could not configure external gitlab tests. "
             f"Config in CI is expected via env vars.\n{e}",
         ),
+        stacklevel=2,
     )
 
 COUNTER = 0

tests/external/test_tap_google_analytics.py

@@ -22,4 +22,5 @@
             "Could not configure external gitlab tests. "
             f"Config in CI is expected via env vars.\n{e}",
         ),
+        stacklevel=2,
     )

tests/samples/test_target_sqlite.py

@@ -103,17 +103,17 @@ def test_sync_sqlite_to_sqlite(
     ):
         try:
             orig_json = json.loads(orig_out)
-        except json.JSONDecodeError:
+        except json.JSONDecodeError as e:
             raise RuntimeError(
                 f"Could not parse JSON in orig line {line_num}: {orig_out}",
-            )
+            ) from e
 
         try:
             tapped_json = json.loads(new_out)
-        except json.JSONDecodeError:
+        except json.JSONDecodeError as e:
             raise RuntimeError(
                 f"Could not parse JSON in new line {line_num}: {new_out}",
-            )
+            ) from e
 
         assert (
             tapped_json["type"] == orig_json["type"]