Dodanie obsługi zapisywania/wyświetlania z brancha.

[redakcja.git] / apps / explorer / models.py

diff --git a/apps/explorer/models.py b/apps/explorer/models.py
index e69de29..2b2e8e8 100644 (file)
--- a/apps/explorer/models.py
+++ b/apps/explorer/models.py
@@ -0,0 +1,16 @@
+import os
+
+from django.conf import settings
+
+
+def get_image_folders():
+    return sorted(fn for fn in os.listdir(os.path.join(settings.MEDIA_ROOT, settings.IMAGE_DIR)) if not fn.startswith('.'))
+
+
+def get_images_from_folder(folder):
+    return sorted(settings.MEDIA_URL + settings.IMAGE_DIR + '/' + folder + '/' + fn for fn 
+            in os.listdir(os.path.join(settings.MEDIA_ROOT, settings.IMAGE_DIR, folder))
+            if not fn.startswith('.'))
+
+def user_branch(user):
+    return 'personal_'+user.username