DeforaNetworks/pentext
DeforaNetworks/pentext
3
0
Fork 0
Code Releases Activity

Pylint fixes

Browse Source
This commit is contained in:
Peter Mosmans 2016-09-13 19:26:43 +10:00
parent 5476c0f466
commit 17da0c9846
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
chatops/python/validate_report.py
View File

@ -172,7 +172,7 @@ def validate_files(filenames, options):
scans = [] scans = []
for filename in filenames: for filename in filenames:
if (filename.lower().endswith('.xml') or if (filename.lower().endswith('.xml') or
filename.lower().endswith('xml"')): filename.lower().endswith('xml"')):
if SNIPPETDIR not in filename and TEMPLATEDIR not in filename: if SNIPPETDIR not in filename and TEMPLATEDIR not in filename:
if (OFFERTE in filename and options['offer']) or \ if (OFFERTE in filename and options['offer']) or \
(REPORT in filename and not options['no_report']): (REPORT in filename and not options['no_report']):
@ -313,7 +313,8 @@ def validate_type(tree, filename, options):
else: else:
if attribute == 'threatLevel' and root.attrib[attribute] not in \ if attribute == 'threatLevel' and root.attrib[attribute] not in \
('Low', 'Moderate', 'Elevated', 'High', 'Extreme'): ('Low', 'Moderate', 'Elevated', 'High', 'Extreme'):
print('[-] threatLevel is not Low, Moderate, High, Elevated or Extreme: {0}'.format(root.attrib[attribute])) print('[-] threatLevel is not Low, Moderate, High, Elevated or Extreme: {0}'.
format(root.attrib[attribute]))
result = False result = False
if attribute == 'type' and (options['capitalization'] and not \ if attribute == 'type' and (options['capitalization'] and not \
is_capitalized(root.attrib[attribute])): is_capitalized(root.attrib[attribute])):
@ -472,16 +473,17 @@ def add_include(filename, identifier, findings):
finding_section.append(new_finding) finding_section.append(new_finding)
tree.write(filename, encoding="utf-8", xml_declaration=True, pretty_print=True) tree.write(filename, encoding="utf-8", xml_declaration=True, pretty_print=True)
def close_file(filename): def close_file(filename):
""" """
Replace placeholder with proper XML include. Replace placeholder with proper XML include.
""" """
f = open(filename,'r') f = open(filename, 'r')
filedata = f.read() filedata = f.read()
f.close() f.close()
newdata = filedata.replace("placeholderinclude","xi:include") newdata = filedata.replace("placeholderinclude", "xi:include")
fileout = filename fileout = filename
f = open(fileout,'w') f = open(fileout, 'w')
f.write(newdata) f.write(newdata)
f.close() f.close()
tree = ElementTree.parse(filename, ElementTree.XMLParser(strip_cdata=False)) tree = ElementTree.parse(filename, ElementTree.XMLParser(strip_cdata=False))