diff --git a/src/parse_nvd_cves.py b/src/parse_nvd_cves.py index d07864e..dc6a627 100755 --- a/src/parse_nvd_cves.py +++ b/src/parse_nvd_cves.py @@ -16,6 +16,8 @@ print(nvd_files) conn = get_connection() +cursor = conn.cursor() +cursor.execute("USE {}".format(DB_NAME)) for f in nvd_files: f = open(join(NVDPATH,f))