A script to notify when a process finishes execution (#169)

Author: rohan843

README.md

@@ -79,6 +79,7 @@ More information on contributing and the general code of conduct for discussion
 | Planet Simulation                    | [Planet Simulation](https://github.com/DhanushNehru/Python-Scripts/tree/master/planetSimulation)                                              | A simulation of several planets rotating around the sun.                                                            |
 | Remove Background                    | [Remove Background](https://github.com/DhanushNehru/Python-Scripts/tree/master/Remove%20Background)                                           | Removes the background of images.                                                                                   |
 | ROCK-PAPER-SCISSOR                   | [ROCK-PAPER-SCISSOR](https://github.com/DhanushNehru/Python-Scripts/tree/master/ROCK-PAPER-SCISSOR)                                           | A game of Rock Paper Scissors.                                                                                      |
+| Run Then Notify                   | [Run Then Notify](https://github.com/DhanushNehru/Python-Scripts/tree/master/Run%20Then%20Notify)                                           | Runs a slow command and mails you when it completes execution.                                                                                      |
 | Selfie with Python                   | [Selfie_with_Python](https://github.com/DhanushNehru/Python-Scripts/tree/master/Selfie_with_Python)                                           | Take your selfie with python .                                                                                      |
 | Simple TCP Chat Server               | [Simple TCP Chat Server](https://github.com/DhanushNehru/Python-Scripts/tree/master/TCP%20Chat%20Server)                                      | Creates a local server on your LAN for receiving and sending messages!                                              |
 | Snake Water Gun                      | [Snake Water Gun](https://github.com/DhanushNehru/Python-Scripts/tree/master/Snake-Water-Gun)                                                 | A game similar to Rock Paper Scissors.                                                                              |

Run Then Notify/README.md

@@ -0,0 +1,11 @@
+# Run then Notify
+
+This is a python script that will run any terminal command on a system and once the command successfully runs, it will notify you via email of its time taken and the exit code.
+
+This can be useful in case of long running commands.
+
+Please enter your email and password when prompted. These values are _not_ stored anywhere and are only used for the SMTP connection.
+
+> This system has a hardcoded gmail smtp server specified. In case of any other smtp server, please update the value in code.
+>
+> **Note**: You may have to enable `less secure app access` in your gmail account.

Run Then Notify/run_then_notify.py

@@ -0,0 +1,81 @@
+import subprocess
+import sys
+import time
+import socket
+from getpass import getpass
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+
+
+user_email = input("Enter your email: ")
+user_password = getpass()
+smtp_server = "smtp.gmail.com"
+smtp_port = 587  # Port 587 for TLS
+
+
+def send_text_email(subject, message, recipient_email, sender_email, sender_password, smtp_server, smtp_port):
+    # Create a text/plain message
+    msg = MIMEMultipart()
+    msg['From'] = sender_email
+    msg['To'] = recipient_email
+    msg['Subject'] = subject
+
+    # Attach the text message
+    msg.attach(MIMEText(message, 'plain'))
+
+    # Connect to the SMTP server
+    try:
+        server = smtplib.SMTP(smtp_server, smtp_port)
+        server.starttls()
+        server.login(sender_email, sender_password)
+        text = msg.as_string()
+        server.sendmail(sender_email, recipient_email, text)
+        server.quit()
+    except Exception as e:
+        print(f"Email could not be sent: {str(e)}")
+
+
+def run_then_notify(command):
+    try:
+        start_time = time.time()
+        completed_process = subprocess.run(
+            command,
+            text=True,
+            shell=True,
+            stdout=sys.stdout,
+            stderr=subprocess.STDOUT,
+            stdin=sys.stdin
+        )
+
+        end_time = time.time()
+        execution_time = end_time - start_time
+
+        notif_subject = '[Run completed] Your command finished execution'
+        notif_message = f"Command '{command}' completed\n\
+            \twith exit code {completed_process.returncode}\n\
+            \tin {execution_time:.2f} seconds\n\
+            \n\
+            This is an automated message sent from your device {socket.gethostname()}"
+
+        send_text_email(
+            subject=notif_subject,
+            message=notif_message,
+            recipient_email=user_email,
+            sender_email=user_email,
+            sender_password=user_password,
+            smtp_server=smtp_server,
+            smtp_port=smtp_port
+        )
+
+    except Exception as e:
+        print(f"Error: {e}")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: python run_then_notify.py 'your_command_here'")
+        sys.exit(1)
+
+    command = sys.argv[1]
+    run_then_notify(command)