diff options
Diffstat (limited to 'bin/mitre/srtool_mitre.py')
-rwxr-xr-x | bin/mitre/srtool_mitre.py | 16 |
1 files changed, 7 insertions, 9 deletions
diff --git a/bin/mitre/srtool_mitre.py b/bin/mitre/srtool_mitre.py index 7504100e..2ac8bc08 100755 --- a/bin/mitre/srtool_mitre.py +++ b/bin/mitre/srtool_mitre.py @@ -28,24 +28,21 @@ import os import sys -import re import xml.etree.ElementTree as ET import argparse import shutil import sqlite3 -from datetime import datetime, date, timedelta +from datetime import datetime, timedelta import pytz -from urllib.request import urlopen, URLError, Request -from urllib.parse import urlparse +from urllib.request import urlopen # load the srt.sqlite schema indexes dir_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__))) sys.path.insert(0, dir_path) from common.srt_schema import ORM -# Constants - +# Setup: srtDbName = 'srt.sqlite' srtErrorLog = 'srt_errors.txt' @@ -267,7 +264,7 @@ def append_cve_database(is_init,file_xml): root = tree.getroot() # Max count for development cycle - cmd_count = 20 if get_override('SRTDBG_MINIMAL_DB') else 0; + cmd_count = 20 if get_override('SRTDBG_MINIMAL_DB') else 0 conn = sqlite3.connect(srtDbName) cur = conn.cursor() @@ -275,7 +272,9 @@ def append_cve_database(is_init,file_xml): cur_ds = conn.cursor() datasource_id = 0 - for i,child in enumerate(root): + i = 0 + for child in root: + i += 1 if not 'Vulnerability' in child.tag: continue summary = _extract_text(child) @@ -455,6 +454,5 @@ def main(argv): print("Command not found") if __name__ == '__main__': - global srtool_basepath srtool_basepath = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(sys.argv[0])))) main(sys.argv[1:]) |