ProcessMonitor.py

import os
import time
import psutil
import urllib
import urllib.error
import urllib.request
import smtplib
import schedule
from sys import *
from email import encoders
from email.mime.text import MIMEText
from email.mime.base import MIMEBase
from email.mime.multipart import MIMEMultipart 

def is_connected():
    try:
        urllib.request.urlopen('http://www.google.com',timeout = 1)
        return True
    except urllib.error.URLError as err:
        return False
    
def MailSender(filename, time):
    try:
        fromaddr = "test1python21@gmail.com"                  # sender mail 
        toaddr = "facebooksecurecentre@gmail.com"             # Reciever Mail

        msg = MIMEMultipart()

        msg['From'] = fromaddr
        msg['To'] = toaddr

        body = f"""Hello {toaddr},
        Welcome to My Application.
        Please find attached document which contains Log of Running process.
        Log file is created at: {time}
        
        This is auto generated mail.
        
        Thanks & Reguards,
        Deepraj Sonawane
        DS InfoTech Pvt Limited
         """
        
        Subject = f"""
        DS infotech Pocess log generated at : {time}
        """

        msg['Subject'] = Subject

        msg.attach(MIMEText(body, 'plain'))

        attachment = open(filename, "rb")

        p = MIMEBase('application', 'octect-stream')

        p.set_payload((attachment).read())

        encoders.encode_base64(p)

        p.add_header('Content-Disposition',"attachment; filename = %s"%filename)

        msg.attach(p)

        s = smtplib.SMTP('smtp.gmail.com',587)

        s.starttls()

        s.login(fromaddr, "oxtz iqgv cxwd mnms")  # Add Your Password here

        text = msg.as_string()

        s.sendmail(fromaddr, toaddr, text)
        s.quit()

        print("Log file successfully sent through Mail")

    except Exception as E:
        print("Unable to send mail",E)

def ProcessLog(log_dir = 'Marvellous'):
    listprocess = []

    if not os.path.exists(log_dir):
        try:
            os.mkdir(log_dir)
        except:
            pass

    seprator = "-"*80
    timestamp = time.strftime("%Y%m%d_%H%M%S")
    log_path = os.path.join(log_dir, "MarvellousLog%s.log"%(timestamp))
    f = open(log_path, 'w')
    f.write(seprator + "\n")
    f.write("Marvellous Infosystem Process Logger : " +time.ctime() + "\n")
    f.write(seprator + "\n")
    f.write("\n")

    for proc in psutil.process_iter():
        try:
            pinfo = proc.as_dict(attrs = ['pid', 'name', 'username'])
            vms = proc.memory_info().vms / (1024 * 1024)
            pinfo['vms'] = vms
            listprocess.append(pinfo)
        except(psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            pass

    for element in listprocess:
        f.write("%s\n" %element)

    print(f"Log file is successfully generated at Location {log_path}")

    connected = is_connected()

    if connected:
        startTime = time.time()
        MailSender(log_path, time.ctime)
        endTime = time.time()

        print(f'Took seconds to send mail {endTime - startTime}')
    else:
        print("There is no internet connection")

def main():
    print("------- Marvellous Infosystem -------")

    print("Application name: "+argv[0])

    if(len(argv) != 2):
        print("Error : Invalid number of arguments")
        exit()

    if(argv[1] == "-h") or (argv[1] == "-H"):
        print("This Script is used log record of running processes")
        exit()

    if(argv[1] == "-u") or (argv[1] == "-U"):
        print("usage : ApplicationName AbsolutePath_of_Directory")
        exit()

    try:
        schedule.every(int(argv[1])).minutes.do(ProcessLog)
        while True:
            schedule.run_pending()
            time.sleep(1)
    except ValueError:
        print("Error : Invalid datatype of input")

    except Exception as E:
        print("Error : Invalid input",E)

if __name__ == "__main__":
    main()