#4 features

emailbomber.py

@@ -1,38 +1,67 @@
-print(
-"""
+import os
+import smtplib
+from dataclasses import dataclass, field
+from functools import partial
+
+WELCOME_MSG = """  # noqa
 ███████╗███╗   ███╗ █████╗ ██╗██╗     ██████╗  ██████╗ ███╗   ███╗██████╗ ███████╗██████╗ 
 ██╔════╝████╗ ████║██╔══██╗██║██║     ██╔══██╗██╔═══██╗████╗ ████║██╔══██╗██╔════╝██╔══██╗
 █████╗  ██╔████╔██║███████║██║██║     ██████╔╝██║   ██║██╔████╔██║██████╔╝█████╗  ██████╔╝
 ██╔══╝  ██║╚██╔╝██║██╔══██║██║██║     ██╔══██╗██║   ██║██║╚██╔╝██║██╔══██╗██╔══╝  ██╔══██╗
 ███████╗██║ ╚═╝ ██║██║  ██║██║███████╗██████╔╝╚██████╔╝██║ ╚═╝ ██║██████╔╝███████╗██║  ██║
 ╚══════╝╚═╝     ╚═╝╚═╝  ╚═╝╚═╝╚══════╝╚═════╝  ╚═════╝ ╚═╝     ╚═╝╚═════╝ ╚══════╝╚═╝  ╚═╝
-            version 1.1 : uses app passowrd now, before it used less secure apps
+            version 1.1 : uses app password now, before it used less secure apps
                                Author : linktr.ee/coderatul
                         Before using please read "prerequisite section"
 """
-)
-sender = input("Enetr Sender's Gmail Id -> ")
-app_password = input("Enter App password -> ")
-receiver = input("Enter Receiver's Gmail Id -> ")
-message = input("Enter Your Message -> ")
-count = int(input("Enter number of mail's to be sent -> "))
-import smtplib 
-server = smtplib.SMTP("smtp.gmail.com",587)
-server.starttls()
-try:
-    server.login(user= sender,password= app_password)
-except:
-    print("")
-    print("""check for the following:
-    1. Read prerequisite section 
-    2. Check app password
-    3. try again and enter details carefully
-    """)
-else:
-    for i in range(count):
-        server.sendmail(from_addr= sender,to_addrs= receiver,msg= message)
-    if count < 2:
-        print(count,"mail have been sent to :",receiver)
+SMTP_SERVER = "smtp.gmail.com"
+SMTP_PORT = 587
+
+
+@dataclass
+class InputArgs:
+    sender: str = field(
+        default_factory=partial(input, "Enter Sender's Gmail Id -> ")
+    )
+    password: str = field(
+        default_factory=partial(input, "Enter App password -> ")
+    )
+    receiver: str = field(
+        default_factory=partial(input, "Enter Receiver's Gmail Id -> ")
+    )
+    message: str = field(
+        default_factory=partial(input, "Enter Your Message -> ")
+    )
+    count: int = field(
+        default_factory=partial(input, "Enter number of mail's to be sent -> ")
+    )
+
+
+def run_bomber(args: InputArgs):
+    server = smtplib.SMTP(SMTP_SERVER, SMTP_PORT)
+    server.starttls()
+    try:
+        server.login(user=args.sender, password=args.password)
+    except smtplib.SMTPException as exc:
+        msg = exc.smtp_error.decode()  # noqa
+        if os.name == "nt":
+            msg = msg.replace("\n", os.linesep)
+        print(msg)
+        return
+
+    for i in range(args.count):
+        server.sendmail(
+            from_addr=args.sender,
+            to_addrs=args.receiver,
+            msg=args.message,
+        )
+    if args.count < 2:
+        print(args.count, "mail have been sent to :", args.receiver)
     else:
-        print(count,"mails have been sent to :",receiver)
+        print(args.count, "mails have been sent to :", args.receiver)
     server.close()
+
+
+if __name__ == "__main__":
+    print(WELCOME_MSG)
+    run_bomber(InputArgs())