schema: Pass schema to flatten tool

#15 (comment) Open-Telecoms-Data/lib-cove-ofds#10
Open-Telecoms-Data · Nov 8, 2022 · 12a047a · 12a047a
1 parent dcc1335
commit 12a047a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cove_ofds/process.py b/cove_ofds/process.py
@@ -63,10 +63,13 @@ def process(self, process_data: dict) -> dict:
 
         os.makedirs(output_dir, exist_ok=True)
 
+        schema = OFDSSchema
+
         unflatten_kwargs = {
             "output_name": os.path.join(output_dir, "unflattened.json"),
             "root_list_path": "networks",
             "input_format": _get_file_type(input_filename),
+            "schema": schema.data_schema_url,
         }
 
         flattentool.unflatten(input_filename, **unflatten_kwargs)
@@ -112,10 +115,13 @@ def process(self, process_data: dict) -> dict:
 
         os.makedirs(output_dir, exist_ok=True)
 
+        schema = OFDSSchema
+
         unflatten_kwargs = {
             "output_name": os.path.join(output_dir, "unflattened.json"),
             "root_list_path": "networks",
             "input_format": "csv",
+            "schema": schema.data_schema_url,
         }
 
         flattentool.unflatten(self.supplied_data.upload_dir(), **unflatten_kwargs)
@@ -302,9 +308,12 @@ def process(self, process_data: dict) -> dict:
 
         os.makedirs(self.output_dir, exist_ok=True)
 
+        schema = OFDSSchema
+
         flatten_kwargs = {
             "output_name": self.output_dir,
             "root_list_path": "networks",
+            "schema": schema.data_schema_url,
         }
 
         flattentool.flatten(process_data["json_data_filename"], **flatten_kwargs)