refc: flake8 fixes

Cloud-Code-AI · Apr 6, 2024 · 64ca28e · 64ca28e
1 parent 010c849
commit 64ca28e
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 30 deletions.
diff --git a/api/github_helper/utils.py b/api/github_helper/utils.py
@@ -49,13 +49,13 @@ def is_github_signature_valid(headers, body):
     Validate the signature of the incoming request against the secret.
     """
     github_secret = os.environ.get("GITHUB_APP_WEBHOOK_SECRET", "").encode()
-    signature = headers.get('X-Hub-Signature-256')
+    signature = headers.get("X-Hub-Signature-256")
 
     if not signature or not github_secret:
         return False
 
-    sha_name, signature = signature.split('=')
-    if sha_name != 'sha256':
+    sha_name, signature = signature.split("=")
+    if sha_name != "sha256":
         return False
 
     mac = hmac.new(github_secret, msg=body, digestmod=hashlib.sha256)

diff --git a/api/main.py b/api/main.py
@@ -1,9 +1,7 @@
 from fastapi import FastAPI, Request, BackgroundTasks, HTTPException
 from fastapi.responses import JSONResponse
 from api.github_helper.pull_requests import process_pull_request, ACTIONS_TO_PROCESS_PR
-from api.github_helper.utils import (
-    is_github_signature_valid
-)
+from api.github_helper.utils import is_github_signature_valid
 import logging
 
 logging.basicConfig(

diff --git a/tests/helpers/test_output.py b/tests/helpers/test_output.py
@@ -2,10 +2,7 @@
 from unittest.mock import patch
 from unittest.mock import patch
 import logging
-from cloudcode.helpers.output import (
-    COLLAPSIBLE_TEMPLATE,
-    json_to_markdown
-)
+from cloudcode.helpers.output import COLLAPSIBLE_TEMPLATE, json_to_markdown
 
 
 @pytest.fixture
@@ -16,46 +13,52 @@ def test_data():
                 "topic": "Code Quality",
                 "comment": "The code is well-structured and easy to read.",
                 "reasoning": "The code follows best practices and coding standards.",
-                "confidence": "High"
+                "confidence": "High",
             },
             {
                 "topic": "Performance",
                 "comment": "The code could be optimized for better performance.",
                 "reasoning": "There are some inefficient loops and data structures used.",
-                "confidence": "Medium"
+                "confidence": "Medium",
             },
             {
                 "topic": "Security",
                 "comment": "NA",
                 "reasoning": "NA",
-                "confidence": "NA"
-            }
+                "confidence": "NA",
+            },
         ]
     }
 
+
 def test_json_to_markdown(test_data, capfd):
     logging.getLogger().setLevel(logging.ERROR)
     expected_output = "## Code Review Feedback\n\n"
     expected_output += "### Code Quality\n\n"
-    expected_output += COLLAPSIBLE_TEMPLATE.format(
-        comment="The code is well-structured and easy to read.",
-        reasoning="The code follows best practices and coding standards.",
-        confidence="High"
-    ) + "\n"
+    expected_output += (
+        COLLAPSIBLE_TEMPLATE.format(
+            comment="The code is well-structured and easy to read.",
+            reasoning="The code follows best practices and coding standards.",
+            confidence="High",
+        )
+        + "\n"
+    )
     expected_output += "### Performance\n\n"
-    expected_output += COLLAPSIBLE_TEMPLATE.format(
-        comment="The code could be optimized for better performance.",
-        reasoning="There are some inefficient loops and data structures used.",
-        confidence="Medium"
-    ) + "\n"
+    expected_output += (
+        COLLAPSIBLE_TEMPLATE.format(
+            comment="The code could be optimized for better performance.",
+            reasoning="There are some inefficient loops and data structures used.",
+            confidence="Medium",
+        )
+        + "\n"
+    )
     expected_output += "### Security\n\n"
-    expected_output += COLLAPSIBLE_TEMPLATE.format(
-        comment="NA",
-        reasoning="NA",
-        confidence="NA"
-    ) + "\n"
+    expected_output += (
+        COLLAPSIBLE_TEMPLATE.format(comment="NA", reasoning="NA", confidence="NA")
+        + "\n"
+    )
 
     output = json_to_markdown(test_data)
     captured = capfd.readouterr()
     assert output == expected_output
-    assert captured.out == ""
+    assert captured.out == ""