From 9f7895540746096aff287697a75d295efe68f569 Mon Sep 17 00:00:00 2001
From: Fontrodona Nicolas <nicolas.fontrodona@ens-lyon.fr>
Date: Thu, 5 Mar 2020 16:40:15 +0100
Subject: [PATCH] src/db_utils/config.py: add cpu variable.
 src/db_utils/__main__.py: add the function to fill frequency table

---
 src/db_utils/__main__.py | 2 +-
 src/db_utils/config.py   | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/db_utils/__main__.py b/src/db_utils/__main__.py
index e7b637da..84b9a55f 100755
--- a/src/db_utils/__main__.py
+++ b/src/db_utils/__main__.py
@@ -28,7 +28,7 @@ def launcher(logging_level: str = "INFO"):
     logging.info('Filling splicing lore tables')
     fill_splicing_lore_data('DISABLE')
     logging.info('Filling frequency tables')
-    fill_frequency_tables('DISABLE')
+    fill_frequency_tables(Config.cpu, 'DISABLE')
 
 
 launcher(logging_level="DEBUG")
diff --git a/src/db_utils/config.py b/src/db_utils/config.py
index 6155a2f4..0ec6908a 100755
--- a/src/db_utils/config.py
+++ b/src/db_utils/config.py
@@ -7,12 +7,14 @@ Description: Configuration class
 """
 
 from pathlib import Path
+import multiprocessing as mp
 
 
 class Config:
     """
     A class containing every parameters used in the submodule db_utils
     """
+    cpu = mp.cpu_count()
     data = Path(__file__).parents[2] / 'data'
     results = Path(__file__).parents[2] / "results"
     db_file = results / 'chia_pet_database.db'
-- 
GitLab