WIP: drs import

038af537 · Cresson Remi · 48ee58bd · 038af537
Commit 038af537 authored 3 years ago by Cresson Remi
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 1 deletion
+2 -1
--- a/test/drs_import.py
+++ b/test/drs_import.py
 import spot
 import argparse
+import os
 from glob import glob

 # Arguments
@@ -9,7 +10,7 @@ params = parser.parse_args()

 # List files
 print("List files...")
-files = glob("{}/*/".format(params.root_dir), recursive=True)
+files = [y for x in os.walk(params.root_dir) for y in glob(os.path.join(x[0], "DIM*.XML"))]
 print(len(files))

 for i in range(0,100):