DeforaNetworks/pentext
3
0
Fork 0
Code Releases Activity

Display unique finding code as well

Browse Source
This commit is contained in:
Peter Mosmans
2017-10-16 13:16:41 +10:00
parent 2051d75563
commit 56b6c3b24d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
chatops/python/pentext_id.py
View File

@@ -38,7 +38,7 @@ except ImportError as exception:
sys.exit(-1) sys.exit(-1)
VERSION = '0.1' VERSION = '0.2.0'
class LogFormatter(logging.Formatter): class LogFormatter(logging.Formatter):
@@ -146,6 +146,8 @@ def main():
try: try:
# Parse document into an ElementTree # Parse document into an ElementTree
tree = etree.parse(filename, etree.XMLParser(strip_cdata=False)) tree = etree.parse(filename, etree.XMLParser(strip_cdata=False))
# Read finding code
code = tree.getroot().attrib['findingCode']
# Grab all elements from the findings section # Grab all elements from the findings section
for elements in tree.xpath('//section[@id="findings"]'): for elements in tree.xpath('//section[@id="findings"]'):
# Iterate through all elements, looking for the findings # Iterate through all elements, looking for the findings
@@ -160,7 +162,7 @@ def main():
locate_finding(findings, args) locate_finding(findings, args)
else: else:
for i, href in enumerate(findings, start=1): for i, href in enumerate(findings, start=1):
print("{0:2d} {1}".format(i, href)) print("{0}-{1:02d} {2}".format(code, i, href))
if __name__ == "__main__": if __name__ == "__main__":