From e3a62548c816f5348e4d91a6328783a73d1ecbf0 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 17 Oct 2024 14:37:26 -0500
Subject: Cleanup: Delete all extracted files after processing.

---
 scripts/rqtl2/phenotypes_qc.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'scripts/rqtl2')

diff --git a/scripts/rqtl2/phenotypes_qc.py b/scripts/rqtl2/phenotypes_qc.py
index b2b02ed..76ef7bf 100644
--- a/scripts/rqtl2/phenotypes_qc.py
+++ b/scripts/rqtl2/phenotypes_qc.py
@@ -1,5 +1,6 @@
 """Run quality control on phenotypes-specific files in the bundle."""
 import sys
+import shutil
 import tempfile
 from pathlib import Path
 from zipfile import ZipFile
@@ -167,6 +168,9 @@ def run_qc(# pylint: disable=[too-many-arguments]
     #       - Check the 3 checks above for phenose and phenonum values too
     # qc_phenose_files(…)
     # qc_phenonum_files(…)
+
+    #       - Delete all extracted files
+    shutil.rmtree(extractiondir)
     raise NotImplementedError("WIP!")
 
 
-- 
cgit v1.2.3