From 39938077fa953e74559eda297dd36aedb63ac33d Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Tue, 30 Mar 2021 13:01:40 +0300
Subject: Refactor getting subcategories and resources to 2 helper methods

---
 pydis_site/apps/resources/utils.py | 42 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 pydis_site/apps/resources/utils.py

(limited to 'pydis_site/apps/resources/utils.py')

diff --git a/pydis_site/apps/resources/utils.py b/pydis_site/apps/resources/utils.py
new file mode 100644
index 00000000..1855fc80
--- /dev/null
+++ b/pydis_site/apps/resources/utils.py
@@ -0,0 +1,42 @@
+import typing as t
+from pathlib import Path
+
+import yaml
+
+
+def get_resources(path: Path) -> t.List[t.Dict]:
+    """Loads resource YAMLs from provided path."""
+    resources = []
+
+    for item in path.iterdir():
+        if item.is_file() and item.suffix == ".yaml" and item.name != "_category_info.yaml":
+            resources.append(yaml.safe_load(item.read_text()))
+
+    return resources
+
+
+def get_subcategories(path: Path) -> t.List[t.Dict]:
+    """Loads resources subcategories with their resources by provided path."""
+    subcategories = []
+
+    for item in path.iterdir():
+        if item.is_dir() and item.joinpath("_category_info.yaml").exists():
+            subcategories.append({
+                "category_info": {
+                    **yaml.safe_load(
+                        item.joinpath("_category_info.yaml").read_text()
+                    ),
+                    "raw_name": item.name
+                },
+                "resources": [
+                    yaml.safe_load(subitem.read_text())
+                    for subitem in item.iterdir()
+                    if (
+                        subitem.is_file()
+                        and subitem.suffix == ".yaml"
+                        and subitem.name != "_category_info.yaml"
+                    )
+                ]
+            })
+
+    return subcategories
-- 
cgit v1.2.3