restructure initialize database

cve-search · Sep 2, 2021 · 7f86aef · 7f86aef
1 parent 6c2ce2b
commit 7f86aef
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 11 deletions.
diff --git a/CveXplore/cli_cmds/db_cmds/commands.py b/CveXplore/cli_cmds/db_cmds/commands.py
@@ -15,7 +15,7 @@ def update_cmd(ctx):
     ctx.obj["data_source"].database.update()
 
 
-@db_cmd.group("populate", invoke_without_command=True, help="Populate the database")
+@db_cmd.group("initialize", invoke_without_command=True, help="Initialize the database")
 @click.pass_context
-def populate_cmd(ctx):
-    ctx.obj["data_source"].database.populate()
+def initialize_cmd(ctx):
+    ctx.obj["data_source"].database.initialize()
diff --git a/CveXplore/database/maintenance/main_updater.py b/CveXplore/database/maintenance/main_updater.py
@@ -37,14 +37,12 @@ def update(self):
 
         self.datasource.set_handlers_for_collections()
 
-    def populate(self):
+    def initialize(self):
 
-        for source in self.sources:
-            populator = source["updater"]()
-            populator.populate()
+        cpe_pop = CPEDownloads()
+        cpe_pop.populate()
 
-        for post in self.posts:
-            indexer = post["updater"]()
-            indexer.create_indexes()
+        cve_pop = CVEDownloads()
+        cve_pop.populate()
 
-        self.datasource.set_handlers_for_collections()
+        self.update()