Check for local Folders

loosolab · Dec 4, 2018 · d8655fd · d8655fd
1 parent b82dc6a
commit d8655fd
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/RegGTFExtractor.py b/RegGTFExtractor.py
@@ -5,10 +5,22 @@
 from Modules.ucsc.ucsc import UcscGtf
 from Modules.Uniquifier import UniqueFilter
 from Modules.SaveResults import ResultSaver
+import os
+
+
+def check_for_local_folder():
+
+    if not os.path.isdir("./EnsemblData"):
+
+        os.mkdir("./EnsemblData")
+
+    if not os.path.isdir( "./UCSCData" ):
+        os.mkdir( "./UCSCData" )
 
 
 def main_script(org, tissuetype=None):
 
+    check_for_local_folder()
     ucsc = UcscGtf(org)
     ense = Ensembl(org)
     print("Getting Unique Results")