FIXED: bug

Author: Ruskonert

cve.py

@@ -16,8 +16,7 @@ def __init__(self, db):
             self.host = db.host
             self.password = db.password
             self.cur = db.cur
-            if self.database is not None:
-                self.use_database(self.database)
+            self.execption_list = {}
         else:
             raise TypeInstanceException("The variable 'db' is not instance of MySqlBase")
 

cvemysql.py

@@ -271,6 +271,20 @@ def help(error=None):
     if year_based_task:
         for year in list_year:
             export_filepath.append(configure_task(base, table_name, year, filename, file_path))
+        
+        if len(base.execption_list) != 0:
+            filename = "err-{time}.log".format(time=datetime.now()).replace(" ", "").replace(":", ".")
+            current_key = list(base.execption_list.keys())
+            with open(filename, 'w+', encoding='utf-8') as fs:
+                for ex in current_key:
+                    for v in list(base.execption_list.get(ex)):
+                        if isinstance(v, str) and re.compile("CVE-.").match(v):
+                            value = v
+                            fs.write("실패: Column에 값을 넣을 수 없습니다. -> {cve} | Reason: {r}\n".format(cve=v, r=ex))
+                            break
+                fs.close()
+            print("\n경고: {count} elements could not be inserted into the table. See {f} for more reasons.".format(count=len(base.execption_list), f=filename))
+            base.execption_list.clear()
     else:
         export_filepath.append(configure_task(base, table_name, None, filename, file_path))
 

nvdnist.py

@@ -20,8 +20,6 @@
 NVD_NIST_JSON_URL = 'https://nvd.nist.gov/feeds/json/cve/1.0/'
 NVD_NIST_XML_URL = 'https://nvd.nist.gov/feeds/xml/cve/2.0/'
 
-execption_list = {}
-
 class ThreadList:
     def __init__(self, process):
         self._entry = 0
@@ -409,21 +407,6 @@ def table_insert(self, tablename, jsonCollection):
                 time.sleep(0.1)
         except (KeyboardInterrupt, SystemExit):
             sys.exit(-1)
-
-
-        if len(execption_list) != 0:
-            filename = "err-{time}.log".format(time=datetime.now()).replace(" ", "").replace(":", ".")
-            current_key = list(execption_list.keys())
-            with open(filename, 'w+', encoding='utf-8') as fs:
-                for ex in current_key:
-                    for v in list(execption_list.get(ex)):
-                        if isinstance(v, str) and re.compile("CVE-.").match(v):
-                            value = v
-                            fs.write("실패: Column에 값을 넣을 수 없습니다. -> {cve} | Reason: {r}\n".format(cve=v, r=ex))
-                            break
-                fs.close()
-            print("\n경고: {count} elements could not be inserted into the table. See {f} for more reasons.".format(count=len(execption_list), f=filename))
-            xecption_list.clear()
 
         print("\nAll data has been inserted.")