lint

Author: CTY-git

patchwork/steps/SendEmail/SendEmail.py

@@ -20,12 +20,11 @@ def __init__(self, inputs):
         self.smtp_host = inputs.get("smtp_host", "smtp.gmail.com")
         self.smtp_port = int(inputs.get("smtp_port", 465))
 
-
     def run(self) -> dict:
         msg = MIMEText(mustache_render(self.body, self.email_template_value))
-        msg['Subject'] = mustache_render(self.subject, self.email_template_value)
-        msg['From'] = self.sender_email
-        msg['To'] = self.recipient_email
+        msg["Subject"] = mustache_render(self.subject, self.email_template_value)
+        msg["From"] = self.sender_email
+        msg["To"] = self.recipient_email
 
         # TODO: support smtp without ssl
         with smtplib.SMTP_SSL(self.smtp_host, self.smtp_port) as smtp_server:

patchwork/steps/SendEmail/typed.py

@@ -1,4 +1,4 @@
-from typing_extensions import Annotated, Dict, TypedDict, Any
+from typing_extensions import Annotated, Any, Dict, TypedDict
 
 from patchwork.common.utils.step_typing import StepTypeConfig
 
@@ -16,5 +16,6 @@ class SendEmailInputs(__SendEmailRequiredInputs, total=False):
     smtp_host: str
     smtp_port: int
 
+
 class SendEmailOutputs(TypedDict):
     is_slack_message_sent: bool

patchwork/steps/__init__.py

@@ -47,6 +47,7 @@
 from patchwork.steps.ScanDepscan.ScanDepscan import ScanDepscan
 from patchwork.steps.ScanSemgrep.ScanSemgrep import ScanSemgrep
 from patchwork.steps.ScanSonar.ScanSonar import ScanSonar
+from patchwork.steps.SendEmail.SendEmail import SendEmail
 from patchwork.steps.SimplifiedLLM.SimplifiedLLM import SimplifiedLLM
 from patchwork.steps.SimplifiedLLMOnce.SimplifiedLLMOnce import SimplifiedLLMOnce
 from patchwork.steps.SlackMessage.SlackMessage import SlackMessage
@@ -99,6 +100,7 @@
     "ScanDepscan",
     "ScanSemgrep",
     "ScanSonar",
+    "SendEmail",
     "SimplifiedLLM",
     "SimplifiedLLMOnce",
     "SimplifiedLLMOncePB",