generate general info for file output

1 files changed, 27 insertions, 12 deletions

diff --git a/wqflask/wqflask/correlation/pre_computes.py b/wqflask/wqflask/correlation/pre_computes.py
index c91e3862..81111a3c 100644
--- a/wqflask/wqflask/correlation/pre_computes.py
+++ b/wqflask/wqflask/correlation/pre_computes.py
@@ -210,15 +210,30 @@ def fetch_lmdb_info(db_path: str, db_target_name: str):
     e.g  ProbeSets: ProbestFreeze__112_
     """
     # open db_read if results return none write the file write the file target
-    with lmdb.open(target_file_path, readonly=True, lock=False) as env:
-        with env.begin() as txn:
-            target_key = __generate_file_name__(db_target_name)
-            dataset = txn.get(target_key.encode())
-            if dataset:
-                return {
-                    "lmdb_target_path": f"{db_path}data.mdb",
-                    "lmdb_target_key": target_key,
-                    "file_type": "lmdb",
-                }
-
-            return {}
+    try:
+        with lmdb.open(target_file_path, readonly=True, lock=False) as env:
+            with env.begin() as txn:
+                target_key = __generate_file_name__(db_target_name)
+                dataset = txn.get(target_key.encode())
+                if dataset:
+                    return {
+                        "lmdb_target_path": f"{db_path}data.mdb",
+                        "lmdb_target_key": target_key,
+                        "file_type": "lmdb",
+                    }
+    except Exception:
+        return {}
+
+
+def generate_general_info(trait_name, sample_names,
+                          strains_vals, file_type_info):
+    if not file_type_info:
+        #! code should not be reached at this point
+        pass
+    # implement fetch code
+    return {
+        "trait_name": target_name,
+        "primary_sample_names": primary_sample_names,
+        "primary_strains_vals": strain_vals,
+        **file_type_info
+    }