feat: Add method to write list of data to csv file.

1 files changed, 23 insertions, 1 deletions

diff --git a/gn3/computations/rqtl2.py b/gn3/computations/rqtl2.py
index 052ceae..d515e74 100644
--- a/gn3/computations/rqtl2.py
+++ b/gn3/computations/rqtl2.py
@@ -1 +1,23 @@
-"""Module contains functions for  parsing and  processing rqtl2 input and output."""
+import os
+import csv
+import uuid
+import json
+from pathlib import Path
+from typing import Dict
+
+
+def write_to_csv(work_dir, file_name, data:list[dict],
+                      headers= None, delimiter=","):
+    """Functions to write data list  to csv file
+    if headers is not provided use the keys for first boject.
+    """
+    file_path = os.path.join(work_dir, file_name)
+    if headers is None and data:
+        headers = data[0].keys()
+    with open(file_path, "w", encoding="utf-8") as file_handler:
+        writer = csv.DictWriter(file_handler, fieldnames=headers,
+                               delimiter=delimiter)
+        writer.writeheader()
+        for row in  data:
+            writer.writerow(row)
+        return file_path