浏览代码

Add script to generate catalog information

sonicaj 2 年之前
父节点
当前提交
d9d0e669ca
共有 1 个文件被更改,包括 20 次插入0 次删除
  1. 20 0
      update_catalog_detail.py

+ 20 - 0
update_catalog_detail.py

@@ -0,0 +1,20 @@
+#!/usr/bin/python3
+import json
+import os
+
+from catalog_validation.items.catalog import get_items_in_trains, retrieve_train_names, retrieve_trains_data
+
+
+def get_trains() -> dict:
+    location: str = os.getcwd()
+    preferred_trains: list = []
+
+    trains_to_traverse = retrieve_train_names(location)
+    items = get_items_in_trains(trains_to_traverse, location)
+
+    return retrieve_trains_data(items, location, preferred_trains, trains_to_traverse)[0]
+
+
+if __name__ == '__main__':
+    with open('catalog.json', 'w') as f:
+        f.write(json.dumps(get_trains(), indent=4))