Ruff Format

Author: TimPansino

newrelic/api/message_trace.py

@@ -154,7 +154,9 @@ def _nr_message_trace_wrapper_(wrapped, instance, args, kwargs):
     return FunctionWrapper(wrapped, _nr_message_trace_wrapper_)
 
 
-def message_trace(library, operation, destination_type, destination_name, params=None, terminal=True, async_wrapper=None):
+def message_trace(
+    library, operation, destination_type, destination_name, params=None, terminal=True, async_wrapper=None
+):
     if params is None:
         params = {}
 

newrelic/hooks/database_aiomysql.py

@@ -57,7 +57,9 @@ def __await__(self):
         except StopIteration as e:
             # Catch the StopIteration and wrap the return value.
             cursor = e.value
-            wrapped_cursor = self.__cursor_wrapper__(cursor, self._nr_dbapi2_module, self._nr_connect_params, self._nr_cursor_args)
+            wrapped_cursor = self.__cursor_wrapper__(
+                cursor, self._nr_dbapi2_module, self._nr_connect_params, self._nr_cursor_args
+            )
             return wrapped_cursor  # Return here instead of raising StopIteration to properly follow generator protocol
 
 

newrelic/hooks/framework_tornado.py

@@ -247,12 +247,13 @@ async def wrapper(req, raise_error):
             try:
                 response = await wrapped(req, raise_error=raise_error)
             except Exception as e:
-                response = getattr(e, 'response', None)
+                response = getattr(e, "response", None)
                 raise
             finally:
                 if response:
                     trace.process_response_headers(response.headers.get_all())
             return response
+
     return wrapper
 
 

tests/datastore_aiomysql/test_sqlalchemy.py

@@ -13,18 +13,13 @@
 # limitations under the License.
 
 from aiomysql.sa import create_engine
+from sqlalchemy import Column, Float, Integer, String
 from sqlalchemy.orm import declarative_base
-from sqlalchemy import Integer, String, Column, Float
 from sqlalchemy.schema import CreateTable, DropTable
-
 from testing_support.db_settings import mysql_settings
 from testing_support.util import instance_hostname
-from testing_support.validators.validate_database_trace_inputs import (
-    validate_database_trace_inputs,
-)
-from testing_support.validators.validate_transaction_metrics import (
-    validate_transaction_metrics,
-)
+from testing_support.validators.validate_database_trace_inputs import validate_database_trace_inputs
+from testing_support.validators.validate_transaction_metrics import validate_transaction_metrics
 
 from newrelic.api.background_task import background_task
 
@@ -38,13 +33,14 @@
 
 Base = declarative_base()
 
+
 class ABCModel(Base):
     __tablename__ = TABLE_NAME
 
     a = Column(Integer, primary_key=True)
     b = Column(Float)
     c = Column(String(100))
-    
+
 
 ABCTable = ABCModel.__table__
 
@@ -99,6 +95,7 @@ async def exercise(engine):
     (f"Datastore/instance/MySQL/{HOST}/{PORT}", 8),
 ]
 
+
 @validate_transaction_metrics(
     "test_sqlalchemy:test_execute_via_engine",
     scoped_metrics=SCOPED_METRICS,

tests/testing_support/validators/validate_database_trace_inputs.py

@@ -55,7 +55,9 @@ def _bind_params(
             assert isinstance(cursor_params[0], tuple)
             assert isinstance(cursor_params[1], dict)
 
-        assert sql_parameters is None or isinstance(sql_parameters, sql_parameters_type), f"Expected: {sql_parameters_type} Got: {type(sql_parameters)}"
+        assert sql_parameters is None or isinstance(sql_parameters, sql_parameters_type), (
+            f"Expected: {sql_parameters_type} Got: {type(sql_parameters)}"
+        )
 
         if execute_params is not None:
             assert len(execute_params) == 2