From 038af537d05cc4209436d84f8a3b5c375bcc2038 Mon Sep 17 00:00:00 2001
From: Remi Cresson <remi.cresson@irstea.fr>
Date: Tue, 15 Feb 2022 15:38:14 +0100
Subject: [PATCH] WIP: drs import

---
 test/drs_import.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/drs_import.py b/test/drs_import.py
index b62bf7c..694d9b3 100644
--- a/test/drs_import.py
+++ b/test/drs_import.py
@@ -1,5 +1,6 @@
 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):
-- 
GitLab