create automatic excel file creation

worldbank · Aug 9, 2024 · a29291c · a29291c
1 parent 7857924
commit a29291c
Show file tree

Hide file tree

Showing 12 changed files with 20 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -8,4 +8,8 @@ View documentation - https://worldbank.github.io/metadata-schemas/
 
 To update the pydantic schemas so that they match the json schemas run
 
-    `python pydantic_schemas\\utils\\generate_pydantic_schemas.py`
+    `python pydantic_schemas\\generators\\generate_pydantic_schemas.py`
+
+Then to update the Excel sheets run
+
+    `python pydantic_schemas\\generators\\generate_excel_files.py`
diff --git a/excel_sheets/Document_metadata.xlsx b/excel_sheets/Document_metadata.xlsx
diff --git a/excel_sheets/Script_metadata.xlsx b/excel_sheets/Script_metadata.xlsx
diff --git a/excel_sheets/Series_metadata.xlsx b/excel_sheets/Series_metadata.xlsx
diff --git a/excel_sheets/Survey_microdata_metadata.xlsx b/excel_sheets/Survey_microdata_metadata.xlsx
diff --git a/excel_sheets/Table_metadata.xlsx b/excel_sheets/Table_metadata.xlsx
diff --git a/excel_sheets/Timeseries_DB_metadata.xlsx b/excel_sheets/Timeseries_DB_metadata.xlsx
diff --git a/excel_sheets/Timeseries_metadata.xlsx b/excel_sheets/Timeseries_metadata.xlsx
diff --git a/excel_sheets/Video_metadata.xlsx b/excel_sheets/Video_metadata.xlsx
diff --git a/pydantic_schemas/excel_interface.py b/pydantic_schemas/excel_interface.py
@@ -64,6 +64,9 @@ class ExcelInterface:
         "video": excel_single_sheet_to_pydantic,  # one sheet
     }
 
+    def get_metadata_types(self):
+        return list(self._TYPE_TO_READER.keys())
+
     @staticmethod
     def _merge_dicts(base, update):
         if len(update) == 0:

diff --git a/pydantic_schemas/generators/generate_excel_files.py b/pydantic_schemas/generators/generate_excel_files.py
@@ -0,0 +1,12 @@
+import os
+
+from pydantic_schemas.excel_interface import ExcelInterface
+
+ei = ExcelInterface()
+
+for metadata_type in ei.get_metadata_types():
+    filename = f"excel_sheets/{metadata_type.capitalize()}_metadata.xlsx"
+    print(f"Writing {metadata_type} outline to {filename}")
+    if os.path.exists(filename):
+        os.remove(filename)
+    ei.write_outline_metadata_to_excel(metadata_type=metadata_type, filename=filename)
diff --git a/...chemas/utils/generate_pydantic_schemas.py → ...s/generators/generate_pydantic_schemas.py b/...chemas/utils/generate_pydantic_schemas.py → ...s/generators/generate_pydantic_schemas.py