Start-task-complete.gmp-py

import sys from base64 import b64decode from pathlib import Path import time import smtplib from email.mime.text import MIMEText from email.mime.multipart import MIMEMultipart from email.mime.base import MIMEBase from email import encoders import os.path import datetime import os

user="heinz.test" password="geheim" mail_from="scanner@xxx.com" mail_to="technik@hubis.de" mail_server="thor.xxxn.de" mail_port=587

task_name=sys.argv[4] portlist="9ddce1ae-57e7-11e1-b13c-406186ea4fc5" scanconf="698f691e-7489-11df-9d8c-002264764cea" openvas="08b69003-5fc2-4037-a479-93b440211c73" pdf_report_format_id = "c402cc3e-b531-11e1-9163-406186ea4fc5"

def send_mail(user,password,from_email,to_email,subject,message,file_location):

 msg = MIMEMultipart()
 msg['From'] = from_email
 msg['To'] =   to_email
 msg['Subject'] = subject
 msg.attach(MIMEText(message, 'plain'))
 # Setup the attachment
 filename = os.path.basename(file_location)
 attachment = open(file_location, "rb")
 part = MIMEBase('application', 'octet-stream')
 part.set_payload(attachment.read())
 encoders.encode_base64(part)
 part.add_header('Content-Disposition', "attachment; filename= %s" % filename)
 # Attach the attachment to the MIMEMultipart object
 msg.attach(part)
 server = smtplib.SMTP('thor.tuxmen.de', 587)
 server.starttls()
 server.login(user, password)
 text = msg.as_string()
 server.sendmail(from_email, to_email, text)
 server.quit()

def find_task(tname):

res = gmp.get_tasks()
for i, task in enumerate(res.xpath('task')):
    id = task.xpath('@id')[0]
    name = task.xpath('name/text()')[0]
    if name == tname:
      return id

def status_task(tname):

res = gmp.get_tasks()
for i, task in enumerate(res.xpath('task')):
    id = task.xpath('@id')[0]
    name = task.xpath('name/text()')[0]
    if name == tname:
      status   = task.xpath('status/text()')[0]
      return status

def progress_task(tname):

res = gmp.get_tasks()
for i, task in enumerate(res.xpath('task')):
    id = task.xpath('@id')[0]
    name = task.xpath('name/text()')[0]
    if name == tname:
      progress = task.xpath('progress/text()')[0]
      return progres

def create_report(tname,reportid):

   pdf_filename = "/tmp/" + tname + "-" + timestamp + ".pdf"
   pdf_report_format_id = "c402cc3e-b531-11e1-9163-406186ea4fc5"
   response = gmp.get_report(
       report_id=report_id, report_format_id=pdf_report_format_id
   )
   report_element = response.find("report")
   content = report_element.find("report_format").tail
   binary_base64_encoded_pdf = content.encode('ascii')
   binary_pdf = b64decode(binary_base64_encoded_pdf)
   pdf_path = Path(pdf_filename).expanduser()
   pdf_path.write_bytes(binary_pdf)
   print("PDF: " + pdf_filename + " created")
   return pdf_filename

time1 = datetime.datetime.now() timestamp=time1.strftime('%Y-%m-%d_%H_%M') print("Task start: " + task_name + " " + timestamp) task_id = find_task(task_name) res=gmp.start_task(task_id) report_id=res[0].text

while True:

 if status_task(task_name) == "Done":
   break
 else:
   print("Status: " +  status_task(task_name) + " " + progress_task(task_name) + "%" )
   time.sleep(30)

time1 = datetime.datetime.now() timestamp=time1.strftime('%Y-%m-%d_%H_%M')

pdf = create_report(task_name,report_id) sub = task_name + "-" + timestamp send_mail(user,password,mail_from,mail_to,sub,pdf,pdf ) print("Task end: " + task_name + " " + timestamp)