aboutsummaryrefslogtreecommitdiffstats
path: root/bin/mitre/srtool_mitre.py
diff options
context:
space:
mode:
Diffstat (limited to 'bin/mitre/srtool_mitre.py')
-rwxr-xr-xbin/mitre/srtool_mitre.py16
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:])