From 2b3edbaa463cae2ecf6f4dcc8b333f23247bd6e9 Mon Sep 17 00:00:00 2001 From: Matthieu Helleboid Date: Fri, 20 Jan 2023 20:51:03 +0100 Subject: [PATCH] use pathlib glob --- .../management/commands/document_importer.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/src/documents/management/commands/document_importer.py b/src/documents/management/commands/document_importer.py index 873a5841d..eeae68e65 100644 --- a/src/documents/management/commands/document_importer.py +++ b/src/documents/management/commands/document_importer.py @@ -83,13 +83,10 @@ class Command(BaseCommand): self.manifest = json.load(f) manifest_paths.append(main_manifest_path) - for root, dirs, files in os.walk(self.source): - for file in files: - if file.endswith("-manifest.json"): - doc_manifest_path = os.path.normpath(os.path.join(root, file)) - with open(doc_manifest_path) as f: - self.manifest += json.load(f) - manifest_paths.append(doc_manifest_path) + for file in Path(self.source).glob("**/*-manifest.json"): + with open(file) as f: + self.manifest += json.load(f) + manifest_paths.append(file) version_path = os.path.normpath(os.path.join(self.source, "version.json")) if os.path.exists(version_path):