import sys
from pathlib import Path
import lxml.etree as et

if len(sys.argv) < 1:
    print("usage: [script].py nessusfile.nessus")
    sys.exit(1)

file = sys.argv[1]

if isinstance(file, str):
    file_path = Path(file)
    content = file_path.read_text()
else:
    content = file.read().decode("utf-8")
    file.close()

tree = et.fromstring(content)

for bad in tree.xpath("//risk_factor"):
    if bad.text.lower() not in ["critical", "high"]:
        # print("Filtering item")
        # remove item from item's parent
        bad.getparent().getparent().remove(bad.getparent())

print(str(et.tostring(tree, pretty_print=True).decode()))