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 3b5848b..2b2e8e8 100644 (file)
--- a/apps/explorer/models.py
+++ b/apps/explorer/models.py
@@ -4,11 +4,13 @@ from django.conf import settings
 
 
 def get_image_folders():
-    return [fn for fn in os.listdir(settings.IMAGE_DIR) if not fn.startswith('.')]
+    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 ['/media/images/' + folder + '/' + fn for fn 
-            in os.listdir(os.path.join(settings.IMAGE_DIR, folder))
-            if not fn.startswith('.')]
+    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