Import curve data before trait data

crocs-muni · Aug 5, 2023 · f6a534d · f6a534d
1 parent a149f31
commit f6a534d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/dissect/utils/database_handler.py b/dissect/utils/database_handler.py
@@ -480,7 +480,16 @@ def export_records(collection_name):
                     with tarfile.open(args.input, "r") as tar:
                         tar.extractall(tmpdir)
 
-                    for file in os.listdir(tmpdir):
+                    files = set(os.listdir(tmpdir))
+                    trait_files = set(
+                        filter(
+                            lambda x: "trait_"
+                            in os.path.basename(x).startswith("trait_"),
+                            files,
+                        )
+                    )
+                    curve_files = files - trait_files
+                    for file in list(curve_files) + list(trait_files):
                         import_file(db, os.path.join(tmpdir, file))
             elif args.input.endswith(".json") or args.input.endswith(".json.bz2"):
                 import_file(db, args.input)