initial push of photoblog

1 files changed, 20 insertions, 0 deletions

diff --git a/app/services/images.py b/app/services/images.py
new file mode 100644
index 0000000..bd02384
--- /dev/null
+++ b/app/services/images.py
@@ -0,0 +1,20 @@
+from pathlib import Path
+import random
+
+print(str(Path.cwd()))
+root_dir = Path.cwd()
+photos_dir = Path(f"{root_dir}/photos")
+
+image_cache = set()
+
+def cache_images():
+    for file_path in photos_dir.rglob("*"):
+        if file_path.is_file():
+            image_path = file_path.relative_to(root_dir).name
+            image_cache.add(str(image_path))
+
+def get_all_images():
+    if not image_cache:
+        cache_images()
+    shuffled = random.sample(list(image_cache), k=len(image_cache))
+    return shuffled