Create-scans-mail.gmp.py

import sys
from base64 import b64decode
from pathlib import Path
import time
import datetime

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.xxx.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 progress


def create_report(tname,reportid):
  time1 = datetime.datetime.now()
  timestamp=time1.strftime('%Y-%m-%d')
  pdf_filename = "/tmp/" + tname + "-" + timestamp + ".pdf"
  pdf_report_format_id = "c402cc3e-b531-11e1-9163-406186ea4fc5"
  response = gmp.get_report(
      report_id=reportid, report_format_id=pdf_report_format_id
   )
  report_element = response[0]
  content = "".join(report_element.itertext())
  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('Done. PDF created: ' + str(pdf_path))
  send_mail("scanner","geheim","scanner@xxx.de","technik@xxx.de","tname" + "-" + timestamp ,pdf_filename ,pdf_filename )

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:
    time.sleep(30)

create_report(task_name,report_id)