From 18d1541f056cc28219dd993e89ed64f8bd73aca7 Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Thu, 23 Jan 2025 13:59:20 +0300
Subject: feat: Add function to prepare files and workspace directory for
 computation.

---
 gn3/computations/rqtl2.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/gn3/computations/rqtl2.py b/gn3/computations/rqtl2.py
index 95d8225..9377fd5 100644
--- a/gn3/computations/rqtl2.py
+++ b/gn3/computations/rqtl2.py
@@ -76,3 +76,20 @@ def create_file(file_path):
     except FileExistsError:
         return False, "File Already Exists"
 
+
+def prepare_files(tmpdir):
+    """Prepare necessary files and workspace dir  for computation."""
+    workspace_dir = os.path.join(tmpdir, str(uuid.uuid4())) #
+    os.makedirs(workspace_dir)
+    input_file = os.path.join(workspace_dir, f"rqtl2-input-{uuid.uuid4()}.json")
+    output_file = os.path.join(workspace_dir, f"rqtl2-output-{uuid.uuid4()}.json")
+
+    # to ensure streaming api has access to file  even after computation ends
+    # .. Create the log file outside the workspace_dir
+    log_file = os.path.join(tmpdir, f"rqtl2-log-{uuid.uuid4()}")
+    for file_path in [input_file, output_file, log_file]:
+        create_file(file_path)
+    return workspace_dir, input_file, output_file, log_file
+
+
+
-- 
cgit 1.4.1