lint and move self.response=None into try block

Author: kimi-p

datadog_lambda/wrapper.py

@@ -153,9 +153,8 @@ def __call__(self, event, context, **kwargs):
             self._after(event, context)
 
     def _before(self, event, context):
-        self.response = None
         try:
-
+            self.response = None
             set_cold_start()
             submit_invocations_metric(context)
             self.trigger_tags = extract_trigger_tags(event, context)

tests/test_wrapper.py

@@ -275,22 +275,19 @@ def lambda_handler(event, context):
             ]
         )
 
-    @patch('datadog_lambda.wrapper.extract_trigger_tags')
+    @patch("datadog_lambda.wrapper.extract_trigger_tags")
     def test_5xx_sends_errors_metric_and_set_tags(self, mock_extract_trigger_tags):
         mock_extract_trigger_tags.return_value = {
             "function_trigger.event_source": "api-gateway",
-            "function_trigger.event_source_arn":
-                "arn:aws:apigateway:us-west-1::/restapis/1234567890/stages/prod",
+            "function_trigger.event_source_arn": "arn:aws:apigateway:us-west-1::/restapis/1234567890/stages/prod",
             "http.url": "70ixmpl4fl.execute-api.us-east-2.amazonaws.com",
             "http.url_details.path": "/prod/path/to/resource",
             "http.method": "GET",
         }
+
         @datadog_lambda_wrapper
         def lambda_handler(event, context):
-            return {
-                "statusCode": 500,
-                "body": "fake response body"
-            }
+            return {"statusCode": 500, "body": "fake response body"}
 
         lambda_event = {}