[WIP] fix path

cnas/pages/gallery.py

@@ -5,6 +5,7 @@
 
 from util.file_util import is_image_file
 from util.config_path import get_gallery_path
+from util.system_path import get_gallery_thumbnail_path
 
 from genhtml.w3c.html import html
 from genhtml.w3c.section import section
@@ -23,43 +24,53 @@ def __init__(self):
         super().__init__()
         self.pictures = []
 
+        self.gallery_path = get_gallery_path()
+        self.thumbnail_path = get_gallery_thumbnail_path()
+        self.current_path = self.get_session("current_path")
+        if not self.current_path:
+            self.current_path = self.gallery_path
+        if not os.path.abspath(self.gallery_path) in\
+            os.path.abspath(self.current_path):
+            self.current_path = self.gallery_path
+
+        self.get_path()
+        self.get_list()
+
     def get_path(self):
-        _gallery_path = None
         if self.json_data and self.json_data.get("path"):
-            _gallery_path = self.json_data.get("path")
-        else:
-          _gallery_path = get_gallery_path()
-        print(_gallery_path)
-
-        if _gallery_path is None or not os.path.exists(_gallery_path):
-            return str(error("No gallery path"))
+            _rel_path = self.json_data.get("path")
+            _full = os.path.join(self.current_path, _rel_path)
+            self.set_session("current_path", os.path.abspath(_full))
+            print(_full)
+            print(_rel_path)
 
     def get_list(self):
         # folder
         for _folder in os.listdir(self.current_path):
             _path = os.path.join(self.current_path, _folder)
             if os.path.isdir(_path):
-                _div.append(
+                self.pictures.append(
                     photo_builder(
                         src="static/images/folder.png",
                         path=_folder))
 
         # images
         for _file in os.listdir(self.current_path):
             _path = os.path.join(self.current_path, _file)
-            _rel_path = ""
-            _thumb_nail_path = ""
             if not is_image_file(_file):
                 continue
-            if not os.path.exists(_path):
-                continue
-            if os.path.exist(_thumb_nail_path):
-                _div.append(
+
+            _rel_path = os.path.relpath(self.current_path, self.gallery_path)
+            _rel_full = os.path.join(_rel_path, _file)
+            _thumb_nail_path = os.path.join(self.thumbnail_path, _rel_path)
+
+            if os.path.exists(_thumb_nail_path):
+                self.pictures.append(
                     photo_builder(
-                        src="gallery_thumbnail/" + _rel_path,
+                        src="gallery_thumbnail/" + _rel_full,
                         path=_file))
             else:
-                _div.append(
+                self.pictures.append(
                     photo_builder(src="static/images/no_cache.png"))
 
 

cnas/pages/page.py

@@ -6,11 +6,11 @@ def __init__(self):
         if request.is_json:
             self.json_data = request.get_json()
 
-    def set_session(key, value):
+    def set_session(self, key, value):
         session[key] = value
 
-    def get_session(key):
+    def get_session(self, key):
         return session.get(key)
 
-    def clear_session(key):
+    def clear_session(self, key):
         session.pop(key, None)

cnas/route/route.py

@@ -16,12 +16,12 @@ def index_page() -> str:
     def gallery_page() -> str:
         return str(gallery())
 
-    @app.route('/gallery_file/<filename>')
+    @app.route('/gallery_file/<path:filename>')
     def gallery_file(filename):
         directory = CONFIG.get('gallery_path')
         return send_from_directory(directory, filename)
 
-    @app.route('/gallery_thumbnail/<filename>')
+    @app.route('/gallery_thumbnail/<path:filename>')
     def gallery_thumbnail(filename):
         directory = get_gallery_thumbnail_path()
         return send_from_directory(directory, filename)

cnas/static/javascripts/cnas_gallery.js

@@ -10,8 +10,8 @@ $(document).ready(function(){
             contentType: 'application/json',
             data: JSON.stringify(data),
             success: function(response) {
-              console.log(response)
               document.body.innerHTML = response
+              window.location.reload();
             },
             error: function(xhr, status, error) {
               console.log(error)