Chore: Format code using Ruff 0.9

cratedb_toolkit/io/dynamodb/copy.py

@@ -76,8 +76,7 @@ def start(self):
             logger.info(f"Bulk processor metrics: {metrics}")
 
             logger.info(
-                "Number of records written: "
-                f"success={metrics.count_success_total}, error={metrics.count_error_total}"
+                f"Number of records written: success={metrics.count_success_total}, error={metrics.count_error_total}"
             )
             if metrics.count_success_total == 0:
                 logger.warning("No data has been copied")

cratedb_toolkit/io/kinesis/adapter.py

@@ -56,7 +56,7 @@ def iterator_type(self):
         """
         if self.start.startswith("seqno"):
             raise NotImplementedError(
-                "Consuming Kinesis Stream from sequence number " "not implemented yet, please file an issue."
+                "Consuming Kinesis Stream from sequence number not implemented yet, please file an issue."
             )
         try:
             return self.start_iterator_type_map[self.start]

cratedb_toolkit/io/mongodb/copy.py

@@ -99,8 +99,7 @@ def start(self):
             logger.info(f"Bulk processor metrics: {metrics}")
 
             logger.info(
-                "Number of records written: "
-                f"success={metrics.count_success_total}, error={metrics.count_error_total}"
+                f"Number of records written: success={metrics.count_success_total}, error={metrics.count_error_total}"
             )
             if metrics.count_success_total == 0:
                 logger.warning("No data has been copied")

cratedb_toolkit/io/mongodb/translate.py

@@ -143,7 +143,7 @@ def proportion_string(types: dict) -> str:
     summary = "-- ⬇️ Types: "
     proportions = []
     for type_ in types:
-        proportions.append(f"{type_}: {round((types[type_]['count']/total)*100, 2)}%")
+        proportions.append(f"{type_}: {round((types[type_]['count'] / total) * 100, 2)}%")
     return " " + (summary + ", ".join(proportions))
 
 

cratedb_toolkit/retention/core.py

@@ -43,7 +43,7 @@ def start(self):
         Resolve retention policies to tasks, and invoke them.
         """
         logger.info(
-            f"Connecting to database {self.settings.database.safe}, " f"table {self.settings.policy_table.fullname}"
+            f"Connecting to database {self.settings.database.safe}, table {self.settings.policy_table.fullname}"
         )
 
         msg = f"Starting data retention using '{self.settings.strategy}', cutoff day '{self.settings.cutoff_day}'"

cratedb_toolkit/retention/store.py

@@ -107,7 +107,7 @@ def __init__(self, settings: JobSettings):
         self.settings = settings
 
         logger.info(
-            f"Connecting to database {self.settings.database.safe}, " f"table {self.settings.policy_table.fullname}"
+            f"Connecting to database {self.settings.database.safe}, table {self.settings.policy_table.fullname}"
         )
 
         if self.settings.policy_table.table is None:

cratedb_toolkit/wtf/query_collector.py

@@ -116,9 +116,7 @@ def write_stats_to_db():
         f"VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)"
     )
     update_query_stmt = (
-        f"UPDATE {stmt_log_table} "
-        f"SET calls = ?, avg_duration = ?, nodes = ?, bucket = ?, last_used = ? "
-        f"WHERE id = ?"
+        f"UPDATE {stmt_log_table} SET calls = ?, avg_duration = ?, nodes = ?, bucket = ?, last_used = ? WHERE id = ?"
     )
     write_params = []
     for key in sys_jobs_log.keys():

examples/cloud_import.py

@@ -104,8 +104,7 @@ def obtain_cluster_id() -> str:
 
     if not cluster_id:
         raise ValueError(
-            "Unable to obtain cluster identifier from command line or "
-            "`CRATEDB_CLOUD_CLUSTER_ID` environment variable"
+            "Unable to obtain cluster identifier from command line or `CRATEDB_CLOUD_CLUSTER_ID` environment variable"
         )
 
     return cluster_id

pyproject.toml

@@ -311,6 +311,8 @@ lint.select = [
 ]
 
 lint.extend-ignore = [
+  # Module `xxx` shadows a Python standard-library module
+  "A005",
   # zip() without an explicit strict= parameter
   "B905",
   # df is a bad variable name. Be kinder to your future self.

tests/datasets/test_loading.py

@@ -34,7 +34,7 @@ def test_dataset_replace(database: DatabaseAdapter, dataset: Dataset):
         "KAGGLE_USERNAME" in os.environ and "KAGGLE_KEY" in os.environ
     )
     if dataset.title and "Weather" in dataset.title and not kaggle_auth_exists:
-        raise pytest.skip(f"Kaggle dataset can not be tested " f"without authentication: {dataset.reference}")
+        raise pytest.skip(f"Kaggle dataset can not be tested without authentication: {dataset.reference}")
 
     dataset = load_dataset(dataset.reference)
     tablename = slugify(dataset.reference, separator="_")